2020-12-25 02:19:27 +00:00
|
|
|
"""Shared pytest config."""
|
2020-12-24 22:21:41 +00:00
|
|
|
import sys
|
|
|
|
|
from tqdm import tqdm
|
|
|
|
|
from pytest import fixture
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@fixture(autouse=True)
|
|
|
|
|
def pretest_posttest():
|
|
|
|
|
"""Fixture for all tests ensuring environment cleanup"""
|
|
|
|
|
try:
|
|
|
|
|
sys.setswitchinterval(1)
|
|
|
|
|
except AttributeError:
|
|
|
|
|
sys.setcheckinterval(100) # deprecated
|
|
|
|
|
|
|
|
|
|
if getattr(tqdm, "_instances", False):
|
|
|
|
|
n = len(tqdm._instances)
|
|
|
|
|
if n:
|
|
|
|
|
tqdm._instances.clear()
|
|
|
|
|
raise EnvironmentError(
|
|
|
|
|
"{0} `tqdm` instances still in existence PRE-test".format(n))
|
|
|
|
|
yield
|
|
|
|
|
if getattr(tqdm, "_instances", False):
|
|
|
|
|
n = len(tqdm._instances)
|
|
|
|
|
if n:
|
|
|
|
|
tqdm._instances.clear()
|
|
|
|
|
raise EnvironmentError(
|
|
|
|
|
"{0} `tqdm` instances still in existence POST-test".format(n))
|
