updated tests

Author: kozlov721

tests/test_data/test_parsers.py

@@ -4,6 +4,7 @@
 
 from luxonis_ml.data import LabelType, LuxonisLoader, LuxonisParser
 from luxonis_ml.enums import DatasetType
+from luxonis_ml.utils import environ
 
 URL_PREFIX: Final[str] = "gs://luxonis-test-bucket/luxonis-ml-test-data"
 WORK_DIR: Final[str] = "tests/data/parser_datasets"
@@ -82,13 +83,24 @@ def prepare_dir():
             "D1_ParkingSlot-solo.zip",
             [LabelType.BOUNDINGBOX, LabelType.SEGMENTATION],
         ),
+        (
+            DatasetType.COCO,
+            "roboflow://team-roboflow/coco-128/2/coco",
+            [LabelType.BOUNDINGBOX, LabelType.CLASSIFICATION],
+        ),
     ],
 )
 def test_dir_parser(
     dataset_type: DatasetType, url: str, expected_label_types: List[LabelType]
 ):
+    if not url.startswith("roboflow://"):
+        url = f"{URL_PREFIX}/{url}"
+
+    elif environ.ROBOFLOW_API_KEY is None:
+        pytest.skip("Roboflow API key is not set")
+
     parser = LuxonisParser(
-        f"{URL_PREFIX}/{url}",
+        url,
         dataset_name=f"test-{dataset_type}",
         delete_existing=True,
         save_dir=WORK_DIR,