def test_apply_to_buffer_raises_stop_iteration(): fn = Mock(return_value='a') utils.apply_to_buffer(fn, xrange(100), size=0)
def test_apply_to_buffer(): fn = Mock(return_value='a') ret = utils.apply_to_buffer(fn, xrange(100), size=5) assert fn.call_count == 5 assert_list_equal(ret, ['a', 'a', 'a', 'a', 'a'])