class TestModelAssignment(unittest.TestCase): def setUp(self): sys.path.insert(0, "../../src") from models import Assignment start_date = datetime.date(2000, 5, 5) due_date = datetime.date(2020, 12, 12) self.assignment1 = Assignment(1) self.assignment2 = Assignment(2, start_date, due_date) def test_assignment_start_date(self): answer1 = datetime.date(2000, 1, 1) answer2 = datetime.date(2000, 5, 5) self.assertEqual(self.assignment1.start_date, answer1) self.assertEqual(self.assignment2.start_date, answer2) def test_assignment_due_date(self): due_date1 = datetime.date(2000, 1, 1) due_date2 = datetime.date(2020, 12, 12) self.assertEqual(self.assignment1.end_date, due_date1) self.assertEqual(self.assignment2.end_date, due_date2) def test_assignment_id(self): self.assertEqual(self.assignment1.assignment_id, 1) self.assertEqual(self.assignment2.assignment_id, 2) def test_assignment_repr(self): self.assertEqual(self.assignment1.__repr__(), "Assignment 1") self.assertEqual(self.assignment2.__repr__(), "Assignment 2")