Fixed unit test.

Author: lu-ohai

tests/unitary/default_setup/model_deployment/test_model_deployment.py

@@ -28,8 +28,10 @@ class ModelDeploymentTestCase(unittest.TestCase):
  )
 
  @patch("requests.post")
- def test_predict(self, mock_post):
+ @patch("ads.model.deployment.model_deployment.ModelDeployment.sync")
+ def test_predict(self, mock_sync, mock_post):
  """Ensures predict model passes with valid input parameters."""
+ mock_sync.return_value = Mock(lifecycle_state="ACTIVE")
  mock_post.return_value = Mock(
  status_code=200, json=lambda: {"result": "result"}
  )
@@ -50,8 +52,10 @@ def test_predict(self, mock_post):
  self.test_model_deployment.predict(data=np.array([1, 2, 3]))
 
  @patch("requests.post")
- def test_predict_with_bytes(self, mock_post):
+ @patch("ads.model.deployment.model_deployment.ModelDeployment.sync")
+ def test_predict_with_bytes(self, mock_sync, mock_post):
  """Ensures predict model passes with bytes input."""
+ mock_sync.return_value = Mock(lifecycle_state="ACTIVE")
  byte_data = b"[[1,2,3,4]]"
  with patch.object(authutil, "default_signer") as mock_auth:
  auth = MagicMock()
@@ -66,8 +70,10 @@ def test_predict_with_bytes(self, mock_post):
  )
 
  @patch("requests.post")
- def test_predict_with_auto_serialize_data(self, mock_post):
+ @patch("ads.model.deployment.model_deployment.ModelDeployment.sync")
+ def test_predict_with_auto_serialize_data(self, mock_sync, mock_post):
  """Ensures predict model passes with valid input parameters."""
+ mock_sync.return_value = Mock(lifecycle_state="ACTIVE")
  mock_post.return_value = Mock(
  status_code=200, json=lambda: {"result": "result"}
  )