tomwer 1.3.27__py3-none-any.whl → 1.4.0rc0__py3-none-any.whl

tomwer/tests/orangecontrib/tomwer/widgets/control/tests/test_nxtomo_concatenate_ow.py

@@ -0,0 +1,64 @@
+import os
+from tomoscan.series import Series
+import pickle
+
+from ewoksorange.bindings.qtapp import QtEvent
+
+from orangecanvas.scheme.readwrite import literal_dumps
+
+from tomwer.tests.conftest import qtapp  # noqa F401
+from tomwer.core.utils.scanutils import MockNXtomo
+
+from orangecontrib.tomwer.widgets.control.NXtomoConcatenate import NXtomoConcatenateOW
+
+
+def test_NabuHelicalPrepareWeightsDoubleOW(
+    qtapp,  # noqa F811
+    tmp_path,
+):
+    """simple test of the _DeltaBetaSelectorDialog"""
+    widget = NXtomoConcatenateOW()
+    widget._loadSettings()
+    # test settings serialization
+    pickle.dumps(widget._ewoks_default_inputs)
+    literal_dumps(widget._ewoks_default_inputs)
+
+    scan_1 = MockNXtomo(
+        scan_path=os.path.join(tmp_path, "scan1"),
+        n_proj=10,
+        n_ini_proj=10,
+        scan_range=180,
+        dim=20,
+        energy=12.3,
+    ).scan
+    scan_2 = MockNXtomo(
+        scan_path=os.path.join(tmp_path, "scan2"),
+        n_proj=10,
+        n_ini_proj=10,
+        scan_range=180,
+        dim=20,
+        energy=12.3,
+    ).scan
+
+    output_scan_file = os.path.join(tmp_path, "concatenation.nx")
+    assert not os.path.exists(output_scan_file)
+
+    widget.widget.setConfiguration(
+        {
+            "output_file": output_scan_file,
+        }
+    )
+
+    # process task
+    finished = QtEvent()
+
+    def finished_callback():
+        finished.set()
+
+    executor = widget.task_executor_queue
+    executor.sigComputationEnded.connect(finished_callback)
+    widget._process_series(series=Series(iterable=[scan_1, scan_2]))
+
+    # wait until processing is finished
+    assert finished.wait(timeout=10)
+    assert os.path.exists(output_scan_file)

tomwer/tests/orangecontrib/tomwer/widgets/control/tests/test_nxtomomill.py

