def program(): '''main program for the Assignment 10''' plt.close('all') filename = raw_input('Please enter the filename of the restaurant_grade:\n') print '\n ' print "Data procession, please wait patiently...\n" raw_data = Data(filename) # create data instance df = raw_data.clean() # use class method to process data into a cleaner dataframe print '\n ' all_restaurant_sum(df) # print out answer to Q4 sum_by_boro(df) # print out answer to Q4 graphs(df) # generate plots for Q5 print "Please check the save pdf files for the corresponding graphs."
def test_Data(self): # test if data contains any empty entries filename = 'DOHMH_New_York_City_Restaurant_Inspection_Results.csv' raw_data = Data(filename) # create data instance df = raw_data.clean() self.assertFalse(df.empty)