def test_activate(): rnn = RNN((2, 2, 1)) rnn.add_link(0, 2) rnn.add_link(0, 3) rnn.add_link(1, 2) rnn.add_link(1, 3) rnn.add_link(2, 4) rnn.add_link(3, 4) res = rnn.activate((1.0, 1.0)) assert res == 4.0
def test_add_link(): rnn = RNN((2, 2, 2)) assert not rnn.add_link(0, 0, 1.1) assert rnn.add_link(0, 2, 1.2) assert rnn.add_link(0, 4, 1.3) assert not rnn.add_link(2, 0, 1.4) assert rnn.add_link(2, 3, 1.5) assert rnn.add_link(2, 4, 1.6) assert not rnn.add_link(4, 0, 1.7) assert not rnn.add_link(4, 5, 1.8) assert rnn.add_link(4, 2, 1.9) assert rnn.wi[0, 0] == 1.2 assert rnn.wo[0, 0] == 1.3 assert rnn.wh[0, 1] == 1.5 assert rnn.wo[0, 2] == 1.6 assert rnn.wb[0, 0] == 1.9