Skip to content
This repository was archived by the owner on Nov 6, 2025. It is now read-only.
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 40 additions & 0 deletions tests/plotting_test/data_viewers_test/ploting_data_all_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,6 +107,9 @@ def test_plot_0D_1D_uniform(self, qtbot):

assert dwa_back == dwa_1D

viewer.parent.close()
viewer.parent.deleteLater()

def test_plot_1D_0D_uniform(self, qtbot):
#%%
NX = 100
Expand All @@ -123,12 +126,18 @@ def test_plot_1D_0D_uniform(self, qtbot):

viewer = dwa_1D.plot('qt')
assert isinstance(viewer, Viewer1D)

qtbot.addWidget(viewer.parent)

with tempfile.TemporaryDirectory() as d:
with DataSaverLoader(Path(d).joinpath('mydatafile.h5')) as saver_loader:
saver_loader.add_data('/RawData', dwa_1D)
dwa_back = saver_loader.load_data('/RawData/Data00', load_all=True)
assert dwa_back == dwa_1D

viewer.parent.close()
viewer.parent.deleteLater()


def test_plot_1D_0D_spread(self, qtbot):

Expand Down Expand Up @@ -165,6 +174,10 @@ def test_plot_1D_0D_spread(self, qtbot):
viewer = data1D_spread.plot('qt')
assert isinstance(viewer, Viewer1D)

qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

def test_plot_0D_1D_spread(self, qtbot, get_h5saver):
h5saver = get_h5saver
data_saver = DataEnlargeableSaver(h5saver)
Expand Down Expand Up @@ -193,6 +206,9 @@ def test_plot_0D_1D_spread(self, qtbot, get_h5saver):
assert dwa_back.inav[2] == data_to_append
viewer = dwa_back.plot('qt')
assert isinstance(viewer, ViewerND)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

class Test2DPlot:
def test_plot_0D_2D_uniform(self, qtbot):
Expand All @@ -210,6 +226,9 @@ def test_plot_0D_2D_uniform(self, qtbot):
assert data2D.dim == 'Data2D'
viewer = data2D.plot('qt')
assert isinstance(viewer, Viewer2D)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

@pytest.mark.parametrize('nav_index', (0, 1))
def test_plot_1D_1D_uniform(self, qtbot, nav_index):
Expand All @@ -228,6 +247,9 @@ def test_plot_1D_1D_uniform(self, qtbot, nav_index):
assert data2D.dim == 'DataND'
viewer = data2D.plot('qt')
assert isinstance(viewer, Viewer2D)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

def test_plot_2D_0D_uniform(self, qtbot):
NX = 100
Expand All @@ -245,6 +267,9 @@ def test_plot_2D_0D_uniform(self, qtbot):
assert data2D.dim == 'DataND'
viewer = data2D.plot('qt')
assert isinstance(viewer, Viewer2D)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

def test_plot_2D_0D_spread(self, qtbot):
N = 100
Expand All @@ -270,6 +295,9 @@ def test_plot_2D_0D_spread(self, qtbot):

viewer = data2D_spread.plot('qt')
assert isinstance(viewer, Viewer2D)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

def test_plot_1D_1D_spread(self, qtbot):
N = 10
Expand Down Expand Up @@ -297,6 +325,9 @@ def test_plot_1D_1D_spread(self, qtbot):

viewer = data2D_spread.plot('qt')
assert isinstance(viewer, ViewerND)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

class Test3DPlot:
@pytest.mark.parametrize('nav_index', ((0,), (1,), (2,), (0, 1), (0, 2), (1, 2)))
Expand All @@ -312,6 +343,9 @@ def test_plot_0D_3D_uniform(self, qtbot, get_3D_array, nav_index):
assert data3D.dim == 'DataND'
viewer = data3D.plot('qt')
assert isinstance(viewer, ViewerND)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

class Test4DPlot:
@pytest.mark.parametrize('nav_indexes',
Expand All @@ -326,6 +360,9 @@ def test_plot_4D_uniform(self, qtbot, get_4D, nav_indexes):
assert dwa.dim == 'DataND'
viewer = dwa.plot('qt')
assert isinstance(viewer, ViewerND)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()

def test_plot_4D_spread(self, qtbot):
N = 100
Expand Down Expand Up @@ -355,3 +392,6 @@ def test_plot_4D_spread(self, qtbot):

viewer = dwa.plot('qt')
assert isinstance(viewer, ViewerND)
qtbot.addWidget(viewer.parent)
viewer.parent.close()
viewer.parent.deleteLater()