X-Git-Url: https://plomlompom.com/repos/%7B%7Bprefix%7D%7D/ledger2?a=blobdiff_plain;f=tests%2Ftest_days.py;h=02f72af140dd13cbc189af14ade4b856cf1a4c63;hb=934d0d4db064ba3cfdc9e826c55b67c75a383f3d;hp=ba627e98cfbfb2c010510f828e965678ba8e783f;hpb=96b8f35df2ed6f7db0c7c8ffa8234e2925e8756c;p=plomtask diff --git a/tests/test_days.py b/tests/test_days.py index ba627e9..02f72af 100644 --- a/tests/test_days.py +++ b/tests/test_days.py @@ -1,5 +1,6 @@ """Test days module.""" from unittest import TestCase +from datetime import datetime from plomtask.days import Day from plomtask.misc import HandledException @@ -11,9 +12,23 @@ class DayTests(TestCase): """Test Day's date format..""" with self.assertRaises(HandledException): Day('foo') + with self.assertRaises(HandledException): Day(None) + with self.assertRaises(HandledException): Day(3) + with self.assertRaises(HandledException): Day('2024-02-30') + with self.assertRaises(HandledException): + Day('2024-02-01 23:00:00') + self.assertEqual(datetime(2024, 1, 1), Day('2024-01-01').datetime) + + def test_Day_sorting(self): + """Test Day.__lt__.""" + day1 = Day('2024-01-01') + day2 = Day('2024-01-02') + day3 = Day('2024-01-03') + days = [day3, day1, day2] + self.assertEqual(sorted(days), [day1, day2, day3]) def test_Day_weekday(self): """Test Day.weekday."""