Fix failing test

Imageomics · Jan 18, 2024 · 7b31c2d · 7b31c2d
1 parent e6ac939
commit 7b31c2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/andromeda/tests/test_dataset.py b/andromeda/tests/test_dataset.py
@@ -53,7 +53,7 @@ def fake_get_path():
 
         self.assertEqual(weights.keys(), set(["R1", "B1", "G1"]))
         self.assertEqual(len(image_coordinates), 3)
-        self.assertEqual(image_coordinates[0].keys(), set(["x", "y", "label", "url"]))
+        self.assertEqual(image_coordinates[0].keys(), set(["x", "y", "label", "url", "R1", "B1", "G1"]))
         self.assertEqual(image_coordinates[0]["label"], "p1")
         self.assertEqual(image_coordinates[0]["url"], "https://example.com/p1.jpg")
         self.assertEqual(image_coordinates[1]["url"], "")