aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/prompt-toolkit/py3/tests/test_utils.py
diff options
context:
space:
mode:
authorshadchin <shadchin@yandex-team.ru>2022-02-10 16:44:39 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:44:39 +0300
commite9656aae26e0358d5378e5b63dcac5c8dbe0e4d0 (patch)
tree64175d5cadab313b3e7039ebaa06c5bc3295e274 /contrib/python/prompt-toolkit/py3/tests/test_utils.py
parent2598ef1d0aee359b4b6d5fdd1758916d5907d04f (diff)
downloadydb-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.py152
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)