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_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_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_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])