def test_create_plots_files(self):
"""
Creates the plots for the Boroughs
Result: Should create 6 plots
"""
try:
os.remove('grade_improvement_bronx.pdf')
os.remove('grade_improvement_brooklyn.pdf')
os.remove('grade_improvement_manhattan.pdf')
os.remove('grade_improvement_nyc.pdf')
os.remove('grade_improvement_queens.pdf')
os.remove('grade_improvement_staten.pdf')
except OSError:
pass
data_reader = DataReader('DOHMH_New_York_City_Restaurant_Inspection_Results.csv')
inspections_df = data_reader.get_dataframe()
restaurant_grades = RestaurantGrader(inspections_df)
restaurant_grades.grade_improvement_plots()
self.assertTrue(os.path.isfile('grade_improvement_bronx.pdf'))
self.assertTrue(os.path.isfile('grade_improvement_brooklyn.pdf'))
self.assertTrue(os.path.isfile('grade_improvement_manhattan.pdf'))
self.assertTrue(os.path.isfile('grade_improvement_nyc.pdf'))
self.assertTrue(os.path.isfile('grade_improvement_queens.pdf'))
self.assertTrue(os.path.isfile('grade_improvement_staten.pdf'))
def test_test_restaurant_grades(self):
"""
Passing the Camis Id: 30112340 to the function test_restaurant_grades
Result: The grade should be 0
"""
data_reader = DataReader('DOHMH_New_York_City_Restaurant_Inspection_Results.csv')
inspections_df = data_reader.get_dataframe()
restaurant_grades = RestaurantGrader(inspections_df)
self.assertEqual(0, restaurant_grades.test_restaurant_grades(30112340))
def main_restaurants_quality():
try:
data_reader = DataReader('DOHMH_New_York_City_Restaurant_Inspection_Results.csv')
inspections_df = data_reader.get_dataframe()
restaurant_grades = RestaurantGrader(inspections_df)
restaurant_grades.get_boroughs_scores()
restaurant_grades.grade_improvement_plots()
except IOError as io_message:
print io_message
except LookupError as lookup_message:
print lookup_message
def test_test_grades(self):
"""
Passing the list of grades ['A', 'C', 'C', 'C', 'C', 'C', 'A'] to the function test_grades
Result: The grade should be 1
"""
self.assertEqual(1, RestaurantGrader.test_grades(['A', 'C', 'C', 'C', 'C', 'C', 'A']))
