def test_next_layer(self):
with self.assertRaises(AssertionError):
Layer(15).next_layer('')
x = Layer(1)
y = Layer(2)
x.next_layer(y)
x.next is y
def test_init_weights(self):
x = Layer(1)
x.init_weights()
self.assertIsNone(x.weights)
x = Layer(1)
y = Layer(1)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0]])
x = Layer(1)
y = Layer(2)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0, 0]])
x = Layer(2)
y = Layer(1)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0], [0]])
x = Layer(2)
y = Layer(2)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0, 0], [0, 0]])
def test_init_weights(self):
x = Layer(1)
x.init_weights()
self.assertIsNone(x.weights)
cnf = lambda: 0
x = Layer(1, cnf)
y = Layer(1, cnf)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0]])
x = Layer(1, cnf)
y = Layer(2, cnf)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0, 0]])
x = Layer(2, cnf)
y = Layer(1, cnf)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0], [0]])
x = Layer(2, cnf)
y = Layer(2, cnf)
x.next_layer(y)
x.init_weights()
self.assertEqual(x.weights, [[0, 0], [0, 0]])
