def test_enrollment_when_student_not_present(self): test_class = ClassList(2) test_class.add_student('Taylor Swift') test_class.add_student('Kanye West') self.assertFalse(test_class.is_enrolled('Sean Bean')) self.assertFalse(test_class.is_enrolled('Martha Stewart'))
def test_enrollment_when_student_present(self): test_class = ClassList(2) test_class.add_student('Snoop Dogg') test_class.add_student('Martha Stewart') self.assertTrue(test_class.is_enrolled('Snoop Dogg')) self.assertTrue(test_class.is_enrolled('Martha Stewart'))