Debug test datasets

minzhou1900 · Feb 16, 2021 · f09988c · f09988c
1 parent c18aa23
commit f09988c
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
@@ -130,16 +130,17 @@ def test_load_emg(self):
 
     def test_load_optical_nerves(self):
         """Test that optical nerves belong to space of landmarks."""
-        data, labels = data_utils.load_optical_nerves()
+        data, labels, monkeys = data_utils.load_optical_nerves()
         result = data.shape
-        n_monkeys = 22
+        n_monkeys = 11
+        n_eyes_per_monkey = 2
         k_landmarks = 5
         dim = 3
-        expected = (n_monkeys, k_landmarks, dim)
+        expected = (n_monkeys * n_eyes_per_monkey, k_landmarks, dim)
         self.assertAllClose(result, expected)
 
         landmarks_space = Landmarks(
-            ambient_manifold=Euclidean(dim=3), k_landmarks=k_landmarks
+            ambient_manifold=Euclidean(dim=dim), k_landmarks=k_landmarks
         )
 
         result = landmarks_space.belongs(data)