@@ -0,0 +1,133 @@
+# coding: utf-8
+from __future__ import annotations
+
+import gc
+import os
+import shutil
+import tempfile
+import time
+
+from orangecanvas.scheme.readwrite import literal_dumps
+from processview.core.manager.manager import DatasetState, ProcessManager
+from silx.gui import qt
+from silx.gui.utils.testutils import TestCaseQt
+from tomoscan.esrf.scan.mock import MockEDF
+
+from orangecontrib.tomwer.widgets.control.EDF2NXTomomillOW import EDF2NXOW
+from orangecontrib.tomwer.widgets.control.NXTomomillOW import NXTomomillOW
+from tomwer.core.scan.blissscan import BlissScan
+from tomwer.core.scan.nxtomoscan import NXtomoScan
+from tomwer.core.utils.scanutils import MockBlissAcquisition
+
+
+class TestNXTomomillOw(TestCaseQt):
+    """
+    Test the NXTomomillOW widget
+    """
+
+    def setUp(self):
+        super().setUp()
+        self.tempdir = tempfile.mkdtemp()
+        mock = MockBlissAcquisition(
+            n_sample=1,
+            n_sequence=1,
+            n_scan_per_sequence=3,
+            n_darks=2,
+            n_flats=2,
+            output_dir=self.tempdir,
+        )
+        self.scan = BlissScan(
+            master_file=mock.samples[0].sample_file,
+            proposal_file=mock.proposal_file,
+            entry="1.1",
+        )
+        self.widget = NXTomomillOW()
+
+    def tearDown(self):
+        self.widget.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.widget.close()
+        self.widget = None
+        shutil.rmtree(self.tempdir)
+        gc.collect()
+
+    def test(self):
+        self.widget.add(self.scan.master_file)
+        self.widget._sendAll()
+
+    def test_literal_dumps(self):
+        self.widget._saveNXTomoCfgFile(cfg_file="")
+        literal_dumps(self.widget._ewoks_default_inputs)
+
+
+class TestEDF2NXOW(TestCaseQt):
+    """
+    Test the EDF2NXOW widget
+    """
+
+    def setUp(self):
+        super().setUp()
+        self.tempdir = tempfile.mkdtemp()
+        self._edf_folder_1 = os.path.join(self.tempdir, "edf_scan_1")
+        self._edf_folder_2 = os.path.join(self.tempdir, "edf_scan_2_")
+        MockEDF(
+            scan_path=self._edf_folder_1,
+            n_radio=10,
+            n_ini_radio=10,
+            n_extra_radio=0,
+            dim=128,
+            dark_n=1,
+            flat_n=1,
+        )
+        MockEDF(
+            scan_path=self._edf_folder_2,
+            n_radio=10,
+            n_ini_radio=10,
+            n_extra_radio=0,
+            dim=128,
+            dark_n=1,
+            flat_n=1,
+        )
+        self.widget = EDF2NXOW()
+        self._process_manager = ProcessManager()
+
+    def tearDown(self):
+        self.widget.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.widget.close()
+        self.widget = None
+        shutil.rmtree(self.tempdir)
+        gc.collect()
+
+    def test(self):
+        self.widget.add(self._edf_folder_1)
+        self.widget.add(self._edf_folder_2)
+        self.widget._sendAll()
+        self.qapp.processEvents()
+        timeout = 10
+        processing_queue = self.widget.task_executor_queue
+        while not processing_queue.is_available and not processing_queue.empty():
+            self.qapp.processEvents()
+            time.sleep(0.1)
+            timeout -= 0.1
+            if timeout <= 0:
+                raise TimeoutError
+        # need to call one more time the processEvents tpo call callbacks and status notification
+        for i in range(3):
+            self.qapp.processEvents()
+            time.sleep(0.1)
+
+        # check the two output files exists and status of the processing is done...
+        for folder in (self._edf_folder_1, self._edf_folder_2):
+            scan = NXtomoScan(scan=folder + ".nx", entry="entry")
+            self.assertTrue(os.path.exists(scan.master_file))
+
+            self.assertEqual(
+                self._process_manager.get_dataset_state(
+                    dataset_id=scan.get_identifier(),
+                    process=self.widget,
+                ),
+                DatasetState.SUCCEED,
+            )
+
+    def test_literal_dumps(self):
+        self.widget._saveNXTomoCfgFile(cfg_file="")
+        literal_dumps(self.widget._ewoks_default_inputs)

tomwer/tests/orangecontrib/tomwer/widgets/control/tests/test_reduce_dark_flat_selector.py

@@ -0,0 +1,40 @@
+import gc
+import pickle
+import numpy
+
+from silx.gui import qt
+from silx.gui.utils.testutils import TestCaseQt
+
+from orangecontrib.tomwer.widgets.control.ReduceDarkFlatSelectorOW import (
+    ReduceDarkFlatSelectorOW,
+)
+from orangecanvas.scheme.readwrite import literal_dumps
+
+
+class TestEmailOW(TestCaseQt):
+    def setUp(self):
+        super().setUp()
+        self.widget = ReduceDarkFlatSelectorOW()
+        self.widget._dialog.addReduceFrames(
+            {
+                "reduce_frames_name": "new dict",
+                -1: numpy.random.random(100 * 100).reshape(100, 100),
+                2: numpy.random.random(100 * 100).reshape(100, 100),
+            },
+            selected=(-1, 2),
+        )
+
+    def tearDown(self):
+        self.widget.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.widget.close()
+        self.widget = None
+        gc.collect()
+
+    def test(self):
+        self.widget.show()
+
+    def test_serializing(self):
+        pickle.dumps(self.widget.getConfiguration())
+
+    def test_literal_dumps(self):
+        literal_dumps(self.widget.getConfiguration())

tomwer/tests/orangecontrib/tomwer/widgets/control/tests/test_singletomoobj.py

