def test_getWeightsForLayer(self): target = NeuralNetwork() target.addLayer(10, 20, lambda x: x**2) tf.debugging.assert_equal(target.getWeightsForLayer(0), tf.zeros([10, 20]))