Open MyReason92 opened 2 months ago
import joblib import numpy as np
def test_model_loading(): model = joblib.load('model.pkl') assert model is not None
def test_model_prediction(): model = joblib.load('model.pkl') sample_input = np.array([[1, 2, 3, 4, 5]]) prediction = model.predict(sample_input) assert prediction is not None
import joblib import numpy as np
def test_model_loading(): model = joblib.load('model.pkl') assert model is not None
def test_model_prediction(): model = joblib.load('model.pkl') sample_input = np.array([[1, 2, 3, 4, 5]]) prediction = model.predict(sample_input) assert prediction is not None