@@ -0,0 +1,40 @@
+import gc
+import os
+import pickle
+import shutil
+import tempfile
+
+import numpy
+from orangecanvas.scheme.readwrite import literal_dumps
+from silx.gui import qt
+from silx.gui.utils.testutils import TestCaseQt
+
+from orangecontrib.tomwer.widgets.control.SingleTomoObjOW import SingleTomoObjOW
+from tomwer.core.volume.hdf5volume import HDF5Volume
+
+
+class TestSingleTomoObjOW(TestCaseQt):
+    def setUp(self):
+        super().setUp()
+        self.window = SingleTomoObjOW()
+        self.folder = tempfile.mkdtemp()
+        self.volume_1 = HDF5Volume(
+            file_path=os.path.join(self.folder, "vol_file.hdf5"),
+            data_path="entry0000",
+            data=numpy.linspace(0, 10, 100 * 100 * 3).reshape(3, 100, 100),
+        )
+        self.volume_1.save()
+
+    def tearDown(self):
+        self.window.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.window.close()
+        self.window = None
+        shutil.rmtree(self.folder)
+        gc.collect()
+
+    def test(self):
+        self.window.show()
+        self.window.widget.setTomoObject(self.volume_1)
+        # test serialization
+        pickle.dumps(self.window._tomo_obj_setting)
+        literal_dumps(self.window._tomo_obj_setting)

tomwer/tests/orangecontrib/tomwer/widgets/control/tests/test_timerow.py

@@ -0,0 +1,25 @@
+# coding: utf-8
+from __future__ import annotations
+
+
+import gc
+
+from silx.gui import qt
+from silx.gui.utils.testutils import TestCaseQt
+
+from orangecontrib.tomwer.widgets.control.TimerOW import TimerOW
+
+
+class TestTimerOW(TestCaseQt):
+    def setUp(self):
+        super().setUp()
+        self.widget = TimerOW()
+
+    def tearDown(self):
+        self.widget.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.widget.close()
+        self.widget = None
+        gc.collect()
+
+    def test(self):
+        self.widget.show()

tomwer/tests/orangecontrib/tomwer/widgets/control/tests/test_tomoobj_series.py

@@ -0,0 +1,96 @@
+import gc
+import os
+import shutil
+import tempfile
+
+import numpy
+from silx.gui import qt
+from silx.gui.utils.testutils import SignalListener, TestCaseQt
+
+from orangecontrib.tomwer.widgets.control.TomoObjSeriesOW import TomoObjSeriesOW
+from tomwer.core.utils.scanutils import HDF5MockContext
+from tomwer.core.volume.hdf5volume import HDF5Volume
+
+
+class TestTomoObjSeriesOW(TestCaseQt):
+    """
+    Test the TomoObjSeriesOW orange widget
+    """
+
+    def setUp(self):
+        super().setUp()
+        self.window = TomoObjSeriesOW()
+        self.folder = tempfile.mkdtemp()
+
+        # connect to the series created signal
+        self._listener = SignalListener()
+        self.window._widget.sigSeriesSelected.connect(self._listener)
+
+    def tearDown(self):
+        self.window.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.window.close()
+        self.window = None
+        self._listener = None
+        shutil.rmtree(self.folder)
+        gc.collect()
+
+    def test_with_volumes(self):
+        """test the TomoObjSeriesOW with volumes"""
+        volume_1 = HDF5Volume(
+            file_path=os.path.join(self.folder, "vol_file.hdf5"),
+            data_path="entry0000",
+            data=numpy.linspace(0, 10, 100 * 100 * 3).reshape(3, 100, 100),
+        )
+        volume_1.save()
+        volume_2 = HDF5Volume(
+            file_path=os.path.join(self.folder, "vol_file2.hdf5"),
+            data_path="entry0001",
+            data=numpy.linspace(60, 120, 100 * 100 * 5).reshape(5, 100, 100),
+        )
+        volume_2.save()
+
+        assert self._listener.callCount() == 0
+        assert len(self.window._widget._widget.getSelectedSeries()) == 0
+        # add the volume to the possible objects to be added
+        self.window.addTomoObj(volume_1)
+        self.window.addTomoObj(volume_2)
+        assert len(self.window._widget._widget.getSelectedSeries()) == 0
+
+        # add volume to the series defined
+        newSerieWidget = (
+            self.window._widget._widget._seriesDefinitionWidget._manualDefWidget._newSeriesWidget
+        )
+        newSerieWidget.addTomoObjToCurrentSeries(volume_1)
+        assert len(self.window._widget._widget.getSelectedSeries()) == 1
+        newSerieWidget.addTomoObjToCurrentSeries(
+            volume_1
+        )  # try adding twice the same object
+        assert len(self.window._widget._widget.getSelectedSeries()) == 1
+        newSerieWidget.addTomoObjToCurrentSeries(volume_2)
+        assert len(self.window._widget._widget.getSelectedSeries()) == 2
+        assert self._listener.callCount() == 0
+        self.window._widget._selectButton.released.emit()
+        self.qapp.processEvents()
+        assert self._listener.callCount() == 1
+
+    def test_with_scans(self):
+        """test the TomoObjSeriesOW with scans"""
+        with HDF5MockContext(
+            scan_path=os.path.join(self.folder, "test", "scan"), n_proj=100
+        ) as scan:
+            assert self._listener.callCount() == 0
+
+            # add the scan to possible objects to be added
+            self.window.addTomoObj(scan)
+            assert len(self.window._widget._widget.getSelectedSeries()) == 0
+
+            # add scan to the series defined
+            newSerieWidget = (
+                self.window._widget._widget._seriesDefinitionWidget._manualDefWidget._newSeriesWidget
+            )
+            newSerieWidget.addTomoObjToCurrentSeries(scan)
+
+            assert len(self.window._widget._widget.getSelectedSeries()) == 1
+            self.window._widget._selectButton.released.emit()
+            self.qapp.processEvents()
+            assert self._listener.callCount() == 1

