add more UTs

Author: gwang111

tests/unit/sagemaker/serve/builder/test_model_builder.py

@@ -18,6 +18,7 @@
 from pathlib import Path
 from copy import deepcopy
 
+from sagemaker.model import Model
 from sagemaker.serve import SchemaBuilder
 from sagemaker.serve.builder.model_builder import ModelBuilder
 from sagemaker.serve.mode.function_pointers import Mode
@@ -3300,6 +3301,19 @@ def test_optimize_sharding_with_override_for_js(
 
         assert not model._enable_network_isolation
 
+    def test_model_sharding_with_eni_fails(self):
+        test_model = Model(role="mock role")
+        test_model._is_sharded_model = True
+        test_model._enable_network_isolation = True
+        self.assertRaisesRegex(
+            ValueError,
+            (
+                "EnableNetworkIsolation cannot be set to True since "
+                "SageMaker Fast Model Loading of model requires network access."
+            ),
+            lambda: test_model.deploy(initial_instance_count=1, instance_type="ml.g5.24xlarge"),
+        )
+
 
 class TestModelBuilderOptimizeValidations(unittest.TestCase):