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