def generate_test_transform_treasuries_data(): data = data_loader.download_csv(data_loader.treasuries_config['TREASURIES_URL']) nb_rows = len(data) test_data = data[:10] + data[nb_rows - 10:] utils.serialise_obj(test_data, 'testdata/test_transform_treasuries_data.data.py') utils.serialise_obj(data_loader.transform_treasuries_data(test_data), 'testdata/test_transform_treasuries_data.result.py')
def test_transform_treasuries_data(self): """ Test the transformation of the raw data received from the fed treasury website into the form we use downstream """ test_data = os.path.join(self.data_folder, 'test_transform_treasuries_data.data.py') data = utils.deserialise_obj(test_data) result = data_loader.transform_treasuries_data(data) test_result = os.path.join(self.data_folder, 'test_transform_treasuries_data.result.py') base_result = utils.deserialise_obj(test_result) self.assertEqual(result, base_result)