Audit APL PN scope and localize official synaptic contacts without changing frozen dynamics

2026-09-29 23:58:05 +08:00
parent 3981628ff8
commit d6843e675d
62 changed files with 3974 additions and 0 deletions
@@ -0,0 +1,29 @@
+import tempfile
+from pathlib import Path
+import unittest
+import numpy as np
+import pandas as pd
+import pyarrow as pa
+import pyarrow.feather as feather
+from audit import raw_incident,population,save
+
+class AuditTests(unittest.TestCase):
+    def test_raw_incident_union_keeps_self_loop_once(self):
+        with tempfile.TemporaryDirectory(prefix='apl-scope-test-') as d:
+            p=Path(d)/'raw.feather'
+            frame=pd.DataFrame({'body_pre':[10,10,20,40],'body_post':[10,30,10,50],'weight':[1,2,3,4]},dtype=np.int64)
+            feather.write_feather(pa.Table.from_pandas(frame,preserve_index=False),p,chunksize=2)
+            result=raw_incident(p,np.array([10],np.uint64))
+            self.assertTrue(np.array_equal(result.to_numpy(),frame.iloc[:3].to_numpy()))
+            with self.assertRaises(AssertionError):
+                feather.write_feather(pa.table({'body_pre':[10],'wrong':[30],'weight':[1]}),p)
+                raw_incident(p,np.array([10],np.uint64))
+    def test_population_does_not_equate_dopamine_to_DAN(self):
+        m=pd.DataFrame({'annotation_class':[None,None,'ALPN','DAN'], 'annotation_type':['APL','DPM','VA6_adPN','PAM01']})
+        self.assertEqual(list(population(m)),['APL','DPM','ALPN','DAN'])
+    def test_exclusive_output(self):
+        with tempfile.TemporaryDirectory(prefix='apl-scope-test-') as d:
+            p=Path(d)/'x.json';save(p,{'original':True})
+            with self.assertRaises(FileExistsError):save(p,{'original':False})
+
+if __name__=='__main__':unittest.main()