ioex/tests/datetimeex/test_yaml_timestamp_constructor.py

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

yaml = pytest.importorskip('yaml')
import datetime
import dateutil.tz.tz
import ioex.datetimeex
import pytz

@pytest.mark.parametrize(('loader'), [yaml.Loader, yaml.SafeLoader])
@pytest.mark.parametrize(('yaml_string', 'expected_timestamp'), [
    ['2016-07-14 13:50:04', datetime.datetime(2016, 7, 14, 13, 50, 4, 0)],
    ['2016-07-14 13:50:04Z', pytz.utc.localize(datetime.datetime(2016, 7, 14, 13, 50, 4, 0))],
    ['2016-07-14 13:50:04Z', datetime.datetime(2016, 7, 14, 13, 50, 4, 0, tzinfo = dateutil.tz.tz.tzutc())],
    ['2016-01-14 13:50:04+01:00', pytz.timezone('Europe/Vienna').localize(datetime.datetime(2016, 1, 14, 13, 50, 4, 0))],
    ['2016-07-14 13:50:04+02:00', pytz.timezone('Europe/Vienna').localize(datetime.datetime(2016, 7, 14, 13, 50, 4, 0))],
    ['2016-07-14 13:50:04+02:00', datetime.datetime(2016, 7, 14, 13, 50, 4, 0, tzinfo = dateutil.tz.tz.tzoffset('Vienna', 2 * 60 * 60))],
    ['2016-07-14 13:50:04-07:00', pytz.timezone('US/Pacific').localize(datetime.datetime(2016, 7, 14, 13, 50, 4, 0))],
    ])
def test_from_yaml(yaml_string, expected_timestamp, loader):
    # create subclass so call to class method does not interfere with other tests
    # see yaml.BaseConstructor.add_constructor()
    class TestLoader(loader):
        pass
    ioex.datetimeex.register_yaml_timestamp_constructor(TestLoader)
    loaded_timestamp = yaml.load(yaml_string, Loader = TestLoader)
    assert loaded_timestamp == expected_timestamp
    assert loaded_timestamp.utcoffset() == expected_timestamp.utcoffset()

@pytest.mark.parametrize(('yaml_string', 'tag', 'expected_timestamp'), [
    ['!without_timezone 2016-07-14 13:50:04', '!without_timezone', datetime.datetime(2016, 7, 14, 13, 50, 4, 0)],
    ['!datetime 2016-07-14 13:50:04Z', '!datetime', pytz.utc.localize(datetime.datetime(2016, 7, 14, 13, 50, 4, 0))],
    ])
def test_from_yaml_tag(yaml_string, tag, expected_timestamp):
    # create subclass so call to class method does not interfere with other tests
    # see yaml.BaseConstructor.add_constructor()
    class TestLoader(yaml.SafeLoader):
        pass
    ioex.datetimeex.register_yaml_timestamp_constructor(TestLoader, tag = tag)
    assert yaml.load(yaml_string, Loader = TestLoader) == expected_timestamp

@pytest.mark.parametrize(('yaml_string', 'expected_timestamp'), [
    ['2016-07-14 13:50:04', datetime.datetime(2016, 7, 14, 13, 50, 4, 0)],
    ['2016-07-14 13:50:04Z', pytz.utc.localize(datetime.datetime(2016, 7, 14, 13, 50, 4, 0))],
    ])
def test_from_yaml_repeat(yaml_string, expected_timestamp):
    # create subclass so call to class method does not interfere with other tests
    # see yaml.BaseRepresenter.add_representer()
    class TestLoader(yaml.SafeLoader):
        pass
    ioex.datetimeex.register_yaml_timestamp_constructor(TestLoader)
    assert yaml.load(yaml_string, Loader = TestLoader) == expected_timestamp
    ioex.datetimeex.register_yaml_timestamp_constructor(TestLoader)
    assert yaml.load(yaml_string, Loader = TestLoader) == expected_timestamp