def test_get_merged_dataset_creates_merged_pandas_dataframe(self): data = MergedDataset([('NSE/OIL', { 'column_index': [1, 2] }), ('WIKI/AAPL', { 'column_index': [1] }), ('WIKI/MSFT')]).data() self.assertIsInstance(data.to_pandas(), pandas.core.frame.DataFrame)
def test_get_merged_dataset_data_returns_specified_columns(self): data = MergedDataset( [('NSE/OIL', {'column_index': [1, 2]}), ('SINGLE/COLUMN', {'column_index': [1]}), ('GOOG/NASDAQ_MSFT')]).data() actual = data.to_pandas().columns.tolist() expected = [six.u('NSE/OIL - column.1'), six.u('NSE/OIL - column.2'), six.u('SINGLE/COLUMN - column.1'), six.u('GOOG/NASDAQ_MSFT - column.1'), six.u('GOOG/NASDAQ_MSFT - column.2'), six.u('GOOG/NASDAQ_MSFT - column.3')] self.assertItemsEqual(actual, expected)
def test_get_merged_dataset_data_returns_specified_columns(self): data = MergedDataset([('NSE/OIL', { 'column_index': [1, 2] }), ('SINGLE/COLUMN', { 'column_index': [1] }), ('WIKI/MSFT')]).data() actual = data.to_pandas().columns.tolist() expected = [ six.u('NSE/OIL - column.1'), six.u('NSE/OIL - column.2'), six.u('SINGLE/COLUMN - column.1'), six.u('WIKI/MSFT - column.1'), six.u('WIKI/MSFT - column.2'), six.u('WIKI/MSFT - column.3') ] self.assertItemsEqual(actual, expected)
def test_get_merged_dataset_creates_merged_pandas_dataframe(self): data = MergedDataset( [('NSE/OIL', {'column_index': [1, 2]}), ('GOOG/NASDAQ_AAPL', {'column_index': [1]}), ('GOOG/NASDAQ_MSFT')]).data() self.assertIsInstance(data.to_pandas(), pandas.core.frame.DataFrame)