Your IP : 216.73.216.213
B
R�X' � @ s> d dl Z d dlmZmZ G dd� de j�ZG dd� de�ZdS )� N)�
Comparison�comparec s< e Zd ZdZdZ� fdd�Z� fdd�Z� fdd�Z� ZS ) �
ShouldWarna
This context manager is used to assert that warnings are issued
within the context it is managing.
:param expected: This should be a sequence made up of one or more elements,
each of one of the following types:
* A warning class, indicating that the type
of the warnings is important but not the
parameters it is created with.
* A warning instance, indicating that a
warning exactly matching the one supplied
should have been issued.
If no expected warnings are passed, you will need to inspect
the contents of the list returned by the context manager.
Fc s& t t| �jdd� dd� |D �| _d S )NT)�recordc S s g | ] }t |��qS � )�C)�.0�er r �H/opt/alt/python37/lib/python3.7/site-packages/testfixtures/shouldwarn.py�
<listcomp> s z'ShouldWarn.__init__.<locals>.<listcomp>)�superr �__init__�expected)�selfr )� __class__r r
r
s zShouldWarn.__init__c s t t| ��� | _t�d� | jS )N�always)r r � __enter__�recorded�warnings� simplefilter)r )r r r
r s
zShouldWarn.__enter__c sX t t| ��|||� | js$| jr$d S | js:| jr:| js:d S t| jdd� | jD �d� d S )Nc S s g | ]
}|j �qS r )�message)r Zwmr r r
r + s z'ShouldWarn.__exit__.<locals>.<listcomp>)�actual)r r �__exit__r � _empty_okayr r )r �exc_typeZexc_val�exc_tb)r r r
r % s zShouldWarn.__exit__) �__name__�
__module__� __qualname__�__doc__r r
r r �
__classcell__r r )r r
r s
r c s$ e Zd ZdZdZ� fdd�Z� ZS )�
ShouldNotWarnzs
This context manager is used to assert that no warnings are issued
within the context it is managing.
Tc s t t| ��� d S )N)r r! r
)r )r r r
r
6 s zShouldNotWarn.__init__)r r r r r r
r r r )r r
r! . s r! )r � testfixturesr r r �catch_warningsr r! r r r r
�<module> s (