def testFunction(self): r = Restaurant("sample") self.assertEqual(-1, r.test_grades(["A", "B"])) self.assertEqual(1, r.test_grades(["C", "B", "B"])) self.assertEqual(0, r.test_grades(["A"])) self.assertEqual(0, r.test_grades(["C", "B", "C"]))