20 lines
828 B
Python
20 lines
828 B
Python
import numpy as np
|
|
import pytest
|
|
|
|
|
|
# @pytest.mark.parametrize("loader_type", ["mock"])
|
|
# @pytest.mark.parametrize("estimator_type", ["mock"])
|
|
# @pytest.mark.parametrize("batch_size", [3])
|
|
# def test_load_model_and_classes(model_loader, model_handle_mock, classes):
|
|
# model_loaded, classes_loaded = model_loader.load_model_and_classes("an identifier")
|
|
# assert model_loaded == model_handle_mock
|
|
# assert np.all(classes_loaded == classes)
|
|
|
|
@pytest.mark.parametrize("database_type", ["mock"])
|
|
def test_load_model_and_classes(model_loader, model_database_record_identifier, model, classes):
|
|
model_loaded = model_loader.load_model(model_database_record_identifier)
|
|
classes_loaded = model_loader.load_classes(model_database_record_identifier)
|
|
|
|
assert model_loaded == model
|
|
assert classes_loaded == classes
|