Python transform_treasuries_data Examples

Example #1

File: test_harness.py Project: lhoghu/python-tools

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

Example #2

File: test_basic.py Project: lhoghu/python-tools

    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)