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'))