aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/runtime_py3/test/test_traceback.py
diff options
context:
space:
mode:
authorabodrov <abodrov@yandex-team.ru>2022-02-10 16:47:18 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:47:18 +0300
commitf02a874a7290593efbe4b3aeae69a04b46c1cc86 (patch)
tree9814fbd1c3effac9b8377c5d604b367b14e2db55 /library/python/runtime_py3/test/test_traceback.py
parent1b8c043abfc689b3d27384c4202b20031fc4ae31 (diff)
downloadydb-f02a874a7290593efbe4b3aeae69a04b46c1cc86.tar.gz
Restoring authorship annotation for <abodrov@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/python/runtime_py3/test/test_traceback.py')
-rw-r--r--library/python/runtime_py3/test/test_traceback.py124
1 files changed, 62 insertions, 62 deletions
diff --git a/library/python/runtime_py3/test/test_traceback.py b/library/python/runtime_py3/test/test_traceback.py
index 0cc4dec75a..82087ce98a 100644
--- a/library/python/runtime_py3/test/test_traceback.py
+++ b/library/python/runtime_py3/test/test_traceback.py
@@ -1,63 +1,63 @@
-# -*- coding: utf-8 -*-
-
-from __future__ import print_function, absolute_import, division
-
-import os
-import re
-
-import pytest
-
-import yatest.common as yc
-
-
-def clean_traceback(traceback):
- traceback = re.sub(br'\033\[(\d|;)+?m', b'', traceback) # strip ANSI codes
- traceback = re.sub(br' at 0x[0-9a-fA-F]+', b'', traceback) # remove object ids
- return traceback
-
-
-@pytest.mark.parametrize('mode', [
- 'default',
- 'ultratb_color',
- 'ultratb_verbose',
-])
-@pytest.mark.parametrize('entry_point', [
- 'main',
- 'custom',
-])
-def test_traceback(mode, entry_point):
- tb_tool = yc.build_path('library/python/runtime_py3/test/traceback/traceback')
- stdout_path = yc.test_output_path('stdout_raw.txt')
- stderr_path = yc.test_output_path('stderr_raw.txt')
- filtered_stdout_path = yc.test_output_path('stdout.txt')
- filtered_stderr_path = yc.test_output_path('stderr.txt')
-
- env = os.environ.copy()
+# -*- coding: utf-8 -*-
+
+from __future__ import print_function, absolute_import, division
+
+import os
+import re
+
+import pytest
+
+import yatest.common as yc
+
+
+def clean_traceback(traceback):
+ traceback = re.sub(br'\033\[(\d|;)+?m', b'', traceback) # strip ANSI codes
+ traceback = re.sub(br' at 0x[0-9a-fA-F]+', b'', traceback) # remove object ids
+ return traceback
+
+
+@pytest.mark.parametrize('mode', [
+ 'default',
+ 'ultratb_color',
+ 'ultratb_verbose',
+])
+@pytest.mark.parametrize('entry_point', [
+ 'main',
+ 'custom',
+])
+def test_traceback(mode, entry_point):
+ tb_tool = yc.build_path('library/python/runtime_py3/test/traceback/traceback')
+ stdout_path = yc.test_output_path('stdout_raw.txt')
+ stderr_path = yc.test_output_path('stderr_raw.txt')
+ filtered_stdout_path = yc.test_output_path('stdout.txt')
+ filtered_stderr_path = yc.test_output_path('stderr.txt')
+
+ env = os.environ.copy()
env.pop('PYTHONPATH', None) # Do not let program peek into its sources on filesystem
- if entry_point == 'custom':
- env['Y_PYTHON_ENTRY_POINT'] = 'library.python.runtime_py3.test.traceback.crash:main'
-
- proc = yc.execute(
- command=[tb_tool, mode],
- env=env,
- stdout=stdout_path,
- stderr=stderr_path,
- check_exit_code=False,
- )
-
- with open(filtered_stdout_path, 'wb') as f:
- f.write(clean_traceback(proc.std_out))
-
- with open(filtered_stderr_path, 'wb') as f:
- f.write(clean_traceback(proc.std_err))
-
- return {
- 'stdout': yc.canonical_file(
- filtered_stdout_path,
- local=True,
- ),
- 'stderr': yc.canonical_file(
- filtered_stderr_path,
- local=True,
- ),
- }
+ if entry_point == 'custom':
+ env['Y_PYTHON_ENTRY_POINT'] = 'library.python.runtime_py3.test.traceback.crash:main'
+
+ proc = yc.execute(
+ command=[tb_tool, mode],
+ env=env,
+ stdout=stdout_path,
+ stderr=stderr_path,
+ check_exit_code=False,
+ )
+
+ with open(filtered_stdout_path, 'wb') as f:
+ f.write(clean_traceback(proc.std_out))
+
+ with open(filtered_stderr_path, 'wb') as f:
+ f.write(clean_traceback(proc.std_err))
+
+ return {
+ 'stdout': yc.canonical_file(
+ filtered_stdout_path,
+ local=True,
+ ),
+ 'stderr': yc.canonical_file(
+ filtered_stderr_path,
+ local=True,
+ ),
+ }