Skip to content

Commit 180538d

Browse files
Alexandru OrmenisanAlexandru Ormenisan
Alexandru Ormenisan
authored and
Alexandru Ormenisan
committed
remove tests
1 parent 60485b8 commit 180538d

File tree

1 file changed

+0
-89
lines changed

1 file changed

+0
-89
lines changed

python/tests/test_model.py

-89
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,8 @@
1717
import copy
1818

1919
import humps
20-
import pytest
2120
from hsml import model
2221
from hsml.constants import MODEL
23-
from hsml.core import explicit_provenance
2422

2523

2624
class TestModel:
@@ -343,94 +341,7 @@ class ClientMock:
343341
mock_client_get_instance.assert_called_once()
344342
mock_util_get_hostname_replaced_url.assert_called_once_with(sub_path=path_arg)
345343

346-
# get feature view
347-
348-
def test_get_feature_view_none(self, mocker, backend_fixtures):
349-
# Arrange
350-
m_json = backend_fixtures["model"]["get_python"]["response"]["items"][0]
351-
mock_get_feature_view_provenance = mocker.patch(
352-
"hsml.model.Model.get_feature_view_provenance", return_value=None
353-
)
354-
355-
# Act
356-
m = model.Model.from_response_json(m_json)
357-
fv = m.get_feature_view()
358-
359-
# Assert
360-
assert fv is None
361-
mock_get_feature_view_provenance.assert_called_once()
362-
363-
def test_get_feature_view(self, mocker, backend_fixtures):
364-
# Arrange
365-
m_json = backend_fixtures["model"]["get_python"]["response"]["items"][0]
366-
mock_fv = mocker.MagicMock()
367-
mock_get_feature_view_provenance = mocker.patch(
368-
"hsml.model.Model.get_feature_view_provenance", return_value=mock_fv
369-
)
370-
371-
# Act
372-
m = model.Model.from_response_json(m_json)
373-
fv = m.get_feature_view()
374-
375-
# Assert
376-
assert fv == mock_fv
377-
mock_get_feature_view_provenance.assert_called_once()
378-
379-
def test_get_feature_view_prov_artifact(self, mocker, backend_fixtures):
380-
# Arrange
381-
m_json = backend_fixtures["model"]["get_python"]["response"]["items"][0]
382-
mock_fv = mocker.MagicMock(spec=explicit_provenance.Artifact)
383-
mock_fv.meta_type = "meta_type"
384-
mocker.patch(
385-
"hsml.model.Model.get_feature_view_provenance", return_value=mock_fv
386-
)
387-
388-
# Act
389-
m = model.Model.from_response_json(m_json)
390-
with pytest.raises(Exception) as e_info:
391-
_ = m.get_feature_view()
392-
393-
# Assert
394-
assert "The returned object is not a valid feature view" in str(e_info.value)
395-
396-
# get feature view provenance
397-
398-
def test_get_feature_view_provenance(self, mocker, backend_fixtures):
399-
# Arrange
400-
m_json = backend_fixtures["model"]["get_python"]["response"]["items"][0]
401-
mock_model_engine_get_feature_view_provenance = mocker.patch(
402-
"hsml.engine.model_engine.ModelEngine.get_feature_view_provenance"
403-
)
404-
405-
# Act
406-
m = model.Model.from_response_json(m_json)
407-
m.get_feature_view_provenance()
408-
409-
# Assert
410-
mock_model_engine_get_feature_view_provenance.assert_called_once_with(
411-
model_instance=m
412-
)
413-
414-
# get training dataset provenance
415-
416-
def test_get_training_dataset_provenance(self, mocker, backend_fixtures):
417-
# Arrange
418-
m_json = backend_fixtures["model"]["get_python"]["response"]["items"][0]
419-
mock_model_engine_get_training_dataset_provenance = mocker.patch(
420-
"hsml.engine.model_engine.ModelEngine.get_training_dataset_provenance"
421-
)
422-
423-
# Act
424-
m = model.Model.from_response_json(m_json)
425-
m.get_training_dataset_provenance()
426-
427-
# Assert
428-
mock_model_engine_get_training_dataset_provenance.assert_called_once_with(
429-
model_instance=m
430-
)
431-
432344
# auxiliary methods
433-
434345
def assert_model(self, mocker, m, m_json, model_framework):
435346
assert isinstance(m, model.Model)
436347
assert m.id == m_json["id"]

0 commit comments

Comments
 (0)