fix: Click ctx error

This commit is contained in:
Suraj Shetty 2021-05-05 14:47:41 +05:30
parent ead3d76c84
commit bb833799dc

View file

@ -19,11 +19,13 @@ import click
import requests
import unittest.util
click.get_current_context().color = True
unittest_runner = unittest.TextTestRunner
SLOW_TEST_THRESHOLD = 2
click_ctx = click.get_current_context(True)
if click_ctx:
click_ctx.color = True
def xmlrunner_wrapper(output):
"""Convenience wrapper to keep method signature unchanged for XMLTestRunner and TextTestRunner"""
def _runner(*args, **kwargs):