ibl-neuropixel 1.9.2__tar.gz → 1.9.3__tar.gz

{ibl_neuropixel-1.9.2/src/ibl_neuropixel.egg-info → ibl_neuropixel-1.9.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ibl-neuropixel
-Version: 1.9.2
+Version: 1.9.3
 Summary: Collection of tools for Neuropixel 1.0 and 2.0 probes data
 Home-page: https://github.com/int-brain-lab/ibl-neuropixel
 Author: The International Brain Laboratory

{ibl_neuropixel-1.9.2 → ibl_neuropixel-1.9.3}/setup.py

@@ -8,7 +8,7 @@ with open("requirements.txt") as f:
 
 setuptools.setup(
     name="ibl-neuropixel",
-    version="1.9.2",
+    version="1.9.3",
     author="The International Brain Laboratory",
     description="Collection of tools for Neuropixel 1.0 and 2.0 probes data",
     long_description=long_description,

{ibl_neuropixel-1.9.2 → ibl_neuropixel-1.9.3/src/ibl_neuropixel.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ibl-neuropixel
-Version: 1.9.2
+Version: 1.9.3
 Summary: Collection of tools for Neuropixel 1.0 and 2.0 probes data
 Home-page: https://github.com/int-brain-lab/ibl-neuropixel
 Author: The International Brain Laboratory

{ibl_neuropixel-1.9.2 → ibl_neuropixel-1.9.3}/src/ibl_neuropixel.egg-info/SOURCES.txt

@@ -58,6 +58,7 @@ src/tests/integration/__init__.py
 src/tests/integration/csd_experiments.py
 src/tests/integration/test_destripe.py
 src/tests/unit/__init__.py
+src/tests/unit/test_cadzow.py
 src/tests/unit/test_ephys_np2.py
 src/tests/unit/test_neuropixel.py
 src/tests/unit/test_plots.py

{ibl_neuropixel-1.9.2 → ibl_neuropixel-1.9.3}/src/ibldsp/cadzow.py

@@ -17,8 +17,6 @@ for N-spatial dimensions.
 }
 """
 
-from functools import lru_cache
-
 import numpy as np
 import scipy.fft
 from iblutil.numerical import ismember2d
@@ -48,7 +46,6 @@ def traj_matrix_indices(n):
     return itraj
 
 
-@lru_cache(maxsize=24)
 def trajectory(x, y, dtype=np.complex128):
     """
     Computes the 2 spatial dimensions block-Toeplitz indices from x and y trace coordinates.

ibl_neuropixel-1.9.3/src/tests/unit/test_cadzow.py

@@ -0,0 +1,24 @@
+import unittest
+
+import numpy as np
+
+import ibldsp.cadzow
+import neuropixel
+
+
+class TestCadzow(unittest.TestCase):
+    def test_trajectory_matrixes_indices(self):
+        assert np.all(
+            ibldsp.cadzow.traj_matrix_indices(4) == np.array([[1, 0], [2, 1], [3, 2]])
+        )
+        assert np.all(
+            ibldsp.cadzow.traj_matrix_indices(3) == np.array([[1, 0], [2, 1]])
+        )
+
+    def test_trajectory(self):
+        th = neuropixel.trace_header(version=1)
+        tm, it, ic, trcount = ibldsp.cadzow.trajectory(
+            th["x"][:8], th["y"][:8], dtype=np.int32
+        )
+        # make sure tm can be indexed
+        np.testing.assert_array_equal(tm[it], 0)

{ibl_neuropixel-1.9.2 → ibl_neuropixel-1.9.3}/src/tests/unit/test_voltage.py

@@ -148,13 +148,3 @@ class TestSaturation(unittest.TestCase):
         saturation[45852:45865] = True
         df_sat = ibldsp.voltage.saturation_samples_to_intervals(saturation)
         self.assertEqual(81, np.sum(df_sat["stop_sample"] - df_sat["start_sample"]))
-
-
-class TestCadzow(unittest.TestCase):
-    def test_trajectory_matrixes(self):
-        assert np.all(
-            ibldsp.cadzow.traj_matrix_indices(4) == np.array([[1, 0], [2, 1], [3, 2]])
-        )
-        assert np.all(
-            ibldsp.cadzow.traj_matrix_indices(3) == np.array([[1, 0], [2, 1]])
-        )