def test_removes_column_index_query_param(self, mock):
self.oil_obj.requested_column_indexes = []
mock.return_value = self.oil_obj
md = MergedDataset(
[('NSE/OIL', {'column_index': [1, 2]})], params={'column_index': 1})
md.data_fields()
expected = call(('NSE/OIL', {'column_index': [1, 2]}), params={})
self.assertEqual(mock.call_args, expected)
def test_sets_column_index_on_each_dataset(self):
md = MergedDataset(
[('NSE/OIL', {'column_index': [1, 2]}),
('GOOG/NASDAQ_AAPL', {'column_index': [1]}),
('GOOG/NASDAQ_MSFT')])
md.data_fields()
self.assertItemsEqual([1, 2], md._datasets[0].requested_column_indexes)
self.assertItemsEqual([1], md._datasets[1].requested_column_indexes)
self.assertItemsEqual([], md._datasets[2].requested_column_indexes)
def test_sets_column_index_on_each_dataset(self):
md = MergedDataset(
[('NSE/OIL', {'column_index': [1, 2]}),
('WIKI/AAPL', {'column_index': [1]}),
('WIKI/MSFT')])
md.data_fields()
six.assertCountEqual(self, [1, 2], md._datasets[0].requested_column_indexes)
six.assertCountEqual(self, [1], md._datasets[1].requested_column_indexes)
six.assertCountEqual(self, [], md._datasets[2].requested_column_indexes)
def test_sets_column_index_on_each_dataset(self):
md = MergedDataset([('NSE/OIL', {
'column_index': [1, 2]
}), ('WIKI/AAPL', {
'column_index': [1]
}), ('WIKI/MSFT')])
md.data_fields()
self.assertItemsEqual([1, 2], md._datasets[0].requested_column_indexes)
self.assertItemsEqual([1], md._datasets[1].requested_column_indexes)
self.assertItemsEqual([], md._datasets[2].requested_column_indexes)
def test_removes_column_index_query_param(self, mock):
self.oil_obj.requested_column_indexes = []
mock.return_value = self.oil_obj
md = MergedDataset([('NSE/OIL', {
'column_index': [1, 2]
})],
params={'column_index': 1})
md.data_fields()
expected = call(('NSE/OIL', {'column_index': [1, 2]}), params={})
self.assertEqual(mock.call_args, expected)
def test_merged_dataset_calls_merged_dataset_get_dataset(self, mock):
mock.return_value = self.oil_obj
md = MergedDataset(
[('NSE/OIL', {'column_index': [1, 2]}),
('GOOG/NASDAQ_AAPL', {'column_index': [1]}),
('GOOG/NASDAQ_MSFT')])
md.data_fields()
expected_calls = [
call(('NSE/OIL', {'column_index': [1, 2]})),
call(('GOOG/NASDAQ_AAPL', {'column_index': [1]})),
call('GOOG/NASDAQ_MSFT')
]
self.assertEqual(mock.call_count, 3)
for index, expected in enumerate(expected_calls):
self.assertEqual(mock.mock_calls[index], expected)
def test_merged_dataset_calls_merged_dataset_get_dataset(self, mock):
mock.return_value = self.oil_obj
md = MergedDataset([('NSE/OIL', {
'column_index': [1, 2]
}), ('WIKI/AAPL', {
'column_index': [1]
}), ('WIKI/MSFT')])
md.data_fields()
expected_calls = [
call(('NSE/OIL', {
'column_index': [1, 2]
})),
call(('WIKI/AAPL', {
'column_index': [1]
})),
call('WIKI/MSFT')
]
self.assertEqual(mock.call_count, 3)
for index, expected in enumerate(expected_calls):
self.assertEqual(mock.mock_calls[index], expected)
