aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/testing
diff options
context:
space:
mode:
authorsobols <sobols@yandex-team.ru>2022-02-10 16:47:08 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:47:08 +0300
commit03335cb18337a0ef51966452a66a69b01abea218 (patch)
treeb83306b6e37edeea782e9eed673d89286c4fef35 /library/python/testing
parent09961b69c61f471ddd594e0fd877df62a8021562 (diff)
downloadydb-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__.py6
-rw-r--r--library/python/testing/recipe/ya.make2
-rw-r--r--library/python/testing/yatest_lib/tools.py82
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):