import datetime import pytest import pytz import symuid # pylint: disable=protected-access @pytest.mark.parametrize(('ts_sec', 'expected_dt'), [ (1528795204, datetime.datetime(2018, 6, 12, 9, 20, 4, tzinfo=datetime.timezone.utc)), ]) def test__timestamp_to_utc_dt(ts_sec, expected_dt): assert expected_dt == symuid._timestamp_to_utc_dt(ts_sec) def test__utc_dt_now(): dt_now = symuid._utc_dt_now() pytz_dt = datetime.datetime.now(pytz.timezone('UTC')) assert abs((dt_now - pytz_dt).total_seconds()) < 5