def test_xlabel(self):
fig, ax = plt.subplots(1, 1)
x = np.arange(0, 1, 0.1)
y = np.sin(x)
res = plot._plot(x, y, ax, xlabel='Test')
x_label = res.xaxis.get_label().get_text()
self.assertEqual('Test', x_label)
def test_title(self):
fig, ax = plt.subplots(1, 1)
x = np.arange(0, 1, 0.1)
y = np.sin(x)
res = plot._plot(x, y, ax, ylabel='Test')
y_label = res.get_title()
self.assertEqual('Test', y_label)
def test_axes(self):
fig, ax = plt.subplots(1, 1)
x = np.arange(0, 1, 0.1)
y = np.sin(x)
res = plot._plot(x, y, ax)
x_plot, y_plot = ax.lines[0].get_xydata().T
self.assertTrue(np.all(x == x_plot))
self.assertTrue(np.all(y == y_plot))