def test_kwarg_pass_through_no_kwargs(self):
lazy_func = mock.Mock()
data = mock.sentinel.data
axis = mock.sentinel.axis
aggregator = Aggregator("", None, lazy_func=lazy_func)
aggregator.lazy_aggregate(data, axis)
lazy_func.assert_called_once_with(data, axis=axis)
def test_kwarg_pass_through_no_kwargs(self):
lazy_func = Mock()
data = sentinel.data
axis = sentinel.axis
aggregator = Aggregator('', None, lazy_func=lazy_func)
aggregator.lazy_aggregate(data, axis)
lazy_func.assert_called_once_with(data, axis)
def test_kwarg_pass_through_init_kwargs(self):
lazy_func = Mock()
data = sentinel.data
axis = sentinel.axis
kwargs = dict(wibble='wobble', foo='bar')
aggregator = Aggregator('', None, lazy_func=lazy_func, **kwargs)
aggregator.lazy_aggregate(data, axis)
lazy_func.assert_called_once_with(data, axis, **kwargs)
def test_kwarg_pass_through_init_kwargs(self):
lazy_func = mock.Mock()
data = mock.sentinel.data
axis = mock.sentinel.axis
kwargs = dict(wibble="wobble", foo="bar")
aggregator = Aggregator("", None, lazy_func=lazy_func, **kwargs)
aggregator.lazy_aggregate(data, axis)
lazy_func.assert_called_once_with(data, axis=axis, **kwargs)
def test_kwarg_pass_through_call_kwargs(self):
lazy_func = mock.Mock()
data = mock.sentinel.data
axis = mock.sentinel.axis
kwargs = dict(wibble='wobble', foo='bar')
aggregator = Aggregator('', None, lazy_func=lazy_func)
aggregator.lazy_aggregate(data, axis, **kwargs)
lazy_func.assert_called_once_with(data, axis=axis, **kwargs)
def test_kwarg_pass_through_combined_kwargs(self):
lazy_func = Mock()
data = sentinel.data
axis = sentinel.axis
init_kwargs = dict(wibble='wobble', var=1.0)
call_kwargs = dict(foo='foo', var=0.5)
aggregator = Aggregator('', None, lazy_func=lazy_func, **init_kwargs)
aggregator.lazy_aggregate(data, axis, **call_kwargs)
expected_kwargs = init_kwargs.copy()
expected_kwargs.update(call_kwargs)
lazy_func.assert_called_once_with(data, axis, **expected_kwargs)
def test_kwarg_pass_through_combined_kwargs(self):
lazy_func = mock.Mock()
data = mock.sentinel.data
axis = mock.sentinel.axis
init_kwargs = dict(wibble="wobble", var=1.0)
call_kwargs = dict(foo="foo", var=0.5)
aggregator = Aggregator("", None, lazy_func=lazy_func, **init_kwargs)
aggregator.lazy_aggregate(data, axis, **call_kwargs)
expected_kwargs = init_kwargs.copy()
expected_kwargs.update(call_kwargs)
lazy_func.assert_called_once_with(data, axis=axis, **expected_kwargs)
def test_no_lazy_func(self):
dummy_agg = Aggregator('custom_op', lambda x: 1)
expected = 'custom_op aggregator does not support lazy operation'
with self.assertRaisesRegexp(LazyAggregatorError, expected):
dummy_agg.lazy_aggregate(np.arange(10), axis=0)
def test_no_lazy_func(self):
dummy_agg = Aggregator("custom_op", lambda x: 1)
expected = "custom_op aggregator does not support lazy operation"
with self.assertRaisesRegex(LazyAggregatorError, expected):
dummy_agg.lazy_aggregate(np.arange(10), axis=0)
