diff --git a/python/tests/xor_nn.py b/python/tests/xor_nn.py index dd139ca..b4f5749 100644 --- a/python/tests/xor_nn.py +++ b/python/tests/xor_nn.py @@ -110,14 +110,14 @@ def test_xor_nn(): mag_out = xor_nn_mag() assert [round(x) for x in np_out] == [0, 1, 1, 0] assert [round(x) for x in mag_out] == [0, 1, 1, 0] - assert np_out[0] < 0.08 - assert mag_out[0] < 0.06 + assert np_out[0] < 0.1 + assert mag_out[0] < 0.1 assert np_out[1] > 0.95 - assert mag_out[1] > 0.94 + assert mag_out[1] > 0.95 assert np_out[2] > 0.95 assert mag_out[2] > 0.95 - assert np_out[3] < 0.05 - assert mag_out[3] < 0.04 + assert np_out[3] < 0.1 + assert mag_out[3] < 0.1 print(np_out) print(mag_out) assert np.allclose(np_out, mag_out, atol=0.1) \ No newline at end of file