o
    poi                     @   s`   d dl Z d dlZd dlmZ d dlmZ d dlmZ eedfde	e dee
 defdd	ZdS )
    N)	Generator)contextmanager)Optionalexpected_warningmatchreturnc                 c   s    t jdd}dV  W d   n1 sw   Y  |D ]#}t|j| r?|du s3t|t|jr?t	d| j
 d|jqdS )zDCheck that no warning was raised/emitted under this context manager.T)recordN`z` was raised: )warningscatch_warnings
issubclasscategoryrecompilesearchstrmessageAssertionError__name__)r   r   r   w r   T/home/ubuntu/.local/lib/python3.10/site-packages/lightning_utilities/test/warning.pyno_warning_call   s   *r   )r   r
   collections.abcr   
contextlibr   typingr   Warningtyper   r   r   r   r   r   <module>   s   *