Source code for bmtk.tests.builder.test_connector

from bmtk.builder import connector


[docs] def test_fnc_params(): con_fnc = connector.create(connector=lambda x, p: x**p) assert(con_fnc(2, 3) == 2**3)
[docs] def test_fnc_noparams(): con_fnc = connector.create(connector=lambda x, p, a:x**p+a, a=10) assert(con_fnc(2, 3) == 2**3+10)
[docs] def test_literal(): con_fnc = connector.create(connector=100.0) assert(con_fnc() == 100.0) con_fnc1 = connector.create(connector=101.0, a=10, b='10') # parameters in literals should be ignored assert(con_fnc1() == 101.0)
[docs] def test_list(): con_fnc = connector.create(connector=['a', 'b', 'c']) assert(con_fnc == ['a', 'b', 'c']) con_fnc1 = connector.create(connector=[100, 200, 300], p1=1, p2='2', p34=(3,4)) assert(con_fnc1 == [100, 200, 300])
[docs] def test_dict(): con_fnc = connector.create(connector={'a': 1, 'b': 'b', 'c': [5, 6]}) assert('a' in con_fnc()) assert('b' in con_fnc()) assert('c' in con_fnc()) con_fnc = connector.create(connector={'a': 1, 'b': 'b', 'c': [5, 6]}, p1='p1', p2=2) assert('a' in con_fnc()) assert('b' in con_fnc()) assert('c' in con_fnc())
#test_dict() #test_connector_fnc_params() #test_connector_fnc_noparams()