Skip to content

Commit bf2123d

Browse files
committed
Update test_learner.py
1 parent 6ce88c3 commit bf2123d

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

adaptive/tests/test_learner.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -339,10 +339,10 @@ def test_learner1d_first_iteration():
339339
assert set(points) == set(rest)
340340

341341
learner = Learner1D(lambda x: None, (-1, 1))
342-
(point,), loss_improvements = learner.choose_points(1)
343-
to_see = set([-1, 1]).remove(point)
344342
points, loss_improvements = learner.choose_points(1)
345-
assert set(points) == set(points)
343+
to_see = set([-1, 1]) - set(points)
344+
points, loss_improvements = learner.choose_points(1)
345+
assert set(points) == set(to_see)
346346

347347

348348
@pytest.mark.xfail

0 commit comments

Comments
 (0)