def setUpClass(cls):
cls.d = {'frequency': [[range(1000)]],
'mode_kappa': [[[[[range(1000,2000)]]]]],
'temperature': [11],
'meta': {}}
cls.fig, cls.ax = one()
cls.ax = frequency.add_waterfall(cls.ax, cls.d, 'mode_kappa',
direction='x', main=True, invert=True)
def test_colourmap(self):
cmap = mpl.cm.get_cmap('viridis')
fig, ax = one()
ax = frequency.add_projected_waterfall(ax,
self.d,
'mode_kappa',
'frequency',
direction='x',
main=True,
colour=cmap)
def test_colour(self):
colour = 'red'
fig, ax = one()
ax = frequency.add_waterfall(ax, self.d, 'mode_kappa', direction='x',
main=True, colour=colour)