merge

Author: jihan-yin

tests/helpers.py

@@ -4,6 +4,7 @@
 TEST_MODEL_REFERENCE = '[PyTest] Test Model Reference'
 TEST_MODEL_RUN = '[PyTest] Test Model Run'
 TEST_DATASET_NAME = '[PyTest] Test Dataset'
+TEST_SLICE_NAME = '[PyTest] Test Slice'
 TEST_IMG_URLS = [
     's3://scaleapi-attachments/BDD/BDD/bdd100k/images/100k/train/6dd63871-831611a6.jpg',
     's3://scaleapi-attachments/BDD/BDD/bdd100k/images/100k/train/82c1005c-e2d1d94f.jpg',

tests/test_dataset.py

@@ -1,6 +1,7 @@
 import pytest
 
 from helpers import (
+    TEST_SLICE_NAME,
     TEST_DATASET_NAME,
     TEST_IMG_URLS,
     reference_id_from_url,
@@ -63,7 +64,7 @@ def check_is_expected_response(response):
         ds_items_with_metadata.append(
             DatasetItem(
                 image_location=url,
-                reference_id=Path(url).name,
+                reference_id=reference_id_from_url(url),
                 metadata={
                     "made_with_pytest": True,
                     "example_int": i,
@@ -93,7 +94,7 @@ def test_slice_create_and_delete(dataset):
     for url in TEST_IMG_URLS:
         ds_items.append(DatasetItem(
             image_location=url,
-            reference_id=Path(url).name,
+            reference_id=reference_id_from_url(url),
         ))
     response = dataset.append(ds_items)
     assert ERROR_PAYLOAD not in response.json()
@@ -122,7 +123,7 @@ def test_slice_append(dataset):
     for url in TEST_IMG_URLS:
         ds_items.append(DatasetItem(
             image_location=url,
-            reference_id=Path(url).name,
+            reference_id=reference_id_from_url(url),
         ))
     response = dataset.append(ds_items)
     assert ERROR_PAYLOAD not in response.json()