@@ -74,7 +74,11 @@ class DSCJobTestCase(unittest.TestCase):
74
74
@property
75
75
def default_datascience_job (self ):
76
76
random .seed (self .random_seed )
77
- return DataScienceJob ().with_project_id (self .PROJECT_ID )
77
+ return (
78
+ DataScienceJob ()
79
+ .with_project_id (self .PROJECT_ID )
80
+ .with_subnet_id (self .SUBNET_ID )
81
+ )
78
82
79
83
@classmethod
80
84
def setUpClass (cls ) -> None :
@@ -632,7 +636,7 @@ def assert_infra_before_build(self, infra):
632
636
self .assertEqual (infra .project_id , self .PROJECT_ID )
633
637
self .assertEqual (infra .compartment_id , None )
634
638
self .assertEqual (infra .block_storage_size , None )
635
- self .assertEqual (infra .subnet_id , None )
639
+ self .assertEqual (infra .subnet_id , self . SUBNET_ID )
636
640
637
641
def test_build_job_within_notebook (self ):
638
642
job = (
@@ -666,4 +670,4 @@ def test_build_job_outside_notebook(self):
666
670
self .assertEqual (job .infrastructure .project_id , self .PROJECT_ID )
667
671
self .assertEqual (job .infrastructure .compartment_id , self .COMPARTMENT_ID )
668
672
self .assertEqual (job .infrastructure .block_storage_size , 50 )
669
- self .assertEqual (job .infrastructure .subnet_id , None )
673
+ self .assertEqual (job .infrastructure .subnet_id , self . SUBNET_ID )
0 commit comments