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)
