U md@shddlmZddlmZmZmZddlZddlmZddl m Z dgZ GdddZ d d d d dZ dS) ) annotations)TupleTypeUnionN)WarningsChecker)warns pytest_warnsc@s$eZdZddZddZddZdS)NoWarningsCheckercCstjdd|_g|_dS)NT)record)warningscatch_warningscwrecselfrR/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/compat/pytest.py__init__ szNoWarningsChecker.__init__cCs|j|_dS)N)r __enter__rrrrrrszNoWarningsChecker.__enter__cCs2|jr.dd|jD}d|}td|dS)NcSsg|] }|jjqSr)category__name__).0wrrr sz.NoWarningsChecker.__exit__..z\nzJFunction is marked as not warning but the following warnings were found: )rjoinAssertionError)rtypevalue tracebackr Zjoinedrrr__exit__s  zNoWarningsChecker.__exit__N)r __module__ __qualname__rrrrrrrr sr z0Type[Warning] | Tuple[Type[Warning], ...] | Nonez)Union[WarningsChecker, NoWarningsChecker])warningreturncCs&|dkrtS|dk stt|SdS)z Parameters ---------- warning : {None, Warning, Tuple[Warning]} None if no warning is produced, or a single or multiple Warnings Returns ------- cm N)r rr)r"rrrrs ) __future__rtypingrrrr Z_pytest.recwarnrZpytestr__all__r rrrrrs