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')