integration test for image indexing

Author: jihan-yin

nucleus/dataset.py

@@ -380,7 +380,8 @@ def set_continuous_indexing(self, enable: bool = True):
         return self._client.set_continuous_indexing(self.id, enable)
 
     def create_image_index(self):
-        return self._client.create_image_index(self.id)
+        response = self._client.create_image_index(self.id)
+        return AsyncJob.from_json(response, self._client)
 
     def check_index_status(self, job_id: str):
         return self._client.check_index_status(job_id)

tests/test_indexing.py

@@ -57,3 +57,10 @@ def test_index_integration(dataset):
     assert STATUS_KEY in job_status_response
     assert JOB_ID_KEY in job_status_response
     assert MESSAGE_KEY in job_status_response
+
+
+def test_generate_image_index_integration(dataset):
+    job = dataset.create_image_index()
+    job.sleep_until_complete()
+    job.status()
+    assert job.job_last_known_status == "Completed"