From: Łukasz Langa <lukasz@langa.pl>
Date: Wed, 21 Mar 2018 01:40:20 +0000 (-0700)
Subject: Support skipping AST printing on test failure
X-Git-Url: https://git.madduck.net/etc/vim.git/commitdiff_plain/591bedc2be0cec92c5f253fd473864c876233114?hp=d1e0d79e38db934a65dded44b903f05ab85829b4

Support skipping AST printing on test failure
---

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: