Audit conditional PN recruitment and graded APL source timing

models/male-cns-lif/pn-recruitment-audit-v1/test_harness.py

@@ -0,0 +1,36 @@
+import tempfile
+import unittest
+from pathlib import Path
+import numpy as np
+import pyarrow.parquet as pq
+from run_audit import first_ticks, temporal_relation, min_observed, EventSink
+
+
+class HarnessTests(unittest.TestCase):
+    def test_first_tick_uses_minimum_not_order(self):
+        self.assertEqual(first_ticks(np.array([4, 2, 3], np.int32), np.array([7, 7, 8], np.int32),
+                                    np.array([7, 8, 9], np.int32)).tolist(), [2, 3, -1])
+
+    def test_strict_temporal_boundary(self):
+        self.assertEqual(temporal_relation(17, 18), 'before')
+        self.assertEqual(temporal_relation(18, 18), 'same_tick')
+        self.assertEqual(temporal_relation(19, 18), 'after')
+        self.assertEqual(temporal_relation(-1, 18), 'unobserved_source')
+        self.assertEqual(temporal_relation(18, -1), 'unobserved_target')
+        self.assertEqual(temporal_relation(-1, -1), 'both_unobserved')
+
+    def test_min_observed(self):
+        self.assertEqual(min_observed([-1, 0, 3]), 0)
+        self.assertEqual(min_observed([-1, -1]), -1)
+
+    def test_empty_event_sink_has_schema(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            path = Path(tmp) / 'events.parquet'
+            sink = EventSink(path)
+            sink.close()
+            self.assertEqual(pq.read_table(path).num_rows, 0)
+            self.assertIn('source_spike_tick', pq.read_schema(path).names)
+
+
+if __name__ == '__main__':
+    unittest.main()