def test_get_merged_dataset_data_is_descending_when_specified_in_params(self): data = MergedDataset(['NSE/OIL', 'GOOG/NASDAQ_AAPL', 'GOOG/NASDAQ_MSFT']).data(params={'order': 'desc'}) results = data.to_list() dates = list([x[0] for x in results]) self.assertTrue(all(dates[i] >= dates[i + 1] for i in range(len(dates) - 1)))
def test_get_merged_dataset_data_is_descending_when_specified_in_params( self): data = MergedDataset(['NSE/OIL', 'WIKI/AAPL', 'WIKI/MSFT']).data(params={'order': 'desc'}) results = data.to_list() dates = list([x[0] for x in results]) self.assertTrue( all(dates[i] >= dates[i + 1] for i in range(len(dates) - 1)))
def test_get_merged_dataset_data_to_list(self): data = MergedDataset( [('NSE/OIL', {'column_index': [1, 2]}), ('SINGLE/COLUMN', {'column_index': [1]}), 'GOOG/NASDAQ_MSFT']).data() results = data.to_list() # NSE/OIL two columns of data # SINGLE/COLUMN one column of data # GOOG/NASDAQ_MSFT all 3 columns of data expected = [[datetime.datetime(2015, 7, 11, 0, 0), 444.3, 10, 444.3, 444.3, 10, 3], [datetime.datetime(2015, 7, 13, 0, 0), 433.3, 4, 433.3, 433.3, 4, 3], [datetime.datetime(2015, 7, 14, 0, 0), 437.5, 3, 437.5, 437.5, 3, 3], [datetime.datetime(2015, 7, 15, 0, 0), 440.0, 2, 440.0, 440.0, 2, 3]] for index, expected_item in enumerate(expected): self.assertItemsEqual(results[index], expected_item)
def test_get_merged_dataset_data_to_list(self): data = MergedDataset([('NSE/OIL', { 'column_index': [1, 2] }), ('SINGLE/COLUMN', { 'column_index': [1] }), 'WIKI/MSFT']).data() results = data.to_list() # NSE/OIL two columns of data # SINGLE/COLUMN one column of data # WIKI/MSFT all 3 columns of data expected = [[ datetime.datetime(2015, 7, 11, 0, 0), 444.3, 10, 444.3, 444.3, 10, 3 ], [ datetime.datetime(2015, 7, 13, 0, 0), 433.3, 4, 433.3, 433.3, 4, 3 ], [ datetime.datetime(2015, 7, 14, 0, 0), 437.5, 3, 437.5, 437.5, 3, 3 ], [ datetime.datetime(2015, 7, 15, 0, 0), 440.0, 2, 440.0, 440.0, 2, 3 ]] for index, expected_item in enumerate(expected): self.assertItemsEqual(results[index], expected_item)