cleaned up a few unit tests to use deterministic syn init vals

Author: Alexander Ororbia

tests/components/synapses/hebbian/test_eventSTDPSynapse.py

@@ -48,14 +48,15 @@ def test_eventSTDPSynapse1():
         evolve_cmd, evolve_args = ctx.compile_by_key(a, compile_key="evolve")
         ctx.add_command(wrap_command(jit(ctx.evolve)), name="adapt")
         """
+    a.weights.set(jnp.ones((1, 1)) * 0.1)
 
     t = 12. ## fake out current time
     ## Case 1: outside of pre-syn time window
     input_tols = jnp.ones((1, 1,)) * 9.
     out_spike = jnp.ones((1, 1))
 
     ## check pre-synaptic STDP only
-    truth = jnp.array([[-0.6296545]])
+    truth = jnp.array([[-0.101]])
     ctx.reset()
     a.pre_tols.set(input_tols)
     a.postSpike.set(out_spike)
@@ -69,7 +70,7 @@ def test_eventSTDPSynapse1():
     out_spike = jnp.ones((1, 1))
 
     ## check pre-synaptic STDP only
-    truth = jnp.array([[0.37034547]])
+    truth = jnp.array([[0.899]])
     ctx.reset()
     a.pre_tols.set(input_tols)
     a.postSpike.set(out_spike)
@@ -79,3 +80,4 @@ def test_eventSTDPSynapse1():
     assert_array_equal(a.dWeights.value, truth)
 
 #test_eventSTDPSynapse1()
+

tests/components/synapses/hebbian/test_expSTDPSynapse.py

@@ -47,14 +47,15 @@ def test_expSTDPSynapse1():
         evolve_cmd, evolve_args = ctx.compile_by_key(a, compile_key="evolve")
         ctx.add_command(wrap_command(jit(ctx.evolve)), name="adapt")
         """
+    a.weights.set(jnp.ones((1, 1)) * 0.1)
 
     in_spike = jnp.ones((1, 1))
     in_trace = jnp.ones((1, 1,)) * 1.25
     out_spike = jnp.ones((1, 1))
     out_trace = jnp.ones((1, 1,)) * 0.65
 
     ## check pre-synaptic STDP only
-    truth = jnp.array([[0.57342285]])
+    truth = jnp.array([[1.1031212]])
     ctx.reset()
     a.preSpike.set(in_spike * 0)
     a.preTrace.set(in_trace)
@@ -65,7 +66,7 @@ def test_expSTDPSynapse1():
     #print(a.dWeights.value)
     assert_array_equal(a.dWeights.value, truth)
 
-    truth = jnp.array([[-0.29817986]])
+    truth = jnp.array([[-0.57362294]])
     ctx.reset()
     a.preSpike.set(in_spike)
     a.preTrace.set(in_trace)
@@ -77,3 +78,4 @@ def test_expSTDPSynapse1():
     assert_array_equal(a.dWeights.value, truth)
 
 #test_expSTDPSynapse1()
+

tests/components/synapses/hebbian/test_traceSTDPSynapse.py

@@ -47,6 +47,7 @@ def test_traceSTDPSynapse1():
         evolve_cmd, evolve_args = ctx.compile_by_key(a, compile_key="evolve")
         ctx.add_command(wrap_command(jit(ctx.evolve)), name="adapt")
         """
+    a.weights.set(jnp.ones((1, 1)) * 0.1)
 
     in_spike = jnp.ones((1, 1))
     in_trace = jnp.ones((1, 1,)) * 1.25