def test_clear_data(self):
with quite.EventLoop(0.1):
self.plot_widget = quite.PlotWidget()
x = np.random.random(10)
y = np.random.random(10)
self.plot_widget.plot(x=x, y=y)
self.plot_widget.clear()
self.assertEqual(self.plot_widget.get_data(), None)
def test_error_plot_data(self):
with quite.EventLoop(0.1):
self.plot_widget = quite.PlotWidget()
x = np.random.random(10)
y = np.random.random(9)
with self.assertRaises(Exception):
self.plot_widget.plot(x=x, y=y)
self.plot_widget.append_plot(x=x, y=y)
def test_set_labels(self):
with quite.EventLoop(0.1):
self.plot_widget = quite.PlotWidget()
valid_labels = ['X', 'Y']
valid_units = ['x', 'y']
self.assertEqual(self.plot_widget.set_labels(text=valid_labels, units=valid_units), None)
in_valid_labels = ['X', 'Y', 'Z']
self.assertRaises(ValueError, self.plot_widget.set_labels, in_valid_labels)
in_valid_units = ['x']
self.assertRaises(ValueError, self.plot_widget.set_labels, valid_labels, in_valid_units)
def test_plot_data(self):
with quite.EventLoop(0.1):
self.plot_widget = quite.PlotWidget()
x = np.random.random(10)
y = np.random.random(10)
self.plot_widget.plot(x=x, y=y)
self.assertTrue((self.plot_widget.get_data() == np.array([x, y])).all())
new_x = np.random.random(9)
new_y = np.random.random(9)
self.plot_widget.plot(x=new_x, y=new_y)
self.assertTrue((self.plot_widget.get_data() == np.array([new_x, new_y])).all())
add_x = np.random.random(5)
add_y = np.random.random(5)
old_data = self.plot_widget.get_data()
self.plot_widget.append_plot(x=add_x, y=add_y)
new_data = self.plot_widget.get_data()
self.assertTrue((new_data == np.array(np.append(old_data[0], add_x)), np.append(old_data[1], add_y)))