(unit-test) test_predict: add dict assert and features

hotosm · Mar 25, 2024 · ba31441 · ba31441
1 parent a0d1132
commit ba31441
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_predict.py b/tests/test_predict.py
@@ -34,8 +34,8 @@ def tearDown(self):
     def test_predict(self):
         zoom_level = 20
         predictions = predict(BBOX, self.model_path, zoom_level, TMS_URL)
-        self.assertIsInstance(predictions, list)
-        self.assertTrue(len(predictions) > 0)
+        self.assertIsInstance(predictions, dict)
+        self.assertTrue(len(predictions["features"]) > 0)
 
 
 if __name__ == "__main__":