ioex/tests/calcex/test_figure_yaml.py

# -*- coding: utf-8 -*-
import pytest

from ioex.calcex import Figure, UnitMismatchError
yaml = pytest.importorskip('yaml')


@pytest.mark.parametrize(('yaml_loader'), [yaml.Loader, yaml.SafeLoader])
@pytest.mark.parametrize(('figure_yaml', 'expected_figure'), [
    ['!fig {value: null, unit: null}', Figure()],
    ['!fig {value: 123.4, unit: null}', Figure(123.4)],
    ['!fig {value: [1, 2], unit: null}', Figure([1, 2])],
    ['!fig {value: null, unit: kg}', Figure(unit='kg')],
    ['!fig {value: 123.4, unit: km/h}', Figure(123.4, 'km/h')],
    ['!fig {value: [1, 2], unit: km/h}', Figure([1, 2], 'km/h')],
])
def test_register_yaml_constructor(figure_yaml, expected_figure, yaml_loader):
    class TestLoader(yaml_loader):
        pass
    Figure.register_yaml_constructor(TestLoader, tag='!fig')
    assert expected_figure == yaml.load(figure_yaml, Loader=TestLoader)


@pytest.mark.parametrize(('yaml_dumper'), [yaml.Dumper, yaml.SafeDumper])
@pytest.mark.parametrize(('figure'), [
    Figure(),
    Figure(123.4),
    Figure([1, 2]),
    Figure(None, u'm/s²'),
    Figure(123.4, u'm/s²'),
    Figure(1234, u'米/s²'),
])
def test_to_yaml(figure, yaml_dumper):
    class TestDumper(yaml_dumper):
        pass
    TestDumper.add_representer(
        type(figure),
        lambda d, f: figure.to_yaml(d, f, '!test-figure'))
    figure_yaml = yaml.dump(figure, Dumper=TestDumper)

    class TestLoader(yaml.SafeLoader):
        pass
    TestLoader.add_constructor(
        '!test-figure',
        lambda loader, node: loader.construct_mapping(node),
    )
    figure_attr = yaml.load(figure_yaml, Loader=TestLoader)
    assert set(['value', 'unit']) == set(figure_attr.keys())
    assert figure.value == figure_attr['value']
    assert figure.unit == figure_attr['unit']


@pytest.mark.parametrize(('yaml_dumper'), [yaml.Dumper, yaml.SafeDumper])
@pytest.mark.parametrize(('figure'), [
    Figure(123.4, u'm/s²'),
])
def test_register_yaml_representer(figure, yaml_dumper):
    class TestDumper(yaml_dumper):
        pass
    figure.register_yaml_representer(TestDumper)
    figure_yaml = yaml.dump(figure, Dumper=TestDumper)

    class TestLoader(yaml.SafeLoader):
        pass
    TestLoader.add_constructor(
        '!figure',
        lambda loader, node: loader.construct_mapping(node),
    )
    figure_attr = yaml.load(figure_yaml, Loader=TestLoader)
    assert set(['value', 'unit']) == set(figure_attr.keys())
    assert figure.value == figure_attr['value']
    assert figure.unit == figure_attr['unit']