Add extra tests for schedules & timeformatters

tests/functions/test_timeFormatters.py

@@ -0,0 +1,64 @@
+from datetime import datetime
+from functions import timeFormatters
+import unittest
+
+
+class TestTimeFormatters(unittest.TestCase):
+    def test_leadingZero(self):
+        self.assertEqual("0123", timeFormatters.leadingZero("123"))
+        self.assertEqual("0123", timeFormatters.leadingZero("0123"))
+
+    def test_delimiter(self):
+        self.assertEqual("0123", timeFormatters.delimiter("0123"))
+        self.assertEqual("01.23", timeFormatters.delimiter("0123", delim="."))
+
+    def test_timeFromInt(self):
+        self.assertEqual("01:23", timeFormatters.timeFromInt(123))
+        self.assertEqual("12:34", timeFormatters.timeFromInt(1234))
+
+    def test_fromArray(self):
+        # Only day/month/year
+        d, m, y = 1, 2, 2021
+        inp = [d, m, y]
+
+        dt = timeFormatters.fromArray(inp)
+        self.assertEqual(d, dt.day)
+        self.assertEqual(m, dt.month)
+        self.assertEqual(y, dt.year)
+
+        # Also hours/minutes/seconds
+        d, m, y, hh, mm, ss = 1, 2, 2021, 1, 2, 3
+        inp = [d, m, y, hh, mm, ss]
+
+        dt = timeFormatters.fromArray(inp)
+        self.assertEqual(d, dt.day)
+        self.assertEqual(m, dt.month)
+        self.assertEqual(y, dt.year)
+        self.assertEqual(hh, dt.hour)
+        self.assertEqual(mm, dt.minute)
+        self.assertEqual(ss, dt.second)
+
+    def test_skipWeekends(self):
+        # Already a weekday
+        weekday = datetime(2021, 8, 11)
+        skipped = timeFormatters.skip_weekends(weekday)
+        self.assertEqual(weekday, skipped)
+
+        # Weekend
+        weekend = datetime(2021, 8, 7)
+        skipped = timeFormatters.skip_weekends(weekend)
+        self.assertEqual(0, skipped.weekday())
+
+    def test_forwardToWeekday(self):
+        mo = datetime(2021, 8, 10)
+        # Before
+        forwarded = timeFormatters.forward_to_weekday(mo, 2)
+        self.assertEqual(1, (forwarded - mo).days)
+
+        # Same day
+        forwarded = timeFormatters.forward_to_weekday(mo, 1)
+        self.assertEqual(7, (forwarded - mo).days)
+
+        # After
+        forwarded = timeFormatters.forward_to_weekday(mo, 0)
+        self.assertEqual(6, (forwarded - mo).days)