test/venv/Lib/site-packages/aniso8601/tests/test_utcoffset.py

# -*- coding: utf-8 -*-

# Copyright (c) 2021, Brandon Nielsen
# All rights reserved.
#
# This software may be modified and distributed under the terms
# of the BSD license.  See the LICENSE file for details.

import datetime
import pickle
import unittest

from aniso8601.utcoffset import UTCOffset


class TestUTCOffset(unittest.TestCase):
    def test_pickle(self):
        # Make sure timezone objects are pickleable
        testutcoffset = UTCOffset(name="UTC", minutes=0)

        utcoffsetpickle = pickle.dumps(testutcoffset)

        resultutcoffset = pickle.loads(utcoffsetpickle)

        self.assertEqual(resultutcoffset._name, testutcoffset._name)
        self.assertEqual(resultutcoffset._utcdelta, testutcoffset._utcdelta)

    def test_repr(self):
        self.assertEqual(str(UTCOffset(minutes=0)), "+0:00:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=60)), "+1:00:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=-60)), "-1:00:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=12)), "+0:12:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=-12)), "-0:12:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=83)), "+1:23:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=-83)), "-1:23:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=1440)), "+1 day, 0:00:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=-1440)), "-1 day, 0:00:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=2967)), "+2 days, 1:27:00 UTC")

        self.assertEqual(str(UTCOffset(minutes=-2967)), "-2 days, 1:27:00 UTC")

    def test_dst(self):
        tzinfoobject = UTCOffset(minutes=240)
        # This would raise ISOFormatError or a TypeError if dst info is invalid
        result = datetime.datetime.now(tzinfoobject)
        # Hacky way to make sure the tzinfo is what we'd expect
        self.assertEqual(result.tzinfo.utcoffset(None), datetime.timedelta(hours=4))