diff options
author | shadchin <shadchin@yandex-team.ru> | 2022-02-10 16:44:39 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:44:39 +0300 |
commit | e9656aae26e0358d5378e5b63dcac5c8dbe0e4d0 (patch) | |
tree | 64175d5cadab313b3e7039ebaa06c5bc3295e274 /contrib/python/prompt-toolkit/py3/tests/test_utils.py | |
parent | 2598ef1d0aee359b4b6d5fdd1758916d5907d04f (diff) | |
download | ydb-e9656aae26e0358d5378e5b63dcac5c8dbe0e4d0.tar.gz |
Restoring authorship annotation for <shadchin@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'contrib/python/prompt-toolkit/py3/tests/test_utils.py')
-rw-r--r-- | contrib/python/prompt-toolkit/py3/tests/test_utils.py | 152 |
1 files changed, 76 insertions, 76 deletions
diff --git a/contrib/python/prompt-toolkit/py3/tests/test_utils.py b/contrib/python/prompt-toolkit/py3/tests/test_utils.py index 4104fbfcb7..6a1c17955c 100644 --- a/contrib/python/prompt-toolkit/py3/tests/test_utils.py +++ b/contrib/python/prompt-toolkit/py3/tests/test_utils.py @@ -1,76 +1,76 @@ -import itertools - -import pytest - -from prompt_toolkit.utils import take_using_weights - - -def test_using_weights(): - def take(generator, count): - return list(itertools.islice(generator, 0, count)) - - # Check distribution. - data = take(take_using_weights(["A", "B", "C"], [5, 10, 20]), 35) - assert data.count("A") == 5 - assert data.count("B") == 10 - assert data.count("C") == 20 - - assert data == [ - "A", - "B", - "C", - "C", - "B", - "C", - "C", - "A", - "B", - "C", - "C", - "B", - "C", - "C", - "A", - "B", - "C", - "C", - "B", - "C", - "C", - "A", - "B", - "C", - "C", - "B", - "C", - "C", - "A", - "B", - "C", - "C", - "B", - "C", - "C", - ] - - # Another order. - data = take(take_using_weights(["A", "B", "C"], [20, 10, 5]), 35) - assert data.count("A") == 20 - assert data.count("B") == 10 - assert data.count("C") == 5 - - # Bigger numbers. - data = take(take_using_weights(["A", "B", "C"], [20, 10, 5]), 70) - assert data.count("A") == 40 - assert data.count("B") == 20 - assert data.count("C") == 10 - - # Negative numbers. - data = take(take_using_weights(["A", "B", "C"], [-20, 10, 0]), 70) - assert data.count("A") == 0 - assert data.count("B") == 70 - assert data.count("C") == 0 - - # All zero-weight items. - with pytest.raises(ValueError): - take(take_using_weights(["A", "B", "C"], [0, 0, 0]), 70) +import itertools + +import pytest + +from prompt_toolkit.utils import take_using_weights + + +def test_using_weights(): + def take(generator, count): + return list(itertools.islice(generator, 0, count)) + + # Check distribution. + data = take(take_using_weights(["A", "B", "C"], [5, 10, 20]), 35) + assert data.count("A") == 5 + assert data.count("B") == 10 + assert data.count("C") == 20 + + assert data == [ + "A", + "B", + "C", + "C", + "B", + "C", + "C", + "A", + "B", + "C", + "C", + "B", + "C", + "C", + "A", + "B", + "C", + "C", + "B", + "C", + "C", + "A", + "B", + "C", + "C", + "B", + "C", + "C", + "A", + "B", + "C", + "C", + "B", + "C", + "C", + ] + + # Another order. + data = take(take_using_weights(["A", "B", "C"], [20, 10, 5]), 35) + assert data.count("A") == 20 + assert data.count("B") == 10 + assert data.count("C") == 5 + + # Bigger numbers. + data = take(take_using_weights(["A", "B", "C"], [20, 10, 5]), 70) + assert data.count("A") == 40 + assert data.count("B") == 20 + assert data.count("C") == 10 + + # Negative numbers. + data = take(take_using_weights(["A", "B", "C"], [-20, 10, 0]), 70) + assert data.count("A") == 0 + assert data.count("B") == 70 + assert data.count("C") == 0 + + # All zero-weight items. + with pytest.raises(ValueError): + take(take_using_weights(["A", "B", "C"], [0, 0, 0]), 70) |