|
17 | 17 | import copy
|
18 | 18 |
|
19 | 19 | import humps
|
20 |
| -import pytest |
21 | 20 | from hsml import model
|
22 | 21 | from hsml.constants import MODEL
|
23 |
| -from hsml.core import explicit_provenance |
24 | 22 |
|
25 | 23 |
|
26 | 24 | class TestModel:
|
@@ -343,94 +341,7 @@ class ClientMock:
|
343 | 341 | mock_client_get_instance.assert_called_once()
|
344 | 342 | mock_util_get_hostname_replaced_url.assert_called_once_with(sub_path=path_arg)
|
345 | 343 |
|
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 |
| - |
432 | 344 | # auxiliary methods
|
433 |
| - |
434 | 345 | def assert_model(self, mocker, m, m_json, model_framework):
|
435 | 346 | assert isinstance(m, model.Model)
|
436 | 347 | assert m.id == m_json["id"]
|
|
0 commit comments