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