def test_display_mode_event(self):
listener = SignalListener()
widget = self.create_widget()
widget.displayedViewChanged.connect(listener)
widget.setData(10)
widget.setData(None)
modes = [v.modeId() for v in listener.arguments(argumentIndex=0)]
self.assertEqual(modes, [DataViews.RAW_MODE, DataViews.EMPTY_MODE])
listener.clear()
def test_display_mode_event(self):
listener = SignalListener()
widget = self.create_widget()
widget.displayedViewChanged.connect(listener)
widget.setData(10)
widget.setData(None)
modes = [v.modeId() for v in listener.arguments(argumentIndex=0)]
self.assertEqual(modes, [DataViews.RAW_MODE, DataViews.EMPTY_MODE])
listener.clear()
def test_selected_data_event(self):
data = numpy.arange(3 * 3 * 3)
data.shape = 3, 3, 3
widget = NumpyAxesSelector()
listener = SignalListener()
widget.selectionChanged.connect(listener)
widget.setData(data)
widget.setAxisNames(["x"])
widget.setData(None)
self.assertEqual(listener.callCount(), 3)
listener.clear()
def test_selected_data_event(self):
data = numpy.arange(3 * 3 * 3)
data.shape = 3, 3, 3
widget = NumpyAxesSelector()
listener = SignalListener()
widget.selectionChanged.connect(listener)
widget.setData(data)
widget.setAxisNames(["x"])
widget.setData(None)
self.assertEqual(listener.callCount(), 3)
listener.clear()
def testFail(self):
listener = SignalListener()
button = ThreadPoolPushButton()
button.setCallable(self._computeFail)
button.beforeExecuting.connect(listener.partial(test="be"))
button.started.connect(listener.partial(test="s"))
button.succeeded.connect(listener.partial(test="Unexpected success"))
button.failed.connect(listener.partial(test="exception"))
button.finished.connect(listener.partial(test="f"))
with TestLogging('silx.gui.widgets.ThreadPoolPushButton', error=1):
button.executeCallable()
self.qapp.processEvents()
time.sleep(0.1)
self.qapp.processEvents()
result = listener.karguments(argumentName="test")
self.assertListEqual(result, ["be", "s", "exception", "f"])
listener.clear()
def testFail(self):
listener = SignalListener()
button = ThreadPoolPushButton()
button.setCallable(self._computeFail)
button.beforeExecuting.connect(listener.partial(test="be"))
button.started.connect(listener.partial(test="s"))
button.succeeded.connect(listener.partial(test="Unexpected success"))
button.failed.connect(listener.partial(test="exception"))
button.finished.connect(listener.partial(test="f"))
with TestLogging('silx.gui.widgets.ThreadPoolPushButton', error=1):
button.executeCallable()
self.qapp.processEvents()
time.sleep(0.1)
self.qapp.processEvents()
result = listener.karguments(argumentName="test")
self.assertListEqual(result, ["be", "s", "exception", "f"])
listener.clear()
def test(self):
"""Test ROI of different shapes"""
tests = ( # shape, points=[list of (x, y), list of (x, y)]
(roi_items.PointROI, numpy.array(([(10., 15.)], [(20., 25.)]))),
(roi_items.RectangleROI,
numpy.array((((1., 10.), (11., 20.)), ((2., 3.), (12., 13.))))),
(roi_items.PolygonROI,
numpy.array((((0., 1.), (0., 10.), (10., 0.)),
((5., 6.), (5., 16.), (15., 6.))))),
(roi_items.LineROI,
numpy.array(
(((10., 20.), (10., 30.)), ((30., 40.), (30., 50.))))),
(roi_items.HorizontalLineROI,
numpy.array(
(((10., 20.), (10., 30.)), ((30., 40.), (30., 50.))))),
(roi_items.VerticalLineROI,
numpy.array(
(((10., 20.), (10., 30.)), ((30., 40.), (30., 50.))))),
(roi_items.HorizontalLineROI,
numpy.array(
(((10., 20.), (10., 30.)), ((30., 40.), (30., 50.))))),
)
for roiClass, points in tests:
with self.subTest(roiClass=roiClass):
manager = roi.RegionOfInterestManager(self.plot)
self.roiTableWidget.setRegionOfInterestManager(manager)
manager.start(roiClass)
self.assertEqual(manager.getRois(), ())
finishListener = SignalListener()
manager.sigInteractiveModeFinished.connect(finishListener)
changedListener = SignalListener()
manager.sigRoiChanged.connect(changedListener)
# Add a point
r = roiClass()
r.setFirstShapePoints(points[0])
manager.addRoi(r)
self.qapp.processEvents()
self.assertTrue(len(manager.getRois()), 1)
self.assertEqual(changedListener.callCount(), 1)
# Remove it
manager.removeRoi(manager.getRois()[0])
self.assertEqual(manager.getRois(), ())
self.assertEqual(changedListener.callCount(), 2)
# Add two point
r = roiClass()
r.setFirstShapePoints(points[0])
manager.addRoi(r)
self.qapp.processEvents()
r = roiClass()
r.setFirstShapePoints(points[1])
manager.addRoi(r)
self.qapp.processEvents()
self.assertTrue(len(manager.getRois()), 2)
self.assertEqual(changedListener.callCount(), 4)
# Reset it
result = manager.clear()
self.assertTrue(result)
self.assertEqual(manager.getRois(), ())
self.assertEqual(changedListener.callCount(), 5)
changedListener.clear()
# Add two point
r = roiClass()
r.setFirstShapePoints(points[0])
manager.addRoi(r)
self.qapp.processEvents()
r = roiClass()
r.setFirstShapePoints(points[1])
manager.addRoi(r)
self.qapp.processEvents()
self.assertTrue(len(manager.getRois()), 2)
self.assertEqual(changedListener.callCount(), 2)
# stop
result = manager.stop()
self.assertTrue(result)
self.assertTrue(len(manager.getRois()), 1)
self.qapp.processEvents()
self.assertEqual(finishListener.callCount(), 1)
manager.clear()
class TestRoiWidgetSignals(TestCaseQt):
"""Test Signals emitted by the RoiWidgetSignals"""
def setUp(self):
self.plot = Plot1D()
x = range(20)
y = range(20)
self.plot.addCurve(x, y, legend='curve0')
self.listener = SignalListener()
self.curves_roi_widget = self.plot.getCurvesRoiWidget()
self.curves_roi_widget.sigROISignal.connect(self.listener)
assert self.curves_roi_widget.isVisible() is False
assert self.listener.callCount() == 0
self.plot.show()
self.qWaitForWindowExposed(self.plot)
toolButton = getQToolButtonFromAction(self.plot.getRoiAction())
self.mouseClick(widget=toolButton, button=qt.Qt.LeftButton)
self.curves_roi_widget.show()
self.qWaitForWindowExposed(self.curves_roi_widget)
def tearDown(self):
self.plot = None
def testSigROISignalAddRmRois(self):
"""Test SigROISignal when adding and removing ROIS"""
print(self.listener.callCount())
self.assertEqual(self.listener.callCount(), 1)
self.listener.clear()
roi1 = CurvesROIWidget.ROI(name='linear', fromdata=0, todata=5)
self.curves_roi_widget.roiTable.addRoi(roi1)
self.assertEqual(self.listener.callCount(), 1)
self.assertTrue(self.listener.arguments()[0][0]['current'] == 'linear')
self.listener.clear()
roi2 = CurvesROIWidget.ROI(name='linear2', fromdata=0, todata=5)
self.curves_roi_widget.roiTable.addRoi(roi2)
self.assertEqual(self.listener.callCount(), 1)
self.assertTrue(
self.listener.arguments()[0][0]['current'] == 'linear2')
self.listener.clear()
self.curves_roi_widget.roiTable.removeROI(roi2)
self.assertEqual(self.listener.callCount(), 1)
self.assertTrue(self.curves_roi_widget.roiTable.activeRoi == roi1)
self.assertTrue(self.listener.arguments()[0][0]['current'] == 'linear')
self.listener.clear()
self.curves_roi_widget.roiTable.deleteActiveRoi()
self.assertEqual(self.listener.callCount(), 1)
self.assertTrue(self.curves_roi_widget.roiTable.activeRoi is None)
self.assertTrue(self.listener.arguments()[0][0]['current'] is None)
self.listener.clear()
self.curves_roi_widget.roiTable.addRoi(roi1)
self.assertEqual(self.listener.callCount(), 1)
self.assertTrue(self.listener.arguments()[0][0]['current'] == 'linear')
self.assertTrue(self.curves_roi_widget.roiTable.activeRoi == roi1)
self.listener.clear()
self.qapp.processEvents()
self.curves_roi_widget.roiTable.removeROI(roi1)
self.qapp.processEvents()
self.assertEqual(self.listener.callCount(), 1)
self.assertTrue(self.listener.arguments()[0][0]['current'] == 'ICR')
self.listener.clear()
def testSigROISignalModifyROI(self):
"""Test SigROISignal when modifying it"""
self.curves_roi_widget.roiTable.setMiddleROIMarkerFlag(True)
roi1 = CurvesROIWidget.ROI(name='linear', fromdata=2, todata=5)
self.curves_roi_widget.roiTable.addRoi(roi1)
self.curves_roi_widget.roiTable.setActiveRoi(roi1)
# test modify the roi2 object
self.listener.clear()
roi1.setFrom(0.56)
self.assertEqual(self.listener.callCount(), 1)
self.listener.clear()
roi1.setTo(2.56)
self.assertEqual(self.listener.callCount(), 1)
self.listener.clear()
roi1.setName('linear2')
self.assertEqual(self.listener.callCount(), 1)
self.listener.clear()
roi1.setType('new type')
self.assertEqual(self.listener.callCount(), 1)
# modify roi limits (from the gui)
roi_marker_handler = self.curves_roi_widget.roiTable._markersHandler.getMarkerHandler(
roi1.getID())
for marker_type in ('min', 'max', 'middle'):
with self.subTest(marker_type=marker_type):
self.listener.clear()
marker = roi_marker_handler.getMarker(marker_type)
self.qapp.processEvents()
items_interaction = ItemsInteraction(plot=self.plot)
x_pix, y_pix = self.plot.dataToPixel(marker.getXPosition(), 1)
items_interaction.beginDrag(x_pix, y_pix)
self.qapp.processEvents()
items_interaction.endDrag(x_pix + 10, y_pix)
self.qapp.processEvents()
self.assertEqual(self.listener.callCount(), 1)
def testSetActiveCurve(self):
"""Test sigRoiSignal when set an active curve"""
roi1 = CurvesROIWidget.ROI(name='linear', fromdata=2, todata=5)
self.curves_roi_widget.roiTable.addRoi(roi1)
self.curves_roi_widget.roiTable.setActiveRoi(roi1)
self.listener.clear()
self.plot.setActiveCurve('curve0')
self.assertEqual(self.listener.callCount(), 0)
def test(self):
"""Test ROI of different shapes"""
tests = ( # shape, points=[list of (x, y), list of (x, y)]
(roi_items.PointROI, numpy.array(([(10., 15.)], [(20., 25.)]))),
(roi_items.RectangleROI,
numpy.array((((1., 10.), (11., 20.)),
((2., 3.), (12., 13.))))),
(roi_items.PolygonROI,
numpy.array((((0., 1.), (0., 10.), (10., 0.)),
((5., 6.), (5., 16.), (15., 6.))))),
(roi_items.LineROI,
numpy.array((((10., 20.), (10., 30.)),
((30., 40.), (30., 50.))))),
(roi_items.HorizontalLineROI,
numpy.array((((10., 20.), (10., 30.)),
((30., 40.), (30., 50.))))),
(roi_items.VerticalLineROI,
numpy.array((((10., 20.), (10., 30.)),
((30., 40.), (30., 50.))))),
)
for roiClass, points in tests:
with self.subTest(roiClass=roiClass):
manager = roi.RegionOfInterestManager(self.plot)
self.roiTableWidget.setRegionOfInterestManager(manager)
manager.start(roiClass)
self.assertEqual(manager.getRois(), ())
finishListener = SignalListener()
manager.sigInteractiveModeFinished.connect(finishListener)
changedListener = SignalListener()
manager.sigRoiChanged.connect(changedListener)
# Add a point
manager.createRoi(roiClass, points[0])
self.qapp.processEvents()
self.assertTrue(len(manager.getRois()), 1)
self.assertEqual(changedListener.callCount(), 1)
# Remove it
manager.removeRoi(manager.getRois()[0])
self.assertEqual(manager.getRois(), ())
self.assertEqual(changedListener.callCount(), 2)
# Add two point
manager.createRoi(roiClass, points[0])
self.qapp.processEvents()
manager.createRoi(roiClass, points[1])
self.qapp.processEvents()
self.assertTrue(len(manager.getRois()), 2)
self.assertEqual(changedListener.callCount(), 4)
# Reset it
result = manager.clear()
self.assertTrue(result)
self.assertEqual(manager.getRois(), ())
self.assertEqual(changedListener.callCount(), 5)
changedListener.clear()
# Add two point
manager.createRoi(roiClass, points[0])
self.qapp.processEvents()
manager.createRoi(roiClass, points[1])
self.qapp.processEvents()
self.assertTrue(len(manager.getRois()), 2)
self.assertEqual(changedListener.callCount(), 2)
# stop
result = manager.stop()
self.assertTrue(result)
self.assertTrue(len(manager.getRois()), 1)
self.qapp.processEvents()
self.assertEqual(finishListener.callCount(), 1)
manager.clear()