chore: updates so tests can run in different environments (#1403)

googleapis · Jun 1, 2022 · 15fe100 · 15fe100
1 parent 3047e29
commit 15fe100
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf1_test.py b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf1_test.py
@@ -26,6 +26,7 @@
 from test_models import upload_model_mock, get_model_mock  # noqa: F401
 
 
+@pytest.mark.usefixtures("google_auth_mock")
 class SavedModelMetadataBuilderTF1Test(tf.test.TestCase):
     def _set_up(self):
         self.sess = tf.Session(graph=tf.Graph())

diff --git a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py
@@ -27,6 +27,7 @@
 from test_models import upload_model_mock, get_model_mock  # noqa: F401
 
 
+@pytest.mark.usefixtures("google_auth_mock")
 class SavedModelMetadataBuilderTF2Test(tf.test.TestCase):
     def _set_up_sequential(self):
         # Set up for the sequential.

diff --git a/tests/unit/aiplatform/test_training_jobs.py b/tests/unit/aiplatform/test_training_jobs.py
@@ -330,6 +330,9 @@ def mock_get_backing_custom_job_with_enable_web_access():
         yield get_custom_job_mock
 
 
+@pytest.mark.skipif(
+    sys.executable is None, reason="requires python path to invoke subprocess"
+)
 @pytest.mark.usefixtures("google_auth_mock")
 class TestTrainingScriptPythonPackagerHelpers:
     def setup_method(self):
@@ -446,6 +449,9 @@ def test_get_python_executable_returns_python_executable(self):
         assert "python" in source_utils._get_python_executable().lower()
 
 
+@pytest.mark.skipif(
+    sys.executable is None, reason="requires python path to invoke subprocess"
+)
 @pytest.mark.usefixtures("google_auth_mock")
 class TestTrainingScriptPythonPackager:
     def setup_method(self):