[library/python/pytest] Better py3 colorization

1 files changed, 36 insertions, 0 deletions

diff --git a/library/python/pytest/ut/test_tools.py b/library/python/pytest/ut/test_tools.py
index 412ec42d24..212a20fb17 100644
--- a/library/python/pytest/ut/test_tools.py
+++ b/library/python/pytest/ut/test_tools.py
@@ -95,3 +95,39 @@ def test_path_resolving_for_local_conftest_load_policy(
     mocker.patch.object(sys, 'extra_modules', extra_modules)
     got = yatest_tools.split_node_id(node_id + parameters)
     assert (expected_class_name, expected_test_name + parameters) == got
+
+
+DATA = [
+    (
+        "simple_bt",
+        """
+path/test.py:15: in test
+    return foo(0)
+path/test.py:5: in foo
+    b, a = 1, 1 / p
+E   ZeroDivisionError: integer division or modulo by zero
+""",
+    ),
+    (
+        "nested_by",
+        """
+path/test.py:13: in test
+    raise Exception("123\n123\npath/test.py:15: in test")
+E   Exception: 123
+E   123
+E   path/test.py:15: in test
+
+During handling of the above exception, another exception occurred:
+path/test.py:15: in test
+    return foo(0)
+path/test.py:5: in foo
+    b, a = 1, 1 / p
+E   ZeroDivisionError: division by zero
+""",
+    ),
+]
+
+
+@pytest.mark.parametrize("bt", [x[1] for x in DATA], ids=([x[0] for x in DATA]))
+def test_colorize_pytest_error(bt):
+    return yatest_tools.colorize_pytest_error(bt)