def test_gen_gen_scaling_info(self):
"""Test gen_scaling_info."""
channel_info = types.ChannelTuple(channel=pulse.DriveChannel(0), scaling=0.5)
obj = generators.gen_scaling_info(channel_info)[0]
# type check
self.assertEqual(type(obj), drawing_objects.TextData)
# data check
self.assertEqual(obj.channel, pulse.DriveChannel(0))
self.assertEqual(obj.text, 'x0.5')
# style check
ref_style = {'zorder': self.style['formatter.layer.axis_label'],
'color': self.style['formatter.color.axis_label'],
'size': self.style['formatter.text_size.annotate'],
'va': 'center',
'ha': 'right'}
self.assertDictEqual(obj.styles, ref_style)
def test_gen_baseline(self):
"""Test gen_baseline."""
channel_info = types.ChannelTuple(channel=pulse.DriveChannel(0), scaling=1)
obj = generators.gen_baseline(channel_info)[0]
# type check
self.assertEqual(type(obj), drawing_objects.LineData)
# data check
self.assertEqual(obj.channel, pulse.DriveChannel(0))
self.assertEqual(obj.x, None)
self.assertEqual(obj.y, 0)
# style check
ref_style = {'alpha': self.style['formatter.alpha.baseline'],
'zorder': self.style['formatter.layer.baseline'],
'linewidth': self.style['formatter.line_width.baseline'],
'linestyle': self.style['formatter.line_style.baseline'],
'color': self.style['formatter.color.baseline']}
self.assertDictEqual(obj.styles, ref_style)
