From 591bedc2be0cec92c5f253fd473864c876233114 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=C5=81ukasz=20Langa?= Date: Tue, 20 Mar 2018 18:40:20 -0700 Subject: [PATCH] Support skipping AST printing on test failure --- tests/test_black.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_black.py b/tests/test_black.py index 7dba611..ec7a883 100644 --- a/tests/test_black.py +++ b/tests/test_black.py @@ -1,6 +1,7 @@ #!/usr/bin/env python3 from functools import partial from io import StringIO +import os from pathlib import Path import sys from typing import Any, List, Tuple @@ -47,7 +48,7 @@ class BlackTestCase(unittest.TestCase): maxDiff = None def assertFormatEqual(self, expected: str, actual: str) -> None: - if actual != expected: + if actual != expected and not os.environ.get('SKIP_AST_PRINT'): bdv: black.DebugVisitor[Any] black.out('Expected tree:', fg='green') try: -- 2.39.5