changes to xcorr and begining of unit tests

bouromain · Oct 17, 2023 · b89f080 · b89f080
1 parent 45012ed
commit b89f080
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_sigtool.py b/tests/test_sigtool.py
@@ -0,0 +1,14 @@
+# %% Import
+import warnings
+from hippocampy import sig_tool
+import unittest
+import numpy as np
+
+
+# %%
+class TestLabel(unittest.TestCase):
+    def test_xcorr(self):
+        # test boolean vector
+        x = np.array([0, 0, 0, 1, 1, 1, 0, 1, 0, 0, 1])
+        out = sig_tool.xcorr(a)
+        assert all(out == np.array([1, 0, 2, 2, 0, 3]))