def test_time_label_image():
m = Movie(dt=1.0 / 14, height_ratio=2)
img = np.arange(100).reshape(4, 5, 5)
m.add_image(img, style='dark_img')
m.add_time_label(color='blue')
assert m.labels[0]['x'] == 0.01
assert m.labels[0]['y'] == 0.08
assert np.allclose(m.labels[0]['values'], np.arange(4) / 14.0)
assert m.labels[0]['axis'] == 0
assert m.labels[0]['s_format'] == '%.2fs'
assert m.labels[0]['size'] == 14
kwargs = m.labels[0]['kwargs']
assert kwargs['color'] == 'blue'
def test_time_label_axis():
m = Movie(dt=1.0 / 14, height_ratio=2)
m.add_axis(x_label='x', y_label='y')
m.add_trace(data=np.arange(10))
m.add_time_label(color='blue')
assert m.labels[0]['x'] == 0.01
assert m.labels[0]['y'] == 0.08
assert np.allclose(m.labels[0]['values'], np.arange(10) / 14.0)
assert m.labels[0]['axis'] == 0
assert m.labels[0]['s_format'] == '%.2fs'
assert m.labels[0]['size'] == 14
kwargs = m.labels[0]['kwargs']
assert kwargs['color'] == 'blue'
def test_time_label_fail():
m = Movie(dt=1.0 / 14, height_ratio=2)
with pytest.raises(RuntimeError) as ex:
m.add_time_label()
assert 'Can not add time labels' in str(ex.value)
