def test_are_equal(self):
a = [np.array([[1, 2, 3], [4, 5, 6]]),
np.array([[1, 2, 3], [4, 5, 6]])]
b = [np.array([[1, 2, 3], [4, 5, 6]]),
np.array([[1, 2, 3], [4, 5, 6]])]
self.assertTrue(util.list_of_arrays_is_equal(a, b))
def test_get_set_wages_give_same_result(self):
nn = layer.NN(5, [3,2,1], 4)
nn.random_init()
w = nn.get_wages()
for l in nn.layers[1:]:
l.W[0,0] = -17
nn.set_wages(w)
w2 = nn.get_wages()
self.assertTrue(util.list_of_arrays_is_equal(w, w2))
