Skip to content

Commit e92e725

Browse files
authored
Merge pull request #311 from nipreps/fix/309
FIX: Disallow the estimator to iterate over mock data
2 parents 642fa15 + 1c16e44 commit e92e725

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

test/test_estimator.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,11 +103,18 @@ def test_estimator_init_model_instance(request):
103103

104104
def test_estimator_init_model_string(request, monkeypatch):
105105
rng = request.node.rng
106+
106107
# Patch ModelFactory.init to return DummyModel
107108
monkeypatch.setattr(
108109
"nifreeze.model.base.ModelFactory.init",
109110
lambda model, dataset, **kwargs: DummyModel(dataset=dataset),
110111
)
112+
113+
def mock_iterator(*_, **kwargs):
114+
return []
115+
116+
monkeypatch.setattr(iterators, "random_iterator", mock_iterator) # Avoid iterator issues
117+
111118
model_name = "dummy"
112119
est = Estimator(model=model_name, model_kwargs={})
113120
_dataset = DummyDataset(rng)

0 commit comments

Comments
 (0)