tomwer/tests/orangecontrib/tomwer/widgets/control/tests/test_volume_selector.py

@@ -0,0 +1,69 @@
+import gc
+import os
+import pickle
+import shutil
+import tempfile
+
+import numpy
+from orangecanvas.scheme.readwrite import literal_dumps
+from silx.gui import qt
+from silx.gui.utils.testutils import TestCaseQt
+
+from orangecontrib.tomwer.widgets.control.VolumeSelector import VolumeSelectorOW
+from tomwer.core.volume.hdf5volume import HDF5Volume
+
+
+class TestVolumeSelectorOW(TestCaseQt):
+    def setUp(self):
+        super().setUp()
+        self.window = VolumeSelectorOW()
+        self.folder = tempfile.mkdtemp()
+        self.volume_1 = HDF5Volume(
+            file_path=os.path.join(self.folder, "vol_file.hdf5"),
+            data_path="entry0000",
+            data=numpy.linspace(0, 10, 100 * 100 * 3).reshape(3, 100, 100),
+        )
+        self.volume_1.save()
+        self.volume_2 = HDF5Volume(
+            file_path=os.path.join(self.folder, "vol_file2.hdf5"),
+            data_path="entry0001",
+            data=numpy.linspace(60, 120, 100 * 100 * 5).reshape(5, 100, 100),
+        )
+        self.volume_2.save()
+
+    def tearDown(self):
+        self.window.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.window.close()
+        self.window = None
+        shutil.rmtree(self.folder)
+        gc.collect()
+
+    def test(self):
+        self.window.show()
+        # check adding one volume
+        self.window.addVolume(self.volume_1)
+        assert self.window.widget.dataList.n_data() == 1
+        # check behavior adding the same volume
+        self.window.addVolume(self.volume_1)
+        assert self.window.widget.dataList.n_data() == 1
+        # check adding another volume
+        self.window.addVolume(self.volume_2)
+        assert self.window.widget.dataList.n_data() == 2
+        # check removing volume(s)
+        self.window.removeVolume(self.volume_1)
+        assert self.window.widget.dataList.n_data() == 1
+        self.window.removeVolume(self.volume_1)
+        assert self.window.widget.dataList.n_data() == 1
+        self.window.removeVolume(self.volume_2)
+        assert self.window.widget.dataList.n_data() == 0
+
+        # and adding them back
+        self.window.addVolume(self.volume_1)
+        self.window.addVolume(self.volume_2)
+        assert self.window.widget.dataList.n_data() == 2
+
+        # test serialization
+        self.window._updateSettings()
+        assert len(self.window._scanIDs) == 2
+        pickle.dumps(self.window._scanIDs)
+        literal_dumps(self.window._scanIDs)

orangecontrib/tomwer/widgets/edit/test/test_image_key_editor.py → tomwer/tests/orangecontrib/tomwer/widgets/debugtools/tests/test_dataset_generator.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 # /*##########################################################################
 #
-# Copyright (c) 2017-2019 European Synchrotron Radiation Facility
+# Copyright (c) 2017-2021 European Synchrotron Radiation Facility
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -22,35 +22,36 @@
 # THE SOFTWARE.
 #
 # ###########################################################################*/
+
 __authors__ = ["H. Payno"]
 __license__ = "MIT"
-__date__ = "10/12/2021"
+__date__ = "21/06/2021"
 
 
 import gc
