aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/testing/yatest_lib/tests
diff options
context:
space:
mode:
authoriaz1607 <iaz1607@yandex-team.ru>2022-02-10 16:45:37 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:45:37 +0300
commit94e51c602b555459333b3c6ae92476c424c930bc (patch)
treeb2cc84ee7850122e7ccf51d0ea21e4fa7e7a5685 /library/python/testing/yatest_lib/tests
parente5437feb4ac2d2dc044e1090b9312dde5ef197e0 (diff)
downloadydb-94e51c602b555459333b3c6ae92476c424c930bc.tar.gz
Restoring authorship annotation for <iaz1607@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/python/testing/yatest_lib/tests')
-rw-r--r--library/python/testing/yatest_lib/tests/test_testsplitter.py206
-rw-r--r--library/python/testing/yatest_lib/tests/ya.make22
2 files changed, 114 insertions, 114 deletions
diff --git a/library/python/testing/yatest_lib/tests/test_testsplitter.py b/library/python/testing/yatest_lib/tests/test_testsplitter.py
index 87842e0ceb..394bfe5a74 100644
--- a/library/python/testing/yatest_lib/tests/test_testsplitter.py
+++ b/library/python/testing/yatest_lib/tests/test_testsplitter.py
@@ -1,103 +1,103 @@
-# coding: utf-8
-from yatest_lib import test_splitter
-
-
-def get_chunks(tests, modulo, mode):
- chunks = []
- if mode == "MODULO":
- for modulo_index in range(modulo):
- chunks.append(test_splitter.get_shuffled_chunk(tests, modulo, modulo_index))
- elif mode == "SEQUENTIAL":
- for modulo_index in range(modulo):
- chunks.append(test_splitter.get_sequential_chunk(tests, modulo, modulo_index))
- else:
- raise ValueError("no such mode")
- return chunks
-
-
-def check_not_intersect(chunk_list):
- test_set = set()
- total_size = 0
- for tests in chunk_list:
- total_size += len(tests)
- test_set.update(tests)
- return total_size == len(test_set)
-
-
-def check_max_diff(chunk_list):
- return max(map(len, chunk_list)) - min(map(len, chunk_list))
-
-
-def test_lot_of_chunks():
- for chunk_count in range(10, 20):
- for tests_count in range(chunk_count):
- chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert chunks.count([]) == chunk_count - tests_count
- assert len(chunks) == chunk_count
- chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert chunks.count([]) == chunk_count - tests_count
- assert len(chunks) == chunk_count
-
-
-def test_lot_of_tests():
- for tests_count in range(10, 20):
- for chunk_count in range(2, tests_count):
- chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert len(chunks) == chunk_count
- chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert len(chunks) == chunk_count
-
-
-def prime_chunk_count():
- for chunk_count in [7, 11, 13, 17, 23, 29]:
- for tests_count in range(chunk_count):
- chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert len(chunks) == chunk_count
- chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert len(chunks) == chunk_count
-
-
-def get_divisors(number):
- divisors = []
- for d in range(1, number + 1):
- if number % d == 0:
- divisors.append(d)
- return divisors
-
-
-def equal_chunks():
- for chunk_count in range(12, 31):
- for tests_count in get_divisors(chunk_count):
- chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) == 0
- assert len(chunks) == chunk_count
- chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) == 0
- assert len(chunks) == chunk_count
-
-
-def chunk_count_equal_tests_count():
- for chunk_count in range(10, 20):
- tests_count = chunk_count
- chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert len(chunks) == chunk_count
- chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
- assert check_not_intersect(chunks)
- assert check_max_diff(chunks) <= 1
- assert len(chunks) == chunk_count
+# coding: utf-8
+from yatest_lib import test_splitter
+
+
+def get_chunks(tests, modulo, mode):
+ chunks = []
+ if mode == "MODULO":
+ for modulo_index in range(modulo):
+ chunks.append(test_splitter.get_shuffled_chunk(tests, modulo, modulo_index))
+ elif mode == "SEQUENTIAL":
+ for modulo_index in range(modulo):
+ chunks.append(test_splitter.get_sequential_chunk(tests, modulo, modulo_index))
+ else:
+ raise ValueError("no such mode")
+ return chunks
+
+
+def check_not_intersect(chunk_list):
+ test_set = set()
+ total_size = 0
+ for tests in chunk_list:
+ total_size += len(tests)
+ test_set.update(tests)
+ return total_size == len(test_set)
+
+
+def check_max_diff(chunk_list):
+ return max(map(len, chunk_list)) - min(map(len, chunk_list))
+
+
+def test_lot_of_chunks():
+ for chunk_count in range(10, 20):
+ for tests_count in range(chunk_count):
+ chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert chunks.count([]) == chunk_count - tests_count
+ assert len(chunks) == chunk_count
+ chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert chunks.count([]) == chunk_count - tests_count
+ assert len(chunks) == chunk_count
+
+
+def test_lot_of_tests():
+ for tests_count in range(10, 20):
+ for chunk_count in range(2, tests_count):
+ chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert len(chunks) == chunk_count
+ chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert len(chunks) == chunk_count
+
+
+def prime_chunk_count():
+ for chunk_count in [7, 11, 13, 17, 23, 29]:
+ for tests_count in range(chunk_count):
+ chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert len(chunks) == chunk_count
+ chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert len(chunks) == chunk_count
+
+
+def get_divisors(number):
+ divisors = []
+ for d in range(1, number + 1):
+ if number % d == 0:
+ divisors.append(d)
+ return divisors
+
+
+def equal_chunks():
+ for chunk_count in range(12, 31):
+ for tests_count in get_divisors(chunk_count):
+ chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) == 0
+ assert len(chunks) == chunk_count
+ chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) == 0
+ assert len(chunks) == chunk_count
+
+
+def chunk_count_equal_tests_count():
+ for chunk_count in range(10, 20):
+ tests_count = chunk_count
+ chunks = get_chunks(range(tests_count), chunk_count, "SEQUENTIAL")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert len(chunks) == chunk_count
+ chunks = get_chunks(range(tests_count), chunk_count, "MODULO")
+ assert check_not_intersect(chunks)
+ assert check_max_diff(chunks) <= 1
+ assert len(chunks) == chunk_count
diff --git a/library/python/testing/yatest_lib/tests/ya.make b/library/python/testing/yatest_lib/tests/ya.make
index 6670106533..8586c6ef7d 100644
--- a/library/python/testing/yatest_lib/tests/ya.make
+++ b/library/python/testing/yatest_lib/tests/ya.make
@@ -1,14 +1,14 @@
OWNER(g:yatest)
-
+
PY23_TEST()
-
-PEERDIR(
- library/python/testing/yatest_lib
-)
-
-TEST_SRCS(
+
+PEERDIR(
+ library/python/testing/yatest_lib
+)
+
+TEST_SRCS(
test_external.py
- test_testsplitter.py
-)
-
-END()
+ test_testsplitter.py
+)
+
+END()