def test_cleaning_function(self):
raw_data = assignment10.load_data()
raw_data = raw_data.ix[:30000]
cleaned_data = assignment10.clean_data(raw_data)
columnNames = cleaned_data.columns.values
for colname in columnNames:
self.assertEqual(cleaned_data[colname].isnull().sum(),0)
self.assertEqual(cleaned_data.GRADE.isin(['P','Z','Not Yet Graded']).sum(),0)
def test_existance_of_generated_files(self):
try:
for filename in glob.glob('grade_improvement*.pdf') :
os.remove( filename ) #first remove all files which are already presented
except IOError:
pass
raw_data = assignment10.load_data()
raw_data = raw_data.ix[:30000]
cleaned_data = assignment10.clean_data(raw_data)
NYC_grade_count = assignment10.GradeCount(cleaned_data)
assignment10.PlotGraphs.Plot(NYC_grade_count,'grade_improvement_nyc.pdf')
Boroughs_grade_count = assignment10.BoroughsGradeCount(cleaned_data)
assignment10.PlotGraphs.PlotBoroughsGraphs(Boroughs_grade_count)
self.assertTrue(os.path.isfile('./grade_improvement_nyc.pdf'))
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_queens.pdf'))
self.assertTrue(os.path.isfile('./grade_improvement_statn.pdf'))
def testing_for_test_restaurant_grades_function(self):
raw_data = assignment10.load_data()
raw_data = raw_data.ix[:30000]
cleaned_data = assignment10.clean_data(raw_data)
self.assertEqual(assignment10.test_restaurant_grades(40358429,cleaned_data),1)
