didier/tests/test_data/test_schedule.py

28 lines
833 B
Python
Raw Normal View History

2021-08-08 20:26:53 +02:00
import pytz
from data import schedule
from datetime import datetime
2021-08-08 18:12:16 +02:00
import unittest
class TestSchedule(unittest.TestCase):
2021-08-08 20:26:53 +02:00
def test_holiday_has_passed(self):
tz = pytz.timezone("Europe/Brussels")
before = datetime(2020, 8, 8, tzinfo=tz)
during = datetime(2021, 6, 2, tzinfo=tz)
after = datetime(2021, 8, 8, tzinfo=tz)
holiday = schedule.Holiday([1, 6, 2021], [2, 7, 2021])
self.assertFalse(holiday.has_passed(before))
self.assertFalse(holiday.has_passed(during))
self.assertTrue(holiday.has_passed(after))
def test_course_str(self):
course = schedule.Course("Test")
self.assertEqual(str(course), "Test")
def test_location_str(self):
location = schedule.Location("C", "B", "R")
self.assertEqual(str(location), "C B R")