adding test for n_splits > 0

MouseLand · Aug 5, 2023 · 9fc08c9 · 9fc08c9
1 parent 01b2fa1
commit 9fc08c9
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_rastermap.py b/tests/test_rastermap.py
@@ -8,6 +8,17 @@ def test_rastermap(test_file):
 
     model = Rastermap().fit(data=spks)
 
+    assert hasattr(model, "embedding")
+    assert hasattr(model, "isort")
+    assert hasattr(model, "Usv")
+    assert hasattr(model, "Vsv")
+
+def test_rastermap_splits(test_file):
+    dat = np.load(test_file)
+    spks = dat["spks"]
+
+    model = Rastermap(n_splits=2, n_clusters=20, nc_splits=10).fit(data=spks)
+
     assert hasattr(model, "embedding")
     assert hasattr(model, "isort")
     assert hasattr(model, "Usv")