Fixed image_url vs. location bug and pytest bug

Author: ardila

nucleus/dataset_item.py

@@ -19,6 +19,7 @@ class DatasetItem:
     metadata: dict = None
 
     def __post_init__(self):
+        self.image_url = self.image_location
         self.local = self._is_local_path(self.image_location)
 
     @classmethod

tests/test_annotation.py

@@ -21,11 +21,12 @@
 )
 from nucleus.constants import ERROR_PAYLOAD
 
-# Have to define here in order to have access to all relevant objects
-def test_repr(test_object: any):
-    assert eval(str(test_object)) == test_object
+
 
 def test_reprs():
+    # Have to define here in order to have access to all relevant objects
+    def test_repr(test_object: any):
+        assert eval(str(test_object)) == test_object
     [
         test_repr(SegmentationAnnotation.from_json(_))
         for _ in TEST_SEGMENTATION_ANNOTATIONS

tests/test_dataset.py

@@ -19,11 +19,12 @@
 )
 
 
-# Have to define here in order to have access to all relevant objects
-def test_repr(test_object: any):
-    assert eval(str(test_object)) == test_object
+
 
 def test_reprs():
+    # Have to define here in order to have access to all relevant objects
+    def test_repr(test_object: any):
+        assert eval(str(test_object)) == test_object
     test_repr(DatasetItem(
                 image_location="test_url",
                 reference_id="test_reference_id",

tests/test_models.py

@@ -24,11 +24,12 @@
     TEST_PREDS,
 )
 
-# Have to define here in order to have access to all relevant objects
-def test_repr(test_object: any):
-    assert eval(str(test_object)) == test_object
+
 
 def test_reprs():
+    # Have to define here in order to have access to all relevant objects
+    def test_repr(test_object: any):
+        assert eval(str(test_object)) == test_object
     client = NucleusClient(api_key="fake_key")
     test_repr(Model(client=client, model_id="fake_model_id", name="fake_name", reference_id="fake_reference_id", metadata={"fake": "metadata"}))
     test_repr(ModelRun(client=client, model_run_id="fake_model_run_id"))

tests/test_prediction.py

@@ -25,11 +25,12 @@
 from nucleus.constants import ERROR_PAYLOAD
 
 
-def test_repr(test_object: any):
-    assert eval(str(test_object)) == test_object
+
 
 
 def test_reprs():
+    def test_repr(test_object: any):
+        assert eval(str(test_object)) == test_object
     [
         test_repr(SegmentationPrediction.from_json(_))
         for _ in TEST_SEGMENTATION_PREDICTIONS

tests/test_slice.py

@@ -0,0 +1,14 @@
+import pytest
+from nucleus import (
+    Slice,
+    NucleusClient
+)
+
+
+
+def test_reprs():
+    # Have to define here in order to have access to all relevant objects
+    def test_repr(test_object: any):
+        assert eval(str(test_object)) == test_object
+    client = NucleusClient(api_key="fake_key")
+    test_repr(Slice(slice_id="fake_slice_id", client=client))

tests/test_upload_response.py

@@ -0,0 +1,17 @@
+import pytest
+from nucleus import (
+    Slice,
+    UploadResponse
+)
+
+
+from nucleus.constants import (
+    DATASET_ID_KEY,
+)
+
+
+def test_reprs():
+    # Have to define here in order to have access to all relevant objects
+    def test_repr(test_object: any):
+        assert eval(str(test_object)) == test_object
+    test_repr(UploadResponse(json = {DATASET_ID_KEY: "fake_dataset_id_key"}))