def test_linked_double_linked3(): ''' Linked to a linked variable, first with indices, second without indices ''' G1 = NeuronGroup(5, 'x : 1') G2 = NeuronGroup(10, 'y : 1 (linked)') G2.y = linked_var(G1.x, index=np.arange(5).repeat(2)) G3 = NeuronGroup(10, 'z: 1 (linked)') G3.z = linked_var(G2.y) G1.x = np.arange(5)*0.1 assert_equal(G3.z[:], np.arange(5).repeat(2)*0.1)
def test_linked_double_linked4(): ''' Linked to a linked variable, both use indices ''' G1 = NeuronGroup(5, 'x : 1') G2 = NeuronGroup(10, 'y : 1 (linked)') G2.y = linked_var(G1.x, index=np.arange(5).repeat(2)) G3 = NeuronGroup(10, 'z: 1 (linked)') G3.z = linked_var(G2.y, index=np.arange(10)[::-1]) G1.x = np.arange(5)*0.1 assert_equal(G3.z[:], np.arange(5).repeat(2)[::-1]*0.1)
def test_linked_double_linked1(): ''' Linked to a linked variable, without indices ''' G1 = NeuronGroup(10, 'x : 1') G2 = NeuronGroup(10, 'y : 1 (linked)') G2.y = linked_var(G1.x) G3 = NeuronGroup(10, 'z: 1 (linked)') G3.z = linked_var(G2.y) G1.x = np.arange(10) assert_equal(G3.z[:], np.arange(10))