update test cases

Author: rxng8

tests/components/neurons/graded/test_bernoulliErrorCell.py

@@ -61,6 +61,6 @@ def clamp_target(x):
   outs = jnp.concatenate(outs, axis=1)
   # print(outs)
   ## output should equal input
-  np.testing.assert_allclose(outs, y_seq, atol=1e-7)
+  np.testing.assert_allclose(outs, y_seq, atol=1e-3)
 
 # test_bernoulliErrorCell()

tests/components/other/test_expKernel.py

@@ -54,6 +54,6 @@ def clamp(x):
     #print(outs)
 
     ## output should equal input
-    assert_array_equal(outs, y_seq)
+    np.testing.assert_allclose(outs, y_seq, atol=1e-8)
 
 #test_expKernel1()

tests/components/synapses/test_STPDenseSynapse.py

@@ -59,7 +59,7 @@ def test_STPDenseSynapse1():
     Wdyn = jnp.concatenate(Wdyn, axis=1)
     # print(outs)
     # print(Wdyn)
-    assert_array_equal(outs, outs_truth)
-    assert_array_equal(Wdyn, Wdyn_truth)
+    np.testing.assert_allclose(outs, outs_truth, atol=1e-8)
+    np.testing.assert_allclose(Wdyn, Wdyn_truth, atol=1e-8)
 
 #test_STPDenseSynapse1()