generated from Python-2026Spring/assignment-05-final-project-template
18 lines
451 B
Plaintext
18 lines
451 B
Plaintext
|
# tests/test_model.py
|
||
|
|
import joblib
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
def test_model_prediction():
|
||
|
|
# 加载模型
|
||
|
|
model = joblib.load("../telco_churn_model.pkl")
|
||
|
|
# 构造测试数据
|
||
|
|
test_data = pd.DataFrame({
|
||
|
|
"tenure": [2],
|
||
|
|
"MonthlyCharges": [89.99],
|
||
|
|
"Contract": ["Month-to-month"]
|
||
|
|
})
|
||
|
|
# 测试预测是否正常运行
|
||
|
|
pred = model.predict(test_data)
|
||
|
|
assert pred is not None # 只要能预测就通过
|
||
|
|
|
||
|
|
|