ioex/tests/datetimeex/test_duration_yaml.py

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

from ioex.datetimeex import Duration
yaml = pytest.importorskip('yaml')


@pytest.mark.parametrize(('loader'), [yaml.Loader, yaml.SafeLoader])
@pytest.mark.parametrize(('expected_duration', 'yaml_string'), [
    [Duration(years=0),  '!duration\nyears: 0'],
    [Duration(years=32), '!duration\nyears: 32'],
    [Duration(years=0),  '!duration\n{}'],
    [Duration(years=0),  '!duration {}'],
])
def test_from_yaml(expected_duration, yaml_string, loader):
    class TestLoader(loader):
        pass
    Duration.register_yaml_constructor(TestLoader)
    loaded_duration = yaml.load(yaml_string, Loader=TestLoader)
    assert expected_duration == loaded_duration


@pytest.mark.parametrize(('loader'), [yaml.Loader, yaml.SafeLoader])
@pytest.mark.parametrize(('expected_duration', 'yaml_string', 'tag'), [
    [Duration(years=2),  '!dur\nyears: 2', '!dur'],
    [Duration(years=0), '!duration_tag {}', '!duration_tag'],
])
def test_from_yaml_tag(expected_duration, yaml_string, tag, loader):
    class TestLoader(loader):
        pass
    Duration.register_yaml_constructor(TestLoader, tag=tag)
    loaded_duration = yaml.load(yaml_string, Loader=TestLoader)
    assert expected_duration == loaded_duration


@pytest.mark.parametrize(('dumper'), [yaml.Dumper, yaml.SafeDumper])
@pytest.mark.parametrize(('duration', 'yaml_string'), [
    [Duration(years=0),  '!duration {}\n'],
    [Duration(years=32), '!duration\nyears: 32\n'],
])
def test_to_yaml(duration, yaml_string, dumper):
    class TestDumper(dumper):
        pass
    Duration.register_yaml_representer(TestDumper)
    assert yaml_string == yaml.dump(
        duration,
        Dumper=TestDumper,
        default_flow_style=False,
    )