change: add image_uris_unit_test pytest mark (#4211)

Author: benieric

tests/conftest.py

@@ -14,6 +14,7 @@
 
 import json
 import os
+import pathlib
 
 import boto3
 import pytest
@@ -86,6 +87,16 @@
 PYTORCH_RENEWED_GPU = "ml.g4dn.xlarge"
 
 
+image_uris_unit_tests_dir = pathlib.Path("tests/unit/sagemaker/image_uris")
+
+
+def pytest_collection_modifyitems(config, items):
+    for item in items:
+        testmod = pathlib.Path(item.fspath)
+        if config.rootdir / image_uris_unit_tests_dir in testmod.parents:
+            item.add_marker(pytest.mark.image_uris_unit_test)
+
+
 def pytest_addoption(parser):
     parser.addoption("--sagemaker-client-config", action="store", default=None)
     parser.addoption("--sagemaker-runtime-config", action="store", default=None)

tox.ini

@@ -59,6 +59,7 @@ markers =
     local_mode
     slow_test
     release
+    image_uris_unit_test
     timeout: mark a test as a timeout.
 
 [testenv]