Digital Drosophila Archive
60 lines
2.8 KiB
Python
60 lines
2.8 KiB
Python
"""Independent fixtures; no access to the MaleCNS raw data or experiment outputs."""
|
|
import unittest
|
|
import numpy as np
|
|
import pandas as pd
|
|
from math_reference import params, state, single, cpu_motif
|
|
|
|
|
|
class Operators(unittest.TestCase):
|
|
def setUp(self):
|
|
self.p = params(dict(dt_ms=.1, synaptic_tau_ms=5., membrane_tau_ms=20.,
|
|
resting_potential_mv=-52., reset_potential_mv=-52., threshold_mv=-45.,
|
|
refractory_ms=2.2, synaptic_delay_ms=1.8))
|
|
|
|
def test_zero_input_and_exact_repeat(self):
|
|
for policy in ('frozen','reset_g_only','refractory_gate_only','reset_and_refractory'):
|
|
a, h = single(np.zeros(1000,np.float32),self.p,policy)
|
|
self.assertEqual(len(a),0)
|
|
np.testing.assert_array_equal(h[:,0],np.full(1000,-52,np.float32))
|
|
self.assertTrue(np.isfinite(h).all())
|
|
np.testing.assert_array_equal(h,single(np.zeros(1000,np.float32),self.p,policy)[1])
|
|
|
|
def test_single_spike_reset(self):
|
|
a = np.zeros(1000,np.float32); a[0] = 2000
|
|
for policy in ('reset_g_only','reset_and_refractory'):
|
|
ev, h = single(a,self.p,policy)
|
|
np.testing.assert_array_equal(ev,[0]); self.assertEqual(h[0,1],0)
|
|
|
|
def test_refractory_release_at_22(self):
|
|
a = np.zeros(30,np.float32); a[0] = 2000; a[1:23] = 10
|
|
ev, h = single(a,self.p,'reset_and_refractory')
|
|
self.assertEqual(ev[0],0)
|
|
np.testing.assert_array_equal(h[:22,1],np.zeros(22))
|
|
self.assertEqual(h[21,2],1); self.assertEqual(h[22,2],0)
|
|
self.assertEqual(h[22,1],10)
|
|
|
|
def test_inhibitory_arrival_sign_preserved(self):
|
|
v,g,r,f = state(np.array([-52],np.float32),np.zeros(1,np.float32),
|
|
np.zeros(1,np.int32),np.array([-10],np.float32),self.p)
|
|
self.assertLess(v[0],-52); self.assertEqual(g[0],-10); self.assertFalse(f[0])
|
|
|
|
def test_exact_delay_and_no_duplicate_count(self):
|
|
edges = pd.DataFrame(dict(edge_id=[17],source_neuron_idx=[100],target_neuron_idx=[200]))
|
|
r = cpu_motif(np.array([100,200],np.int32),edges,np.array([1],np.float32),self.p,40,100)
|
|
np.testing.assert_array_equal(r['step'],[0]); np.testing.assert_array_equal(r['counts'],[1,0])
|
|
self.assertTrue((r['history'][:18,1,1] == 0).all())
|
|
self.assertEqual(r['history'][18,1,1],1)
|
|
np.testing.assert_array_equal(r['queue'],np.zeros((19,2),np.float32))
|
|
|
|
def test_ref_gate_is_not_identical_to_reset(self):
|
|
a = np.zeros(60,np.float32); a[0] = 2000
|
|
_, frozen = single(a,self.p,'frozen')
|
|
_, gate = single(a,self.p,'refractory_gate_only')
|
|
self.assertLess(frozen[1,1],frozen[0,1]); self.assertEqual(gate[1,1],gate[0,1])
|
|
|
|
def test_unknown_policy_rejected(self):
|
|
with self.assertRaises(ValueError): single(np.zeros(1,np.float32),self.p,'guess')
|
|
|
|
|
|
if __name__ == '__main__': unittest.main(verbosity=2)
|