def test_DB_exists(self): """ Function import_data : Must raise an exception is an invalid name is inputted. Must create a dataframe if there is no argument or if the correct name is inputted :return: """ correct_file = "DOHMH_New_York_City_Restaurant_Inspection_Results.csv" incorrect_file="DOHMH_New_York_City_Restaurant_Inspection_Result.csv" self.assertRaises(IOError, import_data, incorrect_file) assert isinstance(import_data(correct_file), pd.DataFrame) assert isinstance(import_data(), pd.DataFrame)
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)