def test_star_and_kw_args(self):
c = 5.0
context_basic = context_function(basic, dict)
kwarg = dict(b=2)
self.assertEqual(basic(b=2, *(1,)), context_basic(b=2, *(1,)))
self.assertEqual(basic(*(1,), **kwarg), context_basic(*(1,), **kwarg))
context_star_and_kw_args = context_function(star_and_kw_args, dict)
a=2
self.assertEqual(star_and_kw_args(a, 2, 3, d=4, e=5),
context_star_and_kw_args(a, 2, 3, d=4, e=5))
self.assertEqual(star_and_kw_args(a, b=2, c=3, d=4, e=5),
context_star_and_kw_args(a, b=2, c=3, d=4, e=5))
self.assertEqual(star_and_kw_args(a=a, b=2, c=3, d=4, e=5),
context_star_and_kw_args(a=a, b=2, c=3, d=4, e=5))
self.assertEqual(star_and_kw_args(a, 2, 3, 4, 5),
context_star_and_kw_args(a, 2, 3, 4, 5))
self.assertEqual(star_and_kw_args(a, 2, *(3, 4, 5)),
context_star_and_kw_args(a, 2, *(3, 4, 5)))
self.assertEqual(star_and_kw_args(*(a, 2, 3, 4, 5)),
context_star_and_kw_args(*(a, 2, 3, 4, 5)))
kwarg = dict(a=a, b=2, c=3, d=4, e=5)
self.assertEqual(star_and_kw_args(**kwarg),
context_star_and_kw_args(**kwarg))
arg = (a, 2, 3); kwarg = dict(d=4, e=5)
self.assertEqual(star_and_kw_args(*arg, **kwarg),
context_star_and_kw_args(*arg, **kwarg))
arg = (2, 3); kwarg = dict(d=4, e=5)
self.assertEqual(star_and_kw_args(a, *arg, **kwarg),
context_star_and_kw_args(a, *arg, **kwarg))
def test_star_and_kw_args(self):
c = 5.0
context_basic = context_function(basic, dict)
kwarg = dict(b=2)
self.assertEqual(basic(b=2, *(1, )), context_basic(b=2, *(1, )))
self.assertEqual(basic(*(1, ), **kwarg),
context_basic(*(1, ), **kwarg))
context_star_and_kw_args = context_function(star_and_kw_args, dict)
a = 2
self.assertEqual(star_and_kw_args(a, 2, 3, d=4, e=5),
context_star_and_kw_args(a, 2, 3, d=4, e=5))
self.assertEqual(star_and_kw_args(a, b=2, c=3, d=4, e=5),
context_star_and_kw_args(a, b=2, c=3, d=4, e=5))
self.assertEqual(star_and_kw_args(a=a, b=2, c=3, d=4, e=5),
context_star_and_kw_args(a=a, b=2, c=3, d=4, e=5))
self.assertEqual(star_and_kw_args(a, 2, 3, 4, 5),
context_star_and_kw_args(a, 2, 3, 4, 5))
self.assertEqual(star_and_kw_args(a, 2, *(3, 4, 5)),
context_star_and_kw_args(a, 2, *(3, 4, 5)))
self.assertEqual(star_and_kw_args(*(a, 2, 3, 4, 5)),
context_star_and_kw_args(*(a, 2, 3, 4, 5)))
kwarg = dict(a=a, b=2, c=3, d=4, e=5)
self.assertEqual(star_and_kw_args(**kwarg),
context_star_and_kw_args(**kwarg))
arg = (a, 2, 3)
kwarg = dict(d=4, e=5)
self.assertEqual(star_and_kw_args(*arg, **kwarg),
context_star_and_kw_args(*arg, **kwarg))
arg = (2, 3)
kwarg = dict(d=4, e=5)
self.assertEqual(star_and_kw_args(a, *arg, **kwarg),
context_star_and_kw_args(a, *arg, **kwarg))
def test_star_args(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic(*(1, 2)), context_basic(*(1, 2)))
context_star_args = context_function(star_args, dict)
a = 2
self.assertEqual(star_args(a, 2, 3, 4, 5),
context_star_args(a, 2, 3, 4, 5))
self.assertEqual(star_args(a, 2, *(3, 4, 5)),
context_star_args(a, 2, *(3, 4, 5)))
self.assertEqual(star_args(*(a, 2, 3, 4, 5)),
context_star_args(*(a, 2, 3, 4, 5)))
def test_star_args(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic(*(1,2)), context_basic(*(1,2)))
context_star_args = context_function(star_args, dict)
a=2
self.assertEqual(star_args(a, 2, 3, 4, 5),
context_star_args(a, 2, 3, 4, 5))
self.assertEqual(star_args(a, 2, *(3, 4, 5)),
context_star_args(a, 2, *(3, 4, 5)))
self.assertEqual(star_args(*(a, 2, 3, 4, 5)),
context_star_args(*(a, 2, 3, 4, 5)))
def test_kw_args(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic(a=1, b=2), context_basic(a=1, b=2))
self.assertEqual(basic(1, b=2), context_basic(1, b=2))
context_kw_args = context_function(kw_args, dict)
a = 2
self.assertEqual(kw_args(a, b=2, c=3, d=4, e=5),
context_kw_args(a, b=2, c=3, d=4, e=5))
self.assertEqual(kw_args(a=a, b=2, c=3, d=4, e=5),
context_kw_args(a=a, b=2, c=3, d=4, e=5))
kwarg = dict(a=a, b=2, c=3, d=4, e=5)
self.assertEqual(kw_args(**kwarg), context_kw_args(**kwarg))
def test_kw_args(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic(a=1, b=2), context_basic(a=1, b=2))
self.assertEqual(basic(1, b=2), context_basic(1, b=2))
context_kw_args = context_function(kw_args, dict)
a=2
self.assertEqual(kw_args(a, b=2, c=3, d=4, e=5),
context_kw_args(a, b=2, c=3, d=4, e=5))
self.assertEqual(kw_args(a=a, b=2, c=3, d=4, e=5),
context_kw_args(a=a, b=2, c=3, d=4, e=5))
kwarg = dict(a=a, b=2, c=3, d=4, e=5)
self.assertEqual(kw_args(**kwarg),
context_kw_args(**kwarg))
def test_func_attributes(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic.__name__, context_basic.__name__)
self.assertEqual(basic.__module__, context_basic.__module__)
self.assertEqual(basic.__doc__, context_basic.__doc__)
self.assertEqual(basic.__dict__, context_basic.__dict__)
def test_nested_functions(self):
c = 5.0
f = nested_functions(10)
context_f = context_function(f, dict)
self.assertEqual(context_f(3),f(3))
self.assertRaises(ContextFunctionError, context_function,
nested_functions, dict)
def test_nested_functions(self):
c = 5.0
f = nested_functions(10)
context_f = context_function(f, dict)
self.assertEqual(context_f(3), f(3))
self.assertRaises(ContextFunctionError, context_function,
nested_functions, dict)
def test_func_attributes(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic.__name__, context_basic.__name__)
self.assertEqual(basic.__module__, context_basic.__module__)
self.assertEqual(basic.__doc__, context_basic.__doc__)
self.assertEqual(basic.__dict__, context_basic.__dict__)
def test_global_replacement(self):
def numpy_math_factory():
return {'math': numpy}
numpy_func = context_function(math_func, numpy_math_factory)
assert numpy.allclose(
numpy_func(numpy.array([0.0, 0.25, 0.5]) * numpy.pi),
numpy.array([1.0, 2.1213203435596424, 2.0]))
def test_basic_functionality(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic(1, 2), context_basic(1, 2))
self.assertEqual(basic(-1, 2), context_basic(-1, 2))
c = 10.0
self.assertEqual(basic(1, 2), context_basic(1, 2))
self.assertEqual(basic(-1, 2), context_basic(-1, 2))
def test_basic_functionality(self):
c = 5.0
context_basic = context_function(basic, dict)
self.assertEqual(basic(1, 2), context_basic(1, 2))
self.assertEqual(basic(-1, 2), context_basic(-1, 2))
c = 10.0
self.assertEqual(basic(1, 2), context_basic(1, 2))
self.assertEqual(basic(-1, 2), context_basic(-1, 2))
def test_accumulator(self):
accumulator_dict = {'total': 0}
def accumulator_factory():
return accumulator_dict
new_accumulator = context_function(accumulator, accumulator_factory)
for i in range(10):
assert new_accumulator(i) == sum(range(i+1))
assert accumulator_dict['total'] == 45
def test_data_structures(self):
context_data = context_function(data_structures, dict)
a = [2, 3, 'a']
b = {
12: 7,
'a': 3,
'4': 56
}
self.assertEqual(data_structures(a, b), context_data(a, b))
def test_accumulator(self):
accumulator_dict = {'total': 0}
def accumulator_factory():
return accumulator_dict
new_accumulator = context_function(accumulator, accumulator_factory)
for i in range(10):
assert new_accumulator(i) == sum(range(i + 1))
assert accumulator_dict['total'] == 45
def test_default_args(self):
c = 5.0
context_default = context_function(default_args, dict)
self.assertEqual(default_args(1, 2), context_default(1, 2))
self.assertEqual(default_args(1), context_default(1))
def test_internal_from(self):
internal_from_context = context_function(internal_from, dict)
self.assertEqual(internal_from_context(0), internal_from(0))
def test_internal_import(self):
internal_import_context = context_function(internal_import, dict)
self.assertEqual(internal_import_context(0), internal_import(0))
def test_global_replacement(self):
def numpy_math_factory():
return {'math': numpy}
numpy_func = context_function(math_func, numpy_math_factory)
assert numpy.allclose(numpy_func(numpy.array([0.0, 0.25, 0.5])*numpy.pi),
numpy.array([1.0, 2.1213203435596424, 2.0]))
def test_data_structures(self):
context_data = context_function(data_structures, dict)
a = [2, 3, 'a']; b = {12: 7, 'a': 3, '4': 56}
self.assertEqual(data_structures(a, b), context_data(a, b))
def test_internal_import(self):
internal_import_context = context_function(internal_import, dict)
self.assertEqual(internal_import_context(0), internal_import(0))
def test_internal_from(self):
internal_from_context = context_function(internal_from, dict)
self.assertEqual(internal_from_context(0), internal_from(0))
def test_default_args(self):
c = 5.0
context_default = context_function(default_args, dict)
self.assertEqual(default_args(1, 2), context_default(1, 2))
self.assertEqual(default_args(1), context_default(1))
