Update test.

Author: qiuosier

ads/jobs/builders/infrastructure/dsc_job.py

@@ -1601,6 +1601,18 @@ def _update_job_infra(self, dsc_job: DSCJob) -> DataScienceJob:
             if value:
                 dsc_job.job_infrastructure_configuration_details[camel_attr] = value
 
+        shape = dsc_job.job_infrastructure_configuration_details.get("shapeName", "")
+        if (
+            shape
+            and not str(shape).endswith("Flex")
+            and dsc_job.job_infrastructure_configuration_details.get(
+                "jobShapeConfigDetails"
+            )
+        ):
+            raise ValueError(
+                "Shape config is not required for non flex shape from user end."
+            )
+
         if dsc_job.job_infrastructure_configuration_details.get("subnetId"):
             dsc_job.job_infrastructure_configuration_details[
                 "jobInfrastructureType"
@@ -1648,7 +1660,9 @@ def init(self, **kwargs) -> DataScienceJob:
     def _config_multi_node(self, runtime: MultiNodeRuntime):
         """Configure the payload for multi-node job run."""
         infra_config: dict = self.dsc_job.job_infrastructure_configuration_details
-        job_config: models.DefaultJobConfigurationDetails = self.dsc_job.job_configuration_details
+        job_config: models.DefaultJobConfigurationDetails = (
+            self.dsc_job.job_configuration_details
+        )
         env_config = self.dsc_job.job_environment_configuration_details
         # For multi-node jobs,
         # the job_infrastructure_configuration_details and job_configuration_details