[+] Add tests

assignments/a2/a2_example_tests.py

@@ -256,23 +256,22 @@ def test_course_compatible() -> None:
     """
     Test is_course_compatible with compatible course
     """
-    # TODO: Create a test
+    assert a2_courses.is_course_compatible(SCHEDULE_1, CON123)
 
 
 def test_course_not_compatible() -> None:
     """
     Test is_course_compatible with incompatible course
     """
-    # TODO: Create a test
+    assert not a2_courses.is_course_compatible(SCHEDULE_1, TMP000)
+    assert not a2_courses.is_course_compatible(SCHEDULE_1, CON333)
 
 
 def test_compatible_sections() -> None:
     """
     Test compatible_sections with compatible sections
     """
-    actual = a2_courses.compatible_sections(SCHEDULE_1, CON123) == {CON123_LEC0123}
+    assert a2_courses.compatible_sections(SCHEDULE_1, CON123) == {CON123_LEC0123}
-    expected = True
-    assert actual == expected
 
 
 # TODO: Create more tests