From 7b31c2df3a0bdd2a7f6577fa9ce33506139d8d98 Mon Sep 17 00:00:00 2001 From: John Bradley Date: Thu, 18 Jan 2024 10:22:01 -0500 Subject: [PATCH] Fix failing test --- andromeda/tests/test_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/andromeda/tests/test_dataset.py b/andromeda/tests/test_dataset.py index 14b31e1..035d85b 100644 --- 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"], "")