faori's picture
Upload folder using huggingface_hub
550665c verified
raw
history blame
858 Bytes
from unittest import TestCase
from beautiful_date import Sept
from gcsa.util.date_time_util import ensure_localisation
class TestReminder(TestCase):
def test_ensure_localisation(self):
initial_date = 23 / Sept / 2022
d = ensure_localisation(initial_date)
# Shouldn't do anything to date
self.assertEqual(initial_date, d)
initial_date_time = initial_date[:]
self.assertIsNone(initial_date_time.tzinfo)
dt_with_tz = ensure_localisation(initial_date_time)
self.assertIsNotNone(dt_with_tz.tzinfo)
self.assertNotEqual(dt_with_tz, initial_date_time)
dt_with_tz_unchanged = ensure_localisation(dt_with_tz)
self.assertEqual(dt_with_tz, dt_with_tz_unchanged)
with self.assertRaises(TypeError):
ensure_localisation('Hello')