U
    ‰dç  ã                   @   s<   d dl mZmZ d dlmZ d dlmZ G dd„ deƒZdS )é    )ÚUseNormalizerÚNormalizerContext)ÚBatchNormalizer)ÚLayersTestCasec                   @   s   e Zd Zdd„ ZdS )ÚTestNormalizerContextc              	   C   s@   t dd}td|iƒ  t ¡  d¡}|  ||¡ W 5 Q R X d S )Ngš™™™™™¹?)ZmomentumZBATCH)r   r   r   ÚcurrentZget_normalizerZassertEquals)ÚselfZbnZ
normalizer© r	   úA/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/normalizer_test.pyÚtest_normalizer_context   s    
z-TestNormalizerContext.test_normalizer_contextN)Ú__name__Ú
__module__Ú__qualname__r   r	   r	   r	   r
   r   
   s   r   N)Z caffe2.python.normalizer_contextr   r   Zcaffe2.python.normalizerr   Zcaffe2.python.layer_test_utilr   r   r	   r	   r	   r
   Ú<module>   s   