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)
