add more test

Author: Captainia

tests/unit/sagemaker/jumpstart/test_notebook_utils.py

@@ -25,6 +25,7 @@
     list_jumpstart_models,
     list_jumpstart_scripts,
     list_jumpstart_tasks,
+    _is_valid_version,
 )
 
 
@@ -185,6 +186,13 @@ def test_list_jumpstart_frameworks(
     patched_get_model_specs.assert_not_called()
 
 
+def test_is_valid_version():
+    valid_version_strs = ["1.0", "1.0.0", "2012.4", "1!1.0", "1.dev0", "1.2.3+abc.dev1"]
+    invalid_version_strs = ["1.1.053_m", "invalid version", "v1-1.0-v2"]
+    assert all(_is_valid_version(v) for v in valid_version_strs)
+    assert not any(_is_valid_version(v) for v in invalid_version_strs)
+
+
 class ListJumpStartModels(TestCase):
     @patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor._get_manifest")
     @patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor.get_model_specs")