Browse Source

[FIX] pass pytest in python 3.11

pull/1/head
Gao Enhao 2 years ago
parent
commit
1379d04191
1 changed files with 6 additions and 8 deletions
  1. +6
    -8
      tests/test_abl_model.py

+ 6
- 8
tests/test_abl_model.py View File

@@ -11,24 +11,22 @@ class TestABLModel(object):
"""Test the initialization method of the ABLModel class."""

invalid_base_model = Mock(spec=[])

with pytest.raises(NotImplementedError):
ABLModel(invalid_base_model)

fit = Mock(return_value=1.0)
predict = Mock(return_value=np.array(1.0))

invalid_base_model = Mock(spec=fit)
invalid_base_model = Mock(spec=["fit"])
invalid_base_model.fit.return_value = 1.0
with pytest.raises(NotImplementedError):
ABLModel(invalid_base_model)

invalid_base_model = Mock(spec=predict)
invalid_base_model = Mock(spec=["predict"])
invalid_base_model.predict.return_value = np.array(1.0)
with pytest.raises(NotImplementedError):
ABLModel(invalid_base_model)

base_model = Mock(spec=["fit", "predict"])
base_model.fit = fit
base_model.predict = predict
base_model.fit.return_value = 1.0
base_model.predict.return_value = np.array(1.0)
model = ABLModel(base_model)
assert hasattr(model, "base_model"), "The model should have a 'base_model' attribute."



Loading…
Cancel
Save