def test_cols(self):
"""Check the cols argument plots columns"""
# One column
fig = plt.plot_frame(self.frame, cols=10)
self.assertEqual(str(type(fig)), "<class 'bokeh.models.layouts.Tabs'>")
# Multiple columns
fig = plt.plot_frame(self.frame, cols=[10, 100, 1000])
self.assertEqual(str(type(fig)), "<class 'bokeh.models.layouts.Tabs'>")
def test_wavecal(self):
"""Check a wavecal array can be supplied"""
# Order 1 only
wavecal = np.ones((256, 2048))
fig = plt.plot_frame(self.frame, wavecal=wavecal)
self.assertEqual(str(type(fig)), "<class 'bokeh.models.layouts.Tabs'>")
# All orders
wavecal = np.ones((3, 256, 2048))
fig = plt.plot_frame(self.frame, wavecal=wavecal)
self.assertEqual(str(type(fig)), "<class 'bokeh.models.layouts.Tabs'>")
def test_coeffs(self):
"""Check the traces are drawn"""
coeffs = lt.trace_polynomial()
fig = plt.plot_frame(self.frame, scale='log', trace_coeffs=coeffs)
self.assertEqual(str(type(fig)), "<class 'bokeh.models.layouts.Tabs'>")
def test_logscale(self):
"""Test that a plot of log scale can be created"""
fig = plt.plot_frame(self.frame, scale='log')
self.assertEqual(str(type(fig)), "<class 'bokeh.models.layouts.Tabs'>")