def test_add_classroom(self): location = os.getcwd() filepath = os.path.join(location, 'schoolbloc', 'data_import', 'testing_data', 'classrooms.csv') csv_import.classrooms_import(filepath) classrooms = Classroom.query.all() classroom = classrooms.pop() self.assertEqual(classroom.room_number, 311)