diff options
author | sobols <sobols@yandex-team.ru> | 2022-02-10 16:47:08 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:47:08 +0300 |
commit | 03335cb18337a0ef51966452a66a69b01abea218 (patch) | |
tree | b83306b6e37edeea782e9eed673d89286c4fef35 /library/python/testing | |
parent | 09961b69c61f471ddd594e0fd877df62a8021562 (diff) | |
download | ydb-03335cb18337a0ef51966452a66a69b01abea218.tar.gz |
Restoring authorship annotation for <sobols@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/python/testing')
-rw-r--r-- | library/python/testing/recipe/__init__.py | 6 | ||||
-rw-r--r-- | library/python/testing/recipe/ya.make | 2 | ||||
-rw-r--r-- | library/python/testing/yatest_lib/tools.py | 82 |
3 files changed, 45 insertions, 45 deletions
diff --git a/library/python/testing/recipe/__init__.py b/library/python/testing/recipe/__init__.py index 91d31d370a..5ef9c5c189 100644 --- a/library/python/testing/recipe/__init__.py +++ b/library/python/testing/recipe/__init__.py @@ -1,5 +1,5 @@ -from __future__ import print_function - +from __future__ import print_function + import os import sys import json @@ -86,7 +86,7 @@ def declare_recipe(start, stop): parsed_args, argv = get_options() if parsed_args.show_cwd: - print("Recipe \"{} {}\" working dir is {}".format(sys.argv[0], " ".join(argv), os.getcwd())) + print("Recipe \"{} {}\" working dir is {}".format(sys.argv[0], " ".join(argv), os.getcwd())) try: if argv[0] == RECIPE_START_OPTION: diff --git a/library/python/testing/recipe/ya.make b/library/python/testing/recipe/ya.make index 0d98ecc3d6..dd323aa245 100644 --- a/library/python/testing/recipe/ya.make +++ b/library/python/testing/recipe/ya.make @@ -3,7 +3,7 @@ OWNER( g:yatest ) -PY23_LIBRARY() +PY23_LIBRARY() PY_SRCS( __init__.py diff --git a/library/python/testing/yatest_lib/tools.py b/library/python/testing/yatest_lib/tools.py index 0cf2ce87c8..b72d79c162 100644 --- a/library/python/testing/yatest_lib/tools.py +++ b/library/python/testing/yatest_lib/tools.py @@ -1,4 +1,4 @@ -import six +import six import sys @@ -16,46 +16,46 @@ def to_utf8(value): return value.encode('utf-8', 'ignore') else: return str(value) - - -def trim_string(s, max_bytes): - """ - Adjusts the length of the string s in order to fit it - into max_bytes bytes of storage after encoding as UTF-8. - Useful when cutting filesystem paths. - :param s: unicode string - :param max_bytes: number of bytes - :return the prefix of s - """ - if isinstance(s, six.text_type): - return _trim_unicode_string(s, max_bytes) - - if isinstance(s, six.binary_type): - if len(s) <= max_bytes: - return s - s = s.decode('utf-8', errors='ignore') - s = _trim_unicode_string(s, max_bytes) - s = s.encode('utf-8', errors='ignore') - return s - - raise TypeError('a string is expected') - - -def _trim_unicode_string(s, max_bytes): - if len(s) * 4 <= max_bytes: - # UTF-8 uses at most 4 bytes per character - return s - - result = [] - cur_byte_length = 0 - - for ch in s: - cur_byte_length += len(ch.encode('utf-8')) - if cur_byte_length > max_bytes: - break - result.append(ch) - - return ''.join(result) + + +def trim_string(s, max_bytes): + """ + Adjusts the length of the string s in order to fit it + into max_bytes bytes of storage after encoding as UTF-8. + Useful when cutting filesystem paths. + :param s: unicode string + :param max_bytes: number of bytes + :return the prefix of s + """ + if isinstance(s, six.text_type): + return _trim_unicode_string(s, max_bytes) + + if isinstance(s, six.binary_type): + if len(s) <= max_bytes: + return s + s = s.decode('utf-8', errors='ignore') + s = _trim_unicode_string(s, max_bytes) + s = s.encode('utf-8', errors='ignore') + return s + + raise TypeError('a string is expected') + + +def _trim_unicode_string(s, max_bytes): + if len(s) * 4 <= max_bytes: + # UTF-8 uses at most 4 bytes per character + return s + + result = [] + cur_byte_length = 0 + + for ch in s: + cur_byte_length += len(ch.encode('utf-8')) + if cur_byte_length > max_bytes: + break + result.append(ch) + + return ''.join(result) def to_str(s): |