def test_add_method_star_str(self):
'''
Tests if the star wrapper is properly applied to a function if the
star_wrap argument to add_method is an string that specifies the
argument key in kargs.
'''
data_1 = (1, 2), (3, 4), (5, 6)
data_2 = 4, 2, 8, -5
data_3 = 5, 6, 3, 6, 2
Pipe.add_method(
gener = min,
is_valve = True,
star_wrap = 'key',
)
self.assertEqual(
Pipe(data_1).min(key=lambda a, b: 1 / a),
(5, 6)
)
self.assertEqual(
Pipe(data_2).min(),
-5
)
# test function that shouldn't be starred
self.assertEqual(
Pipe(data_3).min(key=lambda a: 1 / a),
max(data_3)
)
def test_add_method_as_property(self):
data = 1, 2, 3, 4
Pipe.add_method(enumerate, as_property=True)
self.assertEqual(
tuple(Pipe(data).enumerate),
tuple(enumerate(data))
)
def test_pipe_valve_non_iterable(self):
data_1 = 4, 2, 8, -5
data_1_min = min(data_1)
data_2 = ()
# test method that directly passes the data through
Pipe.add_map_method(lambda val: val, 'pass_through')
# min
Pipe.add_method(min, is_valve=True, empty_error=ValueError)
'''
The Pipe has been preloaded and a valve has been added to the pipe.
When the preloaded and a valve is added the reservoir should be drained
by the pipe into the valve function and the result should be returned.
'''
self.assertEqual(
Pipe(data_1).min(),
data_1_min
)
self.assertEqual(
Pipe(data_1).pass_through().min(),
data_1_min
)
'''
Pipe has not been preloaded and a Pipe should be returned.
'''
pipe_1 = Pipe().min()
self.assertEqual(pipe_1(data_1), data_1_min)
self.assertEqual(pipe_1(data_1), data_1_min) # not a repeat
pipe_2 = Pipe().pass_through().min()
self.assertEqual(pipe_2(data_1), data_1_min)
self.assertEqual(pipe_2(data_1), data_1_min) # not a repeat
pip_3 = Pipe().min().pass_through()
self.assertEqual(next(pip_3(data_1)), data_1_min)
pip_4 = Pipe().min().min()
self.assertEqual(pipe_1(data_1), data_1_min)
pipe_5 = pipe_1.pass_through()
self.assertEqual(
next(pipe_5(data_1)),
next(pip_3(data_1))
)
pipe_6 = pipe_1.min()
self.assertEqual(
pipe_6(data_1),
pip_4(data_1)
)
# empty preloaded iterator
with self.assertRaises(ValueError):
print(Pipe(data_2).min())
def test_add_method_double_star_str(self):
data_1 = tuple(dict(a=a, b=b) for a, b in ((1, 2), (3, 4), (5, 6)))
def min_key_arg(iterable, key):
return min(iterable, key=key)
Pipe.add_method(
gener = min_key_arg,
is_valve = True,
double_star_wrap = 'key',
)
self.assertTrue(hasattr(Pipe, 'min_key_arg'))
self.assertEqual(
Pipe(data_1).min_key_arg(key=lambda a, b: 1 / a),
dict(a=5, b=6)
)
def test_pipe_valve_iterable(self):
data_1 = [(1, 2), (3, 4), (5, 6)]
Pipe.add_method(gener=list, is_valve=True)
self.assertEqual(
Pipe(data_1).list(),
list(data_1)
)
pipe_1 = Pipe().list()
self.assertEqual(
pipe_1(data_1),
list(data_1)
)
def test_add_method(self):
# enumerate
Pipe.add_method(enumerate)
self.assertTrue(hasattr(Pipe, 'enumerate'))
self.assertTrue(isinstance(Pipe.enumerate, types.FunctionType))
self.assertTrue(isinstance(Pipe().enumerate, types.MethodType))
with self.assertRaises(AttributeError):
Pipe.add_method(enumerate)
with self.assertRaises(AttributeError):
Pipe.add_method(map, name='enumerate')
# no_over_write is False
Pipe.add_method(enumerate, no_over_write=False)
self.assertTrue(hasattr(Pipe, 'enumerate'))
self.assertTrue(isinstance(Pipe.enumerate, types.FunctionType))
self.assertTrue(isinstance(Pipe().enumerate, types.MethodType))
Pipe.add_method(enumerate, 'enumerate', no_over_write=False)
self.assertTrue(hasattr(Pipe, 'enumerate'))
self.assertTrue(isinstance(Pipe.enumerate, types.FunctionType))
self.assertTrue(isinstance(Pipe().enumerate, types.MethodType))
# test added method
data = 1, 2, 3, 4
self.assertEqual(
tuple(Pipe(data).enumerate()),
tuple(enumerate(data))
)
self.assertEqual(
tuple(Pipe(data).enumerate(2)),
tuple(enumerate(data, 2))
)
self.assertEqual(
tuple(Pipe(data).enumerate(start=2)),
tuple(enumerate(data, start=2))
)
def setUpClass(self):
Pipe.add_method(gener=filter, iter_index=1)
Pipe.add_method(Expand)
Pipe.add_method(gener=tuple, is_valve=True)
def test_add_method_star_int(self):
'''
Tests if the star wrapper is properly applied to a function if the
star_wrap argument to add_method is an integer that specifies the
argument index in args.
'''
data_1 = (1, 2), (3, 4), (5, 6)
def filter_1_1(a):
return 2 * a[0] > a[1]
def filter_1_2(a, b):
return 2 * a > b
data_1_filtered = tuple(filter(filter_1_1, data_1))
self.assertEqual(
data_1_filtered,
tuple(filter(lambda pt: filter_1_2(*pt), data_1))
)
# function before iterator. filter(function, iterator)
Pipe.add_method(
gener = filter,
iter_index = 1,
star_wrap = 0,
)
self.assertTrue(hasattr(Pipe, 'filter'))
self.assertEqual(
tuple(Pipe(data_1).filter(filter_1_1)),
data_1_filtered
)
self.assertEqual(
tuple(Pipe(data_1).filter(filter_1_2)),
data_1_filtered
)
# function after iterator
def func_after_iter(iterator, function):
for val in iterator:
if function(val):
yield val
Pipe.add_method(
gener = func_after_iter,
star_wrap = 1,
)
self.assertEqual(
tuple(Pipe(data_1).func_after_iter(filter_1_1)),
data_1_filtered
)
self.assertEqual(
tuple(Pipe(data_1).func_after_iter(filter_1_2)),
data_1_filtered
)
# function with extra argument at the end
extra_val = 777
def extra_input_last(iterator, function, extra):
if extra != extra_val:
raise ValueError('extra != extra_val, {} != {}'.format(extra, extra_val))
for val in iterator:
if function(val):
yield val
Pipe.add_method(
gener = extra_input_last,
star_wrap = 1,
)
self.assertEqual(
tuple(Pipe(data_1).extra_input_last(filter_1_1, extra_val)),
data_1_filtered
)
self.assertEqual(
tuple(Pipe(data_1).extra_input_last(filter_1_2, extra_val)),
data_1_filtered
)
# function with extra argument at the beginning
extra_val = 888
def extra_input_first(extra, iterator, function):
if extra != extra_val:
raise ValueError('extra != extra_val, {} != {}'.format(extra, extra_val))
for val in iterator:
if function(val):
yield val
Pipe.add_method(
gener = extra_input_first,
iter_index = 1,
star_wrap = 2,
)
self.assertEqual(
tuple(Pipe(data_1).extra_input_first(extra_val, filter_1_1)),
data_1_filtered
)
self.assertEqual(
tuple(Pipe(data_1).extra_input_first(extra_val, filter_1_2)),
data_1_filtered
)
# function with extra argument in the middle
extra_val = 999
def extra_input_middle(iterator, extra, function):
if extra != extra_val:
raise ValueError('extra != extra_val, {} != {}'.format(extra, extra_val))
for val in iterator:
if function(val):
yield val
Pipe.add_method(
gener = extra_input_middle,
star_wrap = 2,
)
self.assertEqual(
tuple(Pipe(data_1).extra_input_middle(extra_val, filter_1_1)),
data_1_filtered
)
self.assertEqual(
tuple(Pipe(data_1).extra_input_middle(extra_val, filter_1_2)),
data_1_filtered
)
# test with a valve function with a star arg
def min_key_arg(iterable, key):
return min(iterable, key=key)
Pipe.add_method(
gener = min_key_arg,
is_valve = True,
star_wrap = 1,
)
self.assertTrue(hasattr(Pipe, 'min_key_arg'))
self.assertEqual(
Pipe(data_1).min_key_arg(lambda a, b: 1 / a),
(5, 6)
)
