diff options
Diffstat (limited to 'silx/gui/data/test/test_dataviewer.py')
| -rw-r--r-- | silx/gui/data/test/test_dataviewer.py | 28 |
1 files changed, 22 insertions, 6 deletions
diff --git a/silx/gui/data/test/test_dataviewer.py b/silx/gui/data/test/test_dataviewer.py index 5a0de0b..dd3114a 100644 --- a/silx/gui/data/test/test_dataviewer.py +++ b/silx/gui/data/test/test_dataviewer.py @@ -24,7 +24,7 @@ # ###########################################################################*/ __authors__ = ["V. Valls"] __license__ = "MIT" -__date__ = "10/04/2017" +__date__ = "22/08/2017" import os import tempfile @@ -42,8 +42,6 @@ from silx.gui.data.DataViewerFrame import DataViewerFrame from silx.gui.test.utils import SignalListener from silx.gui.test.utils import TestCaseQt -from silx.gui.hdf5.test import _mock - try: import h5py except ImportError: @@ -111,6 +109,24 @@ class AbstractDataViewerTests(TestCaseQt): self.assertEqual(DataViewer.RAW_MODE, widget.displayMode()) self.assertIn(DataViewer.PLOT2D_MODE, availableModes) + def test_plot_2d_bool(self): + data = numpy.zeros((10, 10), dtype=numpy.bool) + data[::2, ::2] = True + widget = self.create_widget() + widget.setData(data) + availableModes = set([v.modeId() for v in widget.currentAvailableViews()]) + self.assertEqual(DataViewer.RAW_MODE, widget.displayMode()) + self.assertIn(DataViewer.PLOT2D_MODE, availableModes) + + def test_plot_2d_complex_data(self): + data = numpy.arange(3 ** 2, dtype=numpy.complex) + data.shape = [3] * 2 + widget = self.create_widget() + widget.setData(data) + availableModes = set([v.modeId() for v in widget.currentAvailableViews()]) + self.assertEqual(DataViewer.RAW_MODE, widget.displayMode()) + self.assertIn(DataViewer.PLOT2D_MODE, availableModes) + def test_plot_3d_data(self): data = numpy.arange(3 ** 3) data.shape = [3] * 3 @@ -212,6 +228,7 @@ class AbstractDataViewerTests(TestCaseQt): self.assertTrue(view not in widget.availableViews()) self.assertTrue(view not in widget.currentAvailableViews()) + class TestDataViewer(AbstractDataViewerTests): def create_widget(self): return DataViewer() @@ -225,11 +242,10 @@ class TestDataViewerFrame(AbstractDataViewerTests): class TestDataView(TestCaseQt): def createComplexData(self): - line = [1, 2j, 3+3j, 4] + line = [1, 2j, 3 + 3j, 4] image = [line, line, line, line] cube = [image, image, image, image] - data = numpy.array(cube, - dtype=numpy.complex) + data = numpy.array(cube, dtype=numpy.complex) return data def createDataViewWithData(self, dataViewClass, data): |