add unit test related to versioning fix

Author: Narrohag

tests/unit/sagemaker/jumpstart/constants.py

@@ -15990,6 +15990,18 @@
         "spec_key": "community_models_specs/tensorflow-ic-"
         "imagenet-inception-v3-classification-4/specs_v3.0.0.json",
     },
+    {
+        "model_id": "meta-textgeneration-llama-2-7b",
+        "version": "4.9.0",
+        "min_version": "2.49.0",
+        "spec_key": "community_models/meta-textgeneration-llama-2-7b/specs_v4.9.0.json",
+    },
+    {
+        "model_id": "meta-textgeneration-llama-2-7b",
+        "version": "4.13.0",
+        "min_version": "2.49.0",
+        "spec_key": "community_models/meta-textgeneration-llama-2-7b/specs_v4.13.0.json",
+    },
 ]
 
 BASE_PROPRIETARY_HEADER = {

tests/unit/sagemaker/jumpstart/test_cache.py

@@ -184,6 +184,30 @@ def test_jumpstart_cache_get_header():
         semantic_version_str="1.0.*",
     )
 
+    assert JumpStartModelHeader(
+        {
+            "model_id": "meta-textgeneration-llama-2-7b",
+            "version": "4.13.0",
+            "min_version": "2.49.0",
+            "spec_key": "community_models/meta-textgeneration-llama-2-7b/specs_v4.13.0.json",
+        }
+    ) == cache.get_header(
+        model_id="meta-textgeneration-llama-2-7b",
+        semantic_version_str="*",
+    )
+
+    assert JumpStartModelHeader(
+        {
+            "model_id": "meta-textgeneration-llama-2-7b",
+            "version": "4.13.0",
+            "min_version": "2.49.0",
+            "spec_key": "community_models/meta-textgeneration-llama-2-7b/specs_v4.13.0.json",
+        }
+    ) == cache.get_header(
+        model_id="meta-textgeneration-llama-2-7b",
+        semantic_version_str="4.*",
+    )
+
     assert JumpStartModelHeader(
         {
             "model_id": "ai21-summarization",