def setUp(self):
super(TestProfile3DToolBar, self).setUp()
self.plot = StackView()
self.plot.show()
self.qWaitForWindowExposed(self.plot)
self.plot.setStack(
numpy.array([[[0, 1, 2], [3, 4, 5]], [[6, 7, 8], [9, 10, 11]],
[[12, 13, 14], [15, 16, 17]]]))
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.hist = Plot1D(self)
self.stack = StackView(self)
self.stack.setColormap({'name': 'viridis', 'autoscale': True})
self.setLayout(qt.QHBoxLayout())
splitter = qt.QSplitter()
splitter.addWidget(self.hist)
splitter.addWidget(self.stack)
self.layout().addWidget(splitter)
def testProfile2D(self):
"""Test that the profile plot associated to a stack view is either a
Plot1D or a plot 2D instance."""
plot = StackView()
plot.show()
self.qWaitForWindowExposed(plot)
plot.setStack(numpy.array([[[0, 1], [2, 3]], [[4, 5], [6, 7]]]))
self.assertIsInstance(plot.getProfileToolbar().getProfileMainWindow(),
qt.QMainWindow)
self.assertIsInstance(plot.getProfileToolbar().getProfilePlot(),
Plot2D)
plot.getProfileToolbar().profile3dAction.computeProfileIn1D()
self.assertIsInstance(plot.getProfileToolbar().getProfilePlot(),
Plot1D)
plot.setAttribute(qt.Qt.WA_DeleteOnClose)
plot.close()
del plot
def testProfile2D(self):
"""Test that the profile plot associated to a stack view is either a
Plot1D or a plot 2D instance."""
self.plot = StackView()
self.plot.show()
self.qWaitForWindowExposed(self.plot)
self.plot.setStack(numpy.array([[[0, 1], [2, 3]], [[4, 5], [6, 7]]]))
toolBar = self.plot.getProfileToolbar()
manager = toolBar.getProfileManager()
roiManager = manager.getRoiManager()
roi = rois.ProfileImageStackHorizontalLineROI()
roi.setPosition(0.5)
roi.setProfileType("2D")
roiManager.addRoi(roi)
roiManager.setCurrentRoi(roi)
for _ in range(20):
self.qWait(200)
if not manager.hasPendingOperations():
break
profileWindow = roi.getProfileWindow()
self.assertIsInstance(roi.getProfileWindow(), qt.QMainWindow)
self.assertIsInstance(profileWindow.getCurrentPlotWidget(), Plot2D)
roi.setProfileType("1D")
for _ in range(20):
self.qWait(200)
if not manager.hasPendingOperations():
break
profileWindow = roi.getProfileWindow()
self.assertIsInstance(roi.getProfileWindow(), qt.QMainWindow)
self.assertIsInstance(profileWindow.getCurrentPlotWidget(), Plot1D)