diff --git a/test/test_neural_nets.py b/test/test_neural_nets.py index 5d4380d00bd8ebfcaac819f3736e1e90a73e71c1..0adc1cc54233a7635201e76b283db0c93143433f 100644 --- a/test/test_neural_nets.py +++ b/test/test_neural_nets.py @@ -17,7 +17,7 @@ def test_make_embedding_network_properties(): optimizer=RMSprop(lr=0.7, rho=0.9, epsilon=1e-6)) eq_(nn.layers[0].input_dim, 3) eq_(nn.loss, mse) - eq_(nn.optimizer.lr.eval(), 0.7) + assert np.allclose(nn.optimizer.lr.eval(), 0.7) print(nn.layers) # embedding + flatten + (dense->activation) * hidden layers and last layer eq_(len(nn.layers), 2 + 2 * (1 + len(layer_sizes))) @@ -34,7 +34,7 @@ def test_make_hotshot_network_properties(): optimizer=RMSprop(lr=0.7, rho=0.9, epsilon=1e-6)) eq_(nn.layers[0].input_dim, 6) eq_(nn.loss, mse) - eq_(nn.optimizer.lr.eval(), 0.7) + assert np.allclose(nn.optimizer.lr.eval(), 0.7) print(nn.layers) eq_(len(nn.layers), 2 + 2 * (1 + len(layer_sizes)))