def main(): try: restaurants = RestaurantGrades() restaurants.grade_changes() restaurants.print_changes() restaurants.count_grades() restaurants.plot_grades() except KeyboardInterrupt: print "Exit with interruption." sys.exit() except ArithmeticError, OverflowError: print "Math error occurs."
class TestCase(unittest.TestCase): """This class will test the RestaurantGrades class.""" def setUp(self): self.r = RestaurantGrades() def test_init(self): self.r = RestaurantGrades() self.assertTrue(len(self.r.grades.columns) == 5) self.assertTrue(len(self.r.grades_counts.keys()) == 6) def test_test_restaurant_grades(self): self.r = RestaurantGrades() self.assertTrue(self.r.test_restaurant_grades(30075445) == 0) self.assertTrue(self.r.test_restaurant_grades(40363630) == 1) def test_grade_changes(self): self.r.grade_changes() self.assertTrue(len(self.r.boro_changes) == 5) def test_count_grades(self): self.r.count_grades() self.assertTrue(self.r.grades_counts["total"].isnull().values.any() == False)
def setUp(self): self.r = RestaurantGrades()
def test_test_restaurant_grades(self): self.r = RestaurantGrades() self.assertTrue(self.r.test_restaurant_grades(30075445) == 0) self.assertTrue(self.r.test_restaurant_grades(40363630) == 1)
def test_init(self): self.r = RestaurantGrades() self.assertTrue(len(self.r.grades.columns) == 5) self.assertTrue(len(self.r.grades_counts.keys()) == 6)