adding unit tests

Author: bencrabtree

tests/unit/sagemaker/jumpstart/constants.py

@@ -9813,7 +9813,7 @@
                 "ComponentNames": ["variant1"],
                 "BenchmarkMetrics": {
                     "ml.g5.12xlarge": [
-                        {"Name": "latency", "Unit": "sec", "Value": "0.19", "Concurrency": "1"},
+                        {"name": "latency", "unit": "sec", "value": "0.19", "concurrency": "1"},
                     ]
                 },
             },

tests/unit/sagemaker/jumpstart/hub/test_utils.py

@@ -264,6 +264,7 @@ def test_walk_and_apply_json():
             "CamelCaseObjectChildTwo": "value2",
         },
         "IgnoreMyChildren": {"ShouldNotBeTouchedOne": "const1", "ShouldNotBeTouchedTwo": "const2"},
+        "ShouldNotIgnoreMyChildren": {"NopeNope": "no"}
     }
 
     result = parser_utils.walk_and_apply_json(
@@ -279,4 +280,35 @@ def test_walk_and_apply_json():
             "ShouldNotBeTouchedOne": "const1",
             "ShouldNotBeTouchedTwo": "const2",
         },
+        "should_not_ignore_my_children": { "nope_nope": "no"}
+    }
+
+def test_walk_and_apply_json_no_stop():
+    test_json = {
+        "CamelCaseKey": "value",
+        "CamelCaseObjectKey": {
+            "CamelCaseObjectChildOne": "value1",
+            "CamelCaseObjectChildTwo": "value2",
+        },
+        "CamelCaseObjectListKey": {
+            "instance.ml.type.xlarge": [
+                {"ShouldChangeMe": "string"}
+            ]
+        }
+    }
+
+    result = parser_utils.walk_and_apply_json(
+        test_json, parser_utils.camel_to_snake
+    )
+    assert result == {
+        "camel_case_key": "value",
+        "camel_case_object_key": {
+            "camel_case_object_child_one": "value1",
+            "camel_case_object_child_two": "value2",
+        },
+        "camel_case_object_list_key": {
+            "instance.ml.type.xlarge": [
+                { "should_change_me": "string" }
+            ]
+        },
     }