def test_exhausts_randomly_when_filtering():
def f(chooser):
chooser.choose(range(10), lambda x: False)
tree = ChoiceTree()
random = Random()
tree.step(random_selection_order(random), f)
assert tree.exhausted
def test_exhausts_randomly():
def f(chooser):
chooser.choose(range(10))
tree = ChoiceTree()
random = Random()
seen = set()
for _ in range(10):
seen.add(tree.step(random_selection_order(random), f))
assert len(seen) == 10
assert tree.exhausted