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