def __init__(self, parent=None):
super(plotTestGui, self).__init__(parent)
self._context = PlottingContext()
self.test = MultiPlotWidget(self._context, self)
ws = setUpSubplot()
self.test.add_subplot("test")
self.test.add_subplot("bob")
self.test.add_subplot("moo")
self.test.add_subplot("baa")
self.test.add_subplot("EXTRA")
self.test.plot("test", ws, specNum=26)
self.test.plot("test", ws, specNum=21)
self.test.plot("test", ws, specNum=22)
# defines position of label
dummy = Label("dummy", 10.1, False, 0.9, True, rotation=-90)
dummy2 = Label(
"protected",
5.1,
False,
0.9,
True,
rotation=-90,
protected=True)
dummy3 = Label("just annotate", 14.1, False, 0.9, True)
# defines position of line
# need to add methods to add just a label
# need to add_vline with a name and if protected but no
# annotation
self.test.add_vline_and_annotate("test", 10, dummy)
self.test.add_vline_and_annotate("test", 5, dummy2)
self.test.add_annotate("bob", dummy3)
self.test.add_vline("bob", 1.2, "just a line")
self.test.plot("bob", ws, specNum=1)
self.test.plot("EXTRA", ws, specNum=42)
self.test.plot("moo", ws, specNum=42)
self.test.plot("baa", ws, specNum=2)
self.test.set_all_values()
self.test.connectCloseSignal(self.close)
# add button for adding more plots
self.n = 0
self.ws = ws
self.btn = QtGui.QPushButton("add plot")
self.btn.clicked.connect(self.add)
self.grid = QtGui.QSplitter(QtCore.Qt.Vertical)
self.grid.addWidget(self.test)
self.grid.addWidget(self.btn)
self.setCentralWidget(self.grid)
self.setWindowTitle("plot test")
def test_that_rm_line_returns_if_plot_window_is_none(self):
self.gui.plotting = MultiPlotWidget(mock.Mock())
self.gui.plotting.get_subplots = mock.Mock(return_value=['plot1', 'plot2', 'plot3'])
self.gui.plot_window = None
self.gui._rm_line('line')
self.assertEqual(self.gui.plotting.get_subplots.call_count, 0)
def test_that_rm_line_calls_correct_function_if_window_not_none(self):
self.gui.plotting = MultiPlotWidget(mock.Mock())
self.gui.plotting.get_subplots = mock.Mock(return_value=['plot1', 'plot2', 'plot3'])
self.gui.plotting.rm_vline_and_annotate = mock.Mock()
self.gui.plot_window = mock.create_autospec(MultiPlotWindow)
self.gui._rm_line('line')
self.assertEqual(self.gui.plotting.get_subplots.call_count, 1)
self.assertEqual(self.gui.plotting.rm_vline_and_annotate.call_count, 3)
self.gui.plotting.rm_vline_and_annotate.assert_called_with('plot3', 'line')
def test_that_plot_line_calls_plot_line_once_if_window_not_none(self, mock_plot_line_once,
mock_gen_label):
self.gui.plot_window = mock.create_autospec(MultiPlotWindow)
self.gui.plotting = MultiPlotWidget(mock.Mock())
self.gui.plotting.get_subplots = mock.Mock(return_value=['plot1'])
mock_gen_label.return_value = 'name of the label'
self.gui._plot_line('name', 1.0, 'C0', None)
self.assertEqual(mock_plot_line_once.call_count, 1)
mock_plot_line_once.assert_called_with('plot1', 1.0, 'name of the label', 'C0')
def __init__(self, parent=None):
super(plotTestGui, self).__init__(parent)
self._context = PlottingContext()
self.test = MultiPlotWidget(self._context, self)
ws = setUpSubplot()
self.test.add_subplot("test", 221)
self.test.add_subplot("bob", 222)
self.test.add_subplot("moo", 223)
self.test.add_subplot("baa", 224)
self.test.plot("test", ws, specNum=26)
self.test.plot("test", ws, specNum=21)
self.test.plot("test", ws, specNum=22)
# defines position of label
dummy = Label("dummy", 10.1, False, 0.9, True, rotation=-90)
dummy2 = Label("protected",
5.1,
False,
0.9,
True,
rotation=-90,
protected=True)
dummy3 = Label("just annotate", 14.1, False, 0.9, True)
# defines position of line
# need to add methods to add just a label
# need to add_vline with a name and if protected but no
# annotation
self.test.add_vline_and_annotate("test", 10, dummy)
self.test.add_vline_and_annotate("test", 5, dummy2)
self.test.add_annotate("bob", dummy3)
self.test.add_vline("bob", 1.2, "just a line")
self.test.plot("bob", ws, specNum=1)
self.test.plot("moo", ws, specNum=42)
self.test.plot("baa", ws, specNum=2)
self.test.set_all_values()
self.setCentralWidget(self.test)
self.setWindowTitle("plot test")
def setUp(self):
context = PlottingContext()
self.widget = MultiPlotWidget(context)
class MultiPlotWidgetTest(unittest.TestCase):
def setUp(self):
context = PlottingContext()
self.widget = MultiPlotWidget(context)
def test_add_subplot(self):
with mock.patch("MultiPlotting.QuickEdit.quickEdit_widget.QuickEditWidget.add_subplot") as qe_patch:
self.widget.add_subplot("test")
self.assertEqual(qe_patch.call_count,1)
def test_plot(self):
with mock.patch("MultiPlotting.subplot.subplot.subplot.plot") as patch:
ws = mock.MagicMock()
subplotName = "test"
specNum = 4
self.widget.plot(subplotName, ws, specNum)
patch.assert_called_with(subplotName, ws, specNum=specNum)
self.assertEqual(patch.call_count,1)
def test_setAllValues(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value = list(data().keys()))
self.widget._x_range_changed = mock.MagicMock()
self.widget._y_range_changed = mock.MagicMock()
self.widget._check_all_errors = mock.MagicMock(return_value = False)
self.widget._change_errors = mock.MagicMock()
self.widget.set_all_values()
self.widget._x_range_changed.assert_called_with([-1,20])
self.widget._y_range_changed.assert_called_with([0,50])
def test_updateQuickEditNoMatch(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value = data())
self.widget.quickEdit.rm_subplot = mock.Mock()
self.widget.quickEdit._if_empty_close = mock.Mock()
self.widget._update_quick_edit("no match")
self.assertEqual(self.widget.quickEdit.rm_subplot.call_count, 1)
def test_updateQuickEdit1Match(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("two")
self.widget.quickEdit.set_plot_x_range.assert_called_with([6,10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0,9])
def test_updateQuickEdit1NoMatch(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("three")
self.assertEqual(self.widget.quickEdit.set_plot_x_range.call_count,0)
self.assertEqual(self.widget.quickEdit.set_plot_y_range.call_count,0)
def test_updateQuickEditMany(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two","three"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("two")
self.widget.quickEdit.set_plot_x_range.assert_called_with([6,10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0,9])
def test_selectionChanged(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two"])
self.widget._check_all_errors = mock.MagicMock(return_value = False)
self.widget._change_errors = mock.MagicMock()
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._selection_changed(1)
self.widget.quickEdit.set_plot_x_range.assert_called_with([6,10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0,9])
def test_selectionChangedAll(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two","three"])
xbounds = [-1,2]
ybounds = [-10,20]
self.widget._context.get_xBounds = mock.MagicMock(return_value = xbounds)
self.widget._context.get_yBounds = mock.MagicMock(return_value = ybounds)
self.widget._check_all_errors = mock.MagicMock(return_value = False)
self.widget._change_errors = mock.MagicMock()
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._x_range_changed = mock.MagicMock()
self.widget._y_range_changed = mock.MagicMock()
self.widget._selection_changed(1)
self.widget.quickEdit.set_plot_x_range.assert_called_with(xbounds)
self.widget.quickEdit.set_plot_y_range.assert_called_with(ybounds)
self.widget._x_range_changed.assert_called_with(xbounds)
self.widget._y_range_changed.assert_called_with(ybounds)
def test_xRangeChanged(self):
names = ["two","three"]
xbounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_xBounds = mock.MagicMock()
self.widget.plots.set_plot_x_range = mock.MagicMock()
self.widget._x_range_changed(xbounds)
self.widget._context.set_xBounds.assert_called_with(xbounds)
self.widget.plots.set_plot_x_range.assert_called_with(names,xbounds)
def test_xRangeChanged1(self):
names = ["two"]
xbounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_xBounds = mock.MagicMock()
self.widget.plots.set_plot_x_range = mock.MagicMock()
self.widget._x_range_changed(xbounds)
self.assertEqual(self.widget._context.set_xBounds.call_count, 0)
self.widget.plots.set_plot_x_range.assert_called_with(names,xbounds)
def test_yRangeChanged(self):
names = ["two","three"]
ybounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_yBounds = mock.MagicMock()
self.widget.plots.set_plot_y_range = mock.MagicMock()
self.widget._y_range_changed(ybounds)
self.widget._context.set_yBounds.assert_called_with(ybounds)
self.widget.plots.set_plot_y_range.assert_called_with(names,ybounds)
def test_yRangeChanged1(self):
names = ["two"]
ybounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_yBounds = mock.MagicMock()
self.widget.plots.set_plot_y_range = mock.MagicMock()
self.widget._y_range_changed(ybounds)
self.assertEqual(self.widget._context.set_yBounds.call_count, 0)
self.widget.plots.set_plot_y_range.assert_called_with(names,ybounds)
def test_checkAllErrorsFalse(self):
context = data()
self.widget._context.subplots = context
self.assertEqual(self.widget._check_all_errors(context.keys()),False)
def test_checkAllErrorsTrue(self):
context = data()
for name in context.keys():
context[name].error = True
self.widget._context.subplots = context
self.assertEqual(self.widget._check_all_errors(context.keys()),True)
def test_checkAllErrors1True(self):
context = data()
context["two"].error = True
self.widget._context.subplots = context
self.assertEqual(self.widget._check_all_errors(context.keys()),False)
class MultiPlotWidgetTest(unittest.TestCase):
def setUp(self):
context = PlottingContext()
self.widget = MultiPlotWidget(context)
def test_add_subplot(self):
with mock.patch(
"MultiPlotting.QuickEdit.quickEdit_widget.QuickEditWidget.add_subplot"
) as qe_patch:
self.widget.add_subplot("test")
self.assertEqual(qe_patch.call_count, 1)
def test_plot(self):
with mock.patch("MultiPlotting.subplot.subplot.subplot.plot") as patch:
ws = mock.MagicMock()
subplot_name = "test"
spec_num = 4
self.widget.plot(subplot_name, ws, color='C0', spec_num=spec_num)
patch.assert_called_with(subplot_name,
ws,
color='C0',
spec_num=spec_num)
self.assertEqual(patch.call_count, 1)
def test_set_all_values(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=list(data().keys()))
self.widget._x_range_changed = mock.MagicMock()
self.widget._y_range_changed = mock.MagicMock()
self.widget._check_all_errors = mock.MagicMock(return_value=False)
self.widget._change_errors = mock.MagicMock()
self.widget.set_all_values()
self.widget._x_range_changed.assert_called_with([-1, 20])
self.widget._y_range_changed.assert_called_with([0, 50])
def test_update_quick_edit_no_match(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=data())
self.widget.quickEdit.rm_subplot = mock.Mock()
self.widget.quickEdit._if_empty_close = mock.Mock()
self.widget._update_quick_edit("no match")
self.assertEqual(self.widget.quickEdit.rm_subplot.call_count, 1)
def test_update_quick_edit1_match(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=["two"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("two")
self.widget.quickEdit.set_plot_x_range.assert_called_with([6, 10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0, 9])
def test_update_quick_edit1_no_match(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=["two"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("three")
self.assertEqual(self.widget.quickEdit.set_plot_x_range.call_count, 0)
self.assertEqual(self.widget.quickEdit.set_plot_y_range.call_count, 0)
def test_update_quick_edit_many(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=["two", "three"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("two")
self.widget.quickEdit.set_plot_x_range.assert_called_with([6, 10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0, 9])
def test_selection_changed(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=["two"])
self.widget._check_all_errors = mock.MagicMock(return_value=False)
self.widget._change_errors = mock.MagicMock()
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._selection_changed(1)
self.widget.quickEdit.set_plot_x_range.assert_called_with([6, 10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0, 9])
def test_selection_changed_all(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=["two", "three"])
xbounds = [-1, 2]
ybounds = [-10, 20]
self.widget._context.get_xBounds = mock.MagicMock(return_value=xbounds)
self.widget._context.get_yBounds = mock.MagicMock(return_value=ybounds)
self.widget._check_all_errors = mock.MagicMock(return_value=False)
self.widget._change_errors = mock.MagicMock()
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._x_range_changed = mock.MagicMock()
self.widget._y_range_changed = mock.MagicMock()
self.widget._selection_changed(1)
self.widget.quickEdit.set_plot_x_range.assert_called_with(xbounds)
self.widget.quickEdit.set_plot_y_range.assert_called_with(ybounds)
self.widget._x_range_changed.assert_called_with(xbounds)
self.widget._y_range_changed.assert_called_with(ybounds)
def test_x_range_changed(self):
names = ["two", "three"]
xbounds = [9, 18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=names)
self.widget._context.set_xBounds = mock.MagicMock()
self.widget.plots.set_plot_x_range = mock.MagicMock()
self.widget._x_range_changed(xbounds)
self.widget._context.set_xBounds.assert_called_with(xbounds)
self.widget.plots.set_plot_x_range.assert_called_with(names, xbounds)
def test_x_range_changed1(self):
names = ["two"]
xbounds = [9, 18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=names)
self.widget._context.set_xBounds = mock.MagicMock()
self.widget.plots.set_plot_x_range = mock.MagicMock()
self.widget._x_range_changed(xbounds)
self.assertEqual(self.widget._context.set_xBounds.call_count, 0)
self.widget.plots.set_plot_x_range.assert_called_with(names, xbounds)
def test_y_range_changed(self):
names = ["two", "three"]
ybounds = [9, 18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=names)
self.widget._context.set_yBounds = mock.MagicMock()
self.widget.plots.set_plot_y_range = mock.MagicMock()
self.widget._y_range_changed(ybounds)
self.widget._context.set_yBounds.assert_called_with(ybounds)
self.widget.plots.set_plot_y_range.assert_called_with(names, ybounds)
def test_y_range_changed1(self):
names = ["two"]
ybounds = [9, 18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(
return_value=names)
self.widget._context.set_yBounds = mock.MagicMock()
self.widget.plots.set_plot_y_range = mock.MagicMock()
self.widget._y_range_changed(ybounds)
self.assertEqual(self.widget._context.set_yBounds.call_count, 0)
self.widget.plots.set_plot_y_range.assert_called_with(names, ybounds)
def test_check_all_errors_false(self):
context = data()
self.widget._context.subplots = context
self.assertEqual(self.widget._check_all_errors(context.keys()), False)
def test_check_all_errors_true(self):
context = data()
for name in context.keys():
context[name].error = True
self.widget._context.subplots = context
self.assertEqual(self.widget._check_all_errors(context.keys()), True)
def test_check_all_errors1_true(self):
context = data()
context["two"].error = True
self.widget._context.subplots = context
self.assertEqual(self.widget._check_all_errors(context.keys()), False)
def test_that_remove_line_connection_connects_to_correct_signal(self):
self.widget.plots = mock.Mock()
self.widget.remove_line_connection("slot parameter")
self.widget.plots.connect_rm_line_signal.assert_called_with(
"slot parameter")
def test_that_remove_line_calls_correct_function(self):
self.widget.plots = mock.Mock()
self.widget.remove_line("subplot", 'ws_name')
self.widget.plots.remove_lines.assert_called_with(
"subplot", ["ws_name: spec 1"])
def setUp(self):
self._qapp = mock_widget.mockQapp()
context = PlottingContext()
self.widget = MultiPlotWidget(context)
def setUp(self):
self._qapp = mock_widget.mockQapp()
context = PlottingContext()
self.widget = MultiPlotWidget(context)
class MultiPlotWidgetTest(unittest.TestCase):
def setUp(self):
self._qapp = mock_widget.mockQapp()
context = PlottingContext()
self.widget = MultiPlotWidget(context)
def test_add_subplot(self):
with mock.patch("MultiPlotting.QuickEdit.quickEdit_widget.QuickEditWidget.add_subplot") as qe_patch:
self.widget.add_subplot("test")
self.assertEquals(qe_patch.call_count,1)
def test_plot(self):
with mock.patch("MultiPlotting.subplot.subplot.subplot.plot") as patch:
ws = mock.MagicMock()
subplotName = "test"
specNum = 4
self.widget.plot(subplotName, ws, specNum)
patch.assert_called_with(subplotName, ws, specNum=specNum)
self.assertEquals(patch.call_count,1)
def test_setAllValues(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value = list(data().keys()))
self.widget._x_range_changed = mock.MagicMock()
self.widget._y_range_changed = mock.MagicMock()
self.widget._check_all_errors = mock.MagicMock(return_value = False)
self.widget._change_errors = mock.MagicMock()
self.widget.set_all_values()
self.widget._x_range_changed.assert_called_with([-1,20])
self.widget._y_range_changed.assert_called_with([0,50])
def test_updateQuickEditNoMatch(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value = data())
self.widget.quickEdit.rm_subplot = mock.Mock()
self.widget.quickEdit._if_empty_close = mock.Mock()
self.widget._update_quick_edit("no match")
self.assertEquals(self.widget.quickEdit.rm_subplot.call_count, 1)
def test_updateQuickEdit1Match(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("two")
self.widget.quickEdit.set_plot_x_range.assert_called_with([6,10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0,9])
def test_updateQuickEdit1NoMatch(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("three")
self.assertEquals(self.widget.quickEdit.set_plot_x_range.call_count,0)
self.assertEquals(self.widget.quickEdit.set_plot_y_range.call_count,0)
def test_updateQuickEditMany(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two","three"])
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._update_quick_edit("two")
self.widget.quickEdit.set_plot_x_range.assert_called_with([6,10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0,9])
def test_selectionChanged(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two"])
self.widget._check_all_errors = mock.MagicMock(return_value = False)
self.widget._change_errors = mock.MagicMock()
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._selection_changed(1)
self.widget.quickEdit.set_plot_x_range.assert_called_with([6,10])
self.widget.quickEdit.set_plot_y_range.assert_called_with([0,9])
def test_selectionChangedAll(self):
self.widget._context.subplots = data()
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =["two","three"])
xbounds = [-1,2]
ybounds = [-10,20]
self.widget._context.get_xBounds = mock.MagicMock(return_value = xbounds)
self.widget._context.get_yBounds = mock.MagicMock(return_value = ybounds)
self.widget._check_all_errors = mock.MagicMock(return_value = False)
self.widget._change_errors = mock.MagicMock()
self.widget.quickEdit.set_plot_x_range = mock.MagicMock()
self.widget.quickEdit.set_plot_y_range = mock.MagicMock()
self.widget._x_range_changed = mock.MagicMock()
self.widget._y_range_changed = mock.MagicMock()
self.widget._selection_changed(1)
self.widget.quickEdit.set_plot_x_range.assert_called_with(xbounds)
self.widget.quickEdit.set_plot_y_range.assert_called_with(ybounds)
self.widget._x_range_changed.assert_called_with(xbounds)
self.widget._y_range_changed.assert_called_with(ybounds)
def test_xRangeChanged(self):
names = ["two","three"]
xbounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_xBounds = mock.MagicMock()
self.widget.plots.set_plot_x_range = mock.MagicMock()
self.widget._x_range_changed(xbounds)
self.widget._context.set_xBounds.assert_called_with(xbounds)
self.widget.plots.set_plot_x_range.assert_called_with(names,xbounds)
def test_xRangeChanged1(self):
names = ["two"]
xbounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_xBounds = mock.MagicMock()
self.widget.plots.set_plot_x_range = mock.MagicMock()
self.widget._x_range_changed(xbounds)
self.assertEquals(self.widget._context.set_xBounds.call_count, 0)
self.widget.plots.set_plot_x_range.assert_called_with(names,xbounds)
def test_yRangeChanged(self):
names = ["two","three"]
ybounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_yBounds = mock.MagicMock()
self.widget.plots.set_plot_y_range = mock.MagicMock()
self.widget._y_range_changed(ybounds)
self.widget._context.set_yBounds.assert_called_with(ybounds)
self.widget.plots.set_plot_y_range.assert_called_with(names,ybounds)
def test_yRangeChanged1(self):
names = ["two"]
ybounds = [9,18]
# mocks as we only want to test logic
self.widget.quickEdit.get_selection = mock.MagicMock(return_value =names)
self.widget._context.set_yBounds = mock.MagicMock()
self.widget.plots.set_plot_y_range = mock.MagicMock()
self.widget._y_range_changed(ybounds)
self.assertEquals(self.widget._context.set_yBounds.call_count, 0)
self.widget.plots.set_plot_y_range.assert_called_with(names,ybounds)
def test_checkAllErrorsFalse(self):
context = data()
self.widget._context.subplots = context
self.assertEquals(self.widget._check_all_errors(context.keys()),False)
def test_checkAllErrorsTrue(self):
context = data()
for name in context.keys():
context[name].error = True
self.widget._context.subplots = context
self.assertEquals(self.widget._check_all_errors(context.keys()),True)
def test_checkAllErrors1True(self):
context = data()
context["two"].error = True
self.widget._context.subplots = context
self.assertEquals(self.widget._check_all_errors(context.keys()),False)
