def test_python_func(self):
def myFunction(variable, param1, param2):
return sum(variable[0]) + 2
myMech = ProcessingMechanism(function=myFunction, size=4, name='myMech')
# assert 'param1' in myMech.parameter_ports.names # <- FIX reinstate when problem with function params is fixed
# assert 'param2' in myMech.parameter_ports.names # <- FIX reinstate when problem with function params is fixed
val = myMech.execute(input=[-1, 2, 3, 4])
assert np.allclose(val, [[10]])
def test_user_def_func(self):
def myFunction(variable, param1, param2):
return variable * 2 + 3
U = UserDefinedFunction(custom_function=myFunction, default_variable=[[0, 0]], param2=0)
myMech = ProcessingMechanism(function=U, size=2, name='myMech')
# assert 'param1' in myMech.parameter_ports.names # <- FIX reinstate when problem with function params is fixed
assert 'param2' in myMech.parameter_ports.names
val = myMech.execute([1, 3])
assert np.allclose(val, [[5, 9]])
def test_udf_with_pnl_func(self):
L = Logistic(gain=2)
def myFunction(variable, params, context):
return L(variable) + 2
U = UserDefinedFunction(custom_function=myFunction, default_variable=[[0, 0, 0]])
myMech = ProcessingMechanism(function=myFunction, size=3, name='myMech')
val1 = myMech.execute(input=[1, 2, 3])
val2 = U.execute(variable=[[1, 2, 3]])
assert np.allclose(val1, val2)
assert np.allclose(val1, L([1, 2, 3]) + 2)
def mech_with_autogenerated_udf(self):
def func(input=[[0], [0]], p=0, q=1):
return (p + q) * input
m = ProcessingMechanism(default_variable=[[0], [0]], function=func)
return m
def test_udf_system_terminal(self):
def myFunction(variable, params, context):
return [variable[0][2], variable[0][0]]
myMech = ProcessingMechanism(function=myFunction, size=3, name='myMech')
T2 = TransferMechanism(size=3, function=Linear)
p2 = Process(pathway=[T2, myMech])
s2 = System(processes=[p2])
s2.run(inputs = {T2: [[1, 2, 3]]})
assert(np.allclose(s2.results[0][0], [3, 1]))
def test_udf_system_origin(self):
def myFunction(variable, params, context):
return [variable[0][1], variable[0][0]]
myMech = ProcessingMechanism(function=myFunction, size=3, name='myMech')
T = TransferMechanism(size=2, function=Linear)
p = Process(pathway=[myMech, T])
s = System(processes=[p])
s.run(inputs = {myMech: [[1, 3, 5]]})
assert np.allclose(s.results[0][0], [3, 1])
def test_udf_composition_terminal(comp_mode, benchmark):
def myFunction(variable, context):
return [variable[0][2], variable[0][0]]
myMech = ProcessingMechanism(function=myFunction, size=3, name='myMech')
T2 = TransferMechanism(size=3, function=Linear)
c2 = Composition(pathways=[[T2, myMech]])
benchmark(c2.run, inputs={T2: [[1, 2, 3]]}, execution_mode=comp_mode)
assert(np.allclose(c2.results[0][0], [3, 1]))
def test_udf_composition_origin(comp_mode, benchmark):
def myFunction(variable, context):
return [variable[0][1], variable[0][0]]
myMech = ProcessingMechanism(function=myFunction, size=3, name='myMech')
T = TransferMechanism(size=2, function=Linear)
c = Composition(pathways=[myMech, T])
benchmark(c.run, inputs={myMech: [[1, 3, 5]]}, execution_mode=comp_mode)
assert np.allclose(c.results[0][0], [3, 1])
def test_udf_creates_parameter_ports(self):
def func(input=[[0], [0]], p=0, q=1):
return (p + q) * input
m = ProcessingMechanism(default_variable=[[0], [0]],
function=UserDefinedFunction(func))
assert len(m.parameter_ports) == 2
assert 'p' in m.parameter_ports.names
assert 'q' in m.parameter_ports.names
def test_udf_in_mechanism(mech_mode, benchmark):
def myFunction(variable, param1, param2):
return sum(variable[0]) + 2
myMech = ProcessingMechanism(function=myFunction, size=4, name='myMech')
# assert 'param1' in myMech.parameter_ports.names # <- FIX reinstate when problem with function params is fixed
# assert 'param2' in myMech.parameter_ports.names # <- FIX reinstate when problem with function params is fixed
e = pytest.helpers.get_mech_execution(myMech, mech_mode)
val = benchmark(e, [-1, 2, 3, 4])
assert np.allclose(val, [[10]])
def test_udf_in_mechanism(self, bin_execute, benchmark):
def myFunction(variable, param1, param2):
return sum(variable[0]) + 2
myMech = ProcessingMechanism(function=myFunction,
size=4,
name='myMech')
# assert 'param1' in myMech.parameter_ports.names # <- FIX reinstate when problem with function params is fixed
# assert 'param2' in myMech.parameter_ports.names # <- FIX reinstate when problem with function params is fixed
if bin_execute == 'LLVM':
e = pnlvm.execution.MechExecution(myMech).execute
elif bin_execute == 'PTX':
e = pnlvm.execution.MechExecution(myMech).cuda_execute
else:
e = myMech.execute
val = benchmark(e, [-1, 2, 3, 4])
assert np.allclose(val, [[10]])