+import shutil
+import tempfile
 
-import pytest
 from silx.gui import qt
 from silx.gui.utils.testutils import TestCaseQt
 
-from orangecontrib.tomwer.widgets.edit.ImageKeyEditorOW import ImageKeyEditorOW
-from tomwer.tests.utils import skip_gui_test
-
+from orangecontrib.tomwer.widgets.debugtools.DatasetGeneratorOW import (
+    DatasetGeneratorOW,
+)
 
-@pytest.mark.skipif(skip_gui_test(), reason="skip gui test")
-class TestImageKeyEditor(TestCaseQt):
-    """Test that the ImageKeyEditorOW widget work correctly (at least launched)"""
 
+class TestDatasetGeneratorOW(TestCaseQt):
     def setUp(self):
-        super().setUp()
-        self._window = ImageKeyEditorOW()
+        self.tmp_dir = tempfile.mkdtemp()
+        self.widget = DatasetGeneratorOW()
+        self.widget.generatorGUI.mainWidget.setRootDir(self.tmp_dir)
 
     def tearDown(self):
-        self._window.setAttribute(qt.Qt.WA_DeleteOnClose)
-        self._window.close()
-        self._window = None
+        shutil.rmtree(self.tmp_dir)
+        self.widget.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self.widget.close()
+        self.widget = None
         gc.collect()
 
     def test(self):
-        self._window.show()
-        self.qWaitForWindowExposed(self._window)
+        self.widget._generate().join()

tomwer/tests/orangecontrib/tomwer/widgets/debugtools/tests/test_object_inspector.py

@@ -0,0 +1,36 @@
+# coding: utf-8
+from __future__ import annotations
+
+
+import gc
+import os
+import shutil
+import tempfile
+
+from silx.gui import qt
+from silx.gui.utils.testutils import TestCaseQt
+
+from orangecontrib.tomwer.widgets.debugtools.ObjectInspectorOW import ObjectInspectorOW
+from tomwer.core.utils.scanutils import MockNXtomo
+
+
+class TestObjectInspectorOW(TestCaseQt):
+    def setUp(self):
+        super().setUp()
+        self.tmp_dir = tempfile.mkdtemp()
+        self.scan = MockNXtomo(
+            scan_path=os.path.join(self.tmp_dir, "myscan"),
+            n_proj=20,
+            n_ini_proj=20,
+            dim=10,
+        ).scan
+
+    def tearDown(self):
+        shutil.rmtree(self.tmp_dir)
+        gc.collect()
+
+    def test(self):
+        widget = ObjectInspectorOW()
+        widget.setObject(self.scan)
+        widget.setAttribute(qt.Qt.WA_DeleteOnClose)
+        widget.close()

{orangecontrib/tomwer/widgets/edit/test → tomwer/tests/orangecontrib/tomwer/widgets/edit/tests}/test_dark_flat_patch.py

@@ -1,31 +1,5 @@
 # coding: utf-8
-# /*##########################################################################
-#
-# Copyright (c) 2017-2019 European Synchrotron Radiation Facility
-#
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-#
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-#
-# ###########################################################################*/
-__authors__ = ["H. Payno"]
-__license__ = "MIT"
-__date__ = "10/12/2021"
-
+from __future__ import annotations
 
 import gc
 import os

tomwer/tests/orangecontrib/tomwer/widgets/edit/tests/test_image_key_editor.py

@@ -0,0 +1,30 @@
+# coding: utf-8
+from __future__ import annotations
+
+import gc
+
+import pytest
+from silx.gui import qt
+from silx.gui.utils.testutils import TestCaseQt
+
+from orangecontrib.tomwer.widgets.edit.ImageKeyEditorOW import ImageKeyEditorOW
+from tomwer.tests.utils import skip_gui_test
+
+
+@pytest.mark.skipif(skip_gui_test(), reason="skip gui test")
+class TestImageKeyEditor(TestCaseQt):
+    """Test that the ImageKeyEditorOW widget work correctly (at least launched)"""
+
+    def setUp(self):
+        super().setUp()
+        self._window = ImageKeyEditorOW()
+
+    def tearDown(self):
+        self._window.setAttribute(qt.Qt.WA_DeleteOnClose)
+        self._window.close()
+        self._window = None
+        gc.collect()
+
+    def test(self):
+        self._window.show()
+        self.qWaitForWindowExposed(self._window)