def testWrongFilename(self): # test to make sure a KeyError is raised if an invalid filename is # entered self.assertRaises(KeyError, bu.extract_logbook_data('fake_filename'))
def testFilenameCorrect(self): # test to make sure the function works if a valid filename is entered test_data = bu.extract_logbook_data('Lookup_UK_WindDirection.csv') self.assertTrue(test_data.columns.values[0], 'ID') self.assertTrue(test_data.columns.values[0], 'WindDirection') self.assertTrue(test_data.columns.values[0], 'ProbWindDD')