From ba31441ac504f44748016ba630112d5d16f3d472 Mon Sep 17 00:00:00 2001
From: kshtiijrajsharma <skshitizraj@gmail.com>
Date: Mon, 25 Mar 2024 09:18:43 +0545
Subject: [PATCH] (unit-test) test_predict: add dict assert and features

---
 tests/test_predict.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_predict.py b/tests/test_predict.py
index db80ce3..db25f1a 100644
--- 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__":