test fixes + constants

Author: jihan-yin

nucleus/constants.py

@@ -92,8 +92,7 @@
 MODEL_RUN_ID_KEY = "model_run_id"
 NAME_KEY = "name"
 NEW_ITEMS = "new_items"
-# NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
-NUCLEUS_ENDPOINT = "https://jihan-yin-api.devbox.internal.scale.com/v1/nucleus"
+NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
 NUM_SENSORS_KEY = "num_sensors"
 ORIGINAL_IMAGE_URL_KEY = "original_image_url"
 PAGE_SIZE = "pageSize"

tests/test_dataset.py

@@ -562,17 +562,22 @@ def test_dataset_item_iterator(dataset):
 def test_dataset_get_image_indexing_status(CLIENT):
     dataset = Dataset(DATASET_WITH_EMBEDDINGS, CLIENT)
     resp = dataset.get_image_indexing_status()
-    assert resp["embedding_count"]
-    assert resp["image_count"]
-    assert not resp["object_count"]
-    assert resp["percent_indexed"]
+    print(resp)
+    assert resp["embedding_count"] == 170
+    assert resp["image_count"] == 170
+    assert "object_count" not in resp
+    assert round(resp["percent_indexed"], 2) == round(
+        resp["image_count"] / resp["embedding_count"], 2
+    )
 
 
 @pytest.mark.integration
 def test_dataset_get_object_indexing_status(CLIENT):
     dataset = Dataset(DATASET_WITH_EMBEDDINGS, CLIENT)
     resp = dataset.get_object_indexing_status()
-    assert resp["embedding_count"]
-    assert resp["object_count"]
-    assert not resp["image_count"]
-    assert resp["percent_indexed"]
+    assert resp["embedding_count"] == 422
+    assert resp["object_count"] == 423
+    assert "image_count" not in resp
+    assert round(resp["percent_indexed"], 2) == round(
+        resp["object_count"] / resp["embedding_count"], 2
+    )