Update test_catalog_notebook.py and test_catalog_project.py to reset NB_SESSION_COMPARTMENT_OCID.

Author: qiuosier

tests/unitary/default_setup/catalog/test_catalog_notebook.py

@@ -75,7 +75,9 @@ def setUpClass(cls) -> None:
             "NB_SESSION_COMPARTMENT_OCID"
         ] = "ocid1.compartment.oc1.<unique_ocid>"
         reload(ads.config)
-        reload(ads.catalog.notebook)
+        ads.catalog.notebook.NB_SESSION_COMPARTMENT_OCID = (
+            ads.config.NB_SESSION_COMPARTMENT_OCID
+        )
         # Initialize class properties after reloading
         with patch.object(auth, "default_signer"):
             with patch.object(oci_client, "OCIClientFactory"):
@@ -99,7 +101,9 @@ def setUpClass(cls) -> None:
     def tearDownClass(cls) -> None:
         os.environ.pop("NB_SESSION_COMPARTMENT_OCID", None)
         reload(ads.config)
-        reload(ads.catalog.notebook)
+        ads.catalog.notebook.NB_SESSION_COMPARTMENT_OCID = (
+            ads.config.NB_SESSION_COMPARTMENT_OCID
+        )
         return super().tearDownClass()
 
     @staticmethod

tests/unitary/default_setup/catalog/test_catalog_project.py

@@ -68,7 +68,7 @@ def setUpClass(cls) -> None:
             "NB_SESSION_COMPARTMENT_OCID"
         ] = "ocid1.compartment.oc1.<unique_ocid>"
         reload(ads.config)
-        reload(ads.catalog.project)
+        ads.catalog.project.NB_SESSION_COMPARTMENT_OCID = ads.config.NB_SESSION_COMPARTMENT_OCID
         # Initialize class properties after reloading
         with patch.object(auth, "default_signer"):
             with patch.object(oci_client, "OCIClientFactory"):
@@ -92,7 +92,7 @@ def setUpClass(cls) -> None:
     def tearDownClass(cls) -> None:
         os.environ.pop("NB_SESSION_COMPARTMENT_OCID", None)
         reload(ads.config)
-        reload(ads.catalog.project)
+        ads.catalog.project.NB_SESSION_COMPARTMENT_OCID = ads.config.NB_SESSION_COMPARTMENT_OCID
         return super().tearDownClass()
 
     @staticmethod