def test_register_yaml_constructor(figure_yaml, expected_figure, yaml_loader):
class TestLoader(yaml_loader):
pass
Figure.register_yaml_constructor(TestLoader, tag='!fig')
generated_figure = yaml.load(figure_yaml, Loader=TestLoader)
assert expected_figure == generated_figure
assert isinstance(generated_figure.value, type(expected_figure.value))
def test_set_unit(unit):
f = Figure()
assert None == f.unit
f.unit = unit
assert unit == f.unit
f.unit = None
assert None == f.unit
f.set_unit(unit)
assert unit == f.unit
def test_set_value(value):
f = Figure()
assert None == f.value
f.value = value
assert value == f.value
f.value = None
assert None == f.value
f.set_value(value)
assert value == f.value
def test_sub_persistent():
a = Figure(1, ['m'])
b = Figure(2, ['m'])
d = a - b
assert Figure(-1, ['m']) == d
a.value = 3
a.unit[0] = 'g'
b.value = 4
b.unit[0] = 'l'
assert Figure(-1, ['m']) == d
def test_add_persistent():
a = Figure([1], ['m'])
b = Figure([2], ['m'])
s = a + b
assert Figure([1, 2], ['m']) == s
a.value[0] = 3
a.unit[0] = 'g'
b.value[0] = 4
b.unit[0] = 'l'
assert Figure([1, 2], ['m']) == s
