def mainwindow(): print 'Restaurants in NYC info: Assignment 10 - Programming for Data Science' while True: try: optionuser = raw_input('Menu: Press 1 to start showing some plots and results, q for exit :') if optionuser == 'q': break else: if int(optionuser) == 1: ratings = import_data() #Creating CleanDB object and cleaning ratings_db = clean.clean_DB(ratings) rating_clean = ratings_db.clean_all() #Creating grades object grades_db = g.grades(rating_clean) print "Data was cleaned! Processing ..." #Calling NYC results printingresults2(grades_db) else: print "That option was not valid, try again!" except ValueError: print "Oops! That was no valid input. Try again with a number!" except KeyboardInterrupt: print 'Keyboard interrupted, try again!' except IOError: print 'File not found!'
def test_grade_definition(self): data = import_data() #If data is ok then a Grade class object must be created assert isinstance(Grade.grades(data), Grade.grades) #if there is missing some column, exception should rise data = data.drop('BORO',1) self.assertRaises(ValueError, Grade.grades, data)
def test_def_grades(self): #Creating an instance of grades class data = import_data() grades_db = Grade.grades(data) #Invalid inputs invalid_type = ['a',1,] invalid_value = [['A', 1, 1], ['A',1]] #Valid inputs valid = [['A', 'A', 'A'], ['A','B']] #Evaluating for grades in invalid_type: self.assertRaises(TypeError, grades_db.test_grades, grades) for grades in invalid_value: self.assertRaises(ValueError, grades_db.test_grades, grades) for grades in valid: assert isinstance(grades_db.test_grades(grades),int)