Skip to content

Commit 5304a0c

Browse files
lu-ohailu-ohai
committed
Updated unit tests.
1 parent 0f0e302 commit 5304a0c

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

‎tests/unitary/with_extras/aqua/test_deployment.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ class TestDataset:
3939
MODEL_DEPLOYMENT_ID = "ocid1.datasciencemodeldeployment.oc1.<region>.<MD_OCID>"
4040
MODEL_DEPLOYMENT_URL = "https://modeldeployment.customer-oci.com/ocid1.datasciencemodeldeployment.oc1.<region>.<MD_OCID>"
4141
MODEL_ID = "ocid1.datasciencemodeldeployment.oc1.<region>.<MODEL_OCID>"
42+
PRIVATE_ENDPOINT_ID = "ocid1.datascienceprivateendpoint.oc1.<region>.<PRIVATE_ENDPOINT_OCID>"
4243
DEPLOYMENT_IMAGE_NAME = "dsmc://image-name:1.0.0.0"
4344
DEPLOYMENT_SHAPE_NAME = "VM.GPU.A10.1"
4445
DEPLOYMENT_SHAPE_NAME_CPU = "VM.Standard.A1.Flex"
@@ -95,6 +96,7 @@ class TestDataset:
9596
"instance_configuration": oci.data_science.models.InstanceConfiguration(
9697
**{
9798
"instance_shape_name": DEPLOYMENT_SHAPE_NAME,
99+
"private_endpoint_id": PRIVATE_ENDPOINT_ID,
98100
"model_deployment_instance_shape_config_details": null,
99101
}
100102
),
@@ -149,6 +151,7 @@ class TestDataset:
149151
"instance_configuration": oci.data_science.models.InstanceConfiguration(
150152
**{
151153
"instance_shape_name": DEPLOYMENT_SHAPE_NAME_CPU,
154+
"private_endpoint_id": PRIVATE_ENDPOINT_ID,
152155
"model_deployment_instance_shape_config_details": oci.data_science.models.ModelDeploymentInstanceShapeConfigDetails(
153156
**{
154157
"ocpus": 10,
@@ -181,6 +184,7 @@ class TestDataset:
181184
"created_on": "2024-01-01T00:00:00.000000+00:00",
182185
"created_by": "ocid1.user.oc1..<OCID>",
183186
"endpoint": MODEL_DEPLOYMENT_URL,
187+
"private_endpoint_id": PRIVATE_ENDPOINT_ID,
184188
"environment_variables": {
185189
"BASE_MODEL": "service_models/model-name/artifact",
186190
"MODEL_DEPLOY_ENABLE_STREAMING": "true",

‎tests/unitary/with_extras/model/test_generic_model.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1514,6 +1514,7 @@ def test__to_yaml(self):
15141514
"description": None,
15151515
"deployment_instance_shape": None,
15161516
"deployment_instance_subnet_id": None,
1517+
"deployment_instance_private_endpoint_id": None,
15171518
"deployment_instance_count": None,
15181519
"deployment_bandwidth_mbps": None,
15191520
"deployment_log_group_id": None,
@@ -1554,6 +1555,7 @@ def test__to_yaml(self):
15541555
"deployment_description": None,
15551556
"deployment_instance_shape": None,
15561557
"deployment_instance_subnet_id": None,
1558+
"deployment_instance_private_endpoint_id": None,
15571559
"deployment_instance_count": None,
15581560
"deployment_bandwidth_mbps": None,
15591561
"deployment_log_group_id": None,
@@ -1606,6 +1608,7 @@ def test__to_yaml(self):
16061608
"description": None,
16071609
"deployment_instance_shape": None,
16081610
"deployment_instance_subnet_id": None,
1611+
"deployment_instance_private_endpoint_id": None,
16091612
"deployment_instance_count": None,
16101613
"deployment_bandwidth_mbps": None,
16111614
"deployment_log_group_id": None,
@@ -1646,6 +1649,7 @@ def test__to_yaml(self):
16461649
"deployment_description": "fake_deployment_description",
16471650
"deployment_instance_shape": "2.1",
16481651
"deployment_instance_subnet_id": "ocid1.subnet.oc1.iad.<unique_ocid>",
1652+
"deployment_instance_private_endpoint_id": "ocid1.datascienceprivateendpointint.oc1.iad.<unique_ocid>",
16491653
"deployment_instance_count": 1,
16501654
"deployment_bandwidth_mbps": 10,
16511655
"deployment_log_group_id": "ocid1.loggroup.oc1.iad.<unique_ocid>",
@@ -1704,6 +1708,7 @@ def test__to_yaml(self):
17041708
"deployment_instance_count": 1,
17051709
"deployment_bandwidth_mbps": 10,
17061710
"deployment_instance_subnet_id": "ocid1.subnet.oc1.iad.<unique_ocid>",
1711+
"deployment_instance_private_endpoint_id": "ocid1.datascienceprivateendpointint.oc1.iad.<unique_ocid>",
17071712
"deployment_log_group_id": "ocid1.loggroup.oc1.iad.<unique_ocid>",
17081713
"deployment_access_log_id": "ocid1.log.oc1.iad.<unique_ocid>",
17091714
"deployment_predict_log_id": "ocid1.log.oc1.iad.<unique_ocid>",
@@ -1746,6 +1751,7 @@ def test__to_yaml(self):
17461751
"deployment_description": "fake_deployment_description",
17471752
"deployment_instance_shape": "2.1",
17481753
"deployment_instance_subnet_id": "ocid1.subnet.oc1.iad.<unique_ocid>",
1754+
"deployment_instance_private_endpoint_id": "ocid1.datascienceprivateendpointint.oc1.iad.<unique_ocid>",
17491755
"deployment_instance_count": 1,
17501756
"deployment_bandwidth_mbps": 10,
17511757
"deployment_log_group_id": "ocid",
@@ -1810,6 +1816,7 @@ def test__to_yaml(self):
18101816
"deployment_instance_count": 1,
18111817
"deployment_bandwidth_mbps": 10,
18121818
"deployment_instance_subnet_id": "ocid1.subnet.oc1.iad.<unique_ocid>",
1819+
"deployment_instance_private_endpoint_id": "ocid1.datascienceprivateendpointint.oc1.iad.<unique_ocid>",
18131820
"deployment_log_group_id": "ocid",
18141821
"deployment_access_log_id": "ocid",
18151822
"deployment_predict_log_id": "ocid",
@@ -1882,6 +1889,7 @@ def test_prepare_save_deploy_with_default_display_name(
18821889
"description": None,
18831890
"deployment_instance_shape": None,
18841891
"deployment_instance_subnet_id": None,
1892+
"deployment_instance_private_endpoint_id": None,
18851893
"deployment_instance_count": None,
18861894
"deployment_bandwidth_mbps": None,
18871895
"deployment_memory_in_gbs": None,

0 commit comments

Comments
 (0)