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)