U
    ‰dì  ã                   @   s*   d dl mZmZmZ G dd„ dejƒZdS )é    )ÚcoreÚ	test_utilÚ	workspacec                   @   s   e Zd Zdd„ ZdS )Ú
TestFillerc              	   C   s¦   t  d¡}| dddgddg¡ |  t d¡¡ d}t ||||ggdddgg¡ |  t d¡¡ |  t 	d¡j
|¡ |  t¡ t ||ggdgg¡ W 5 Q R X d S )	NÚtest_fillerZX0ZX1ZX2Z
concat_outZ
split_info)é   é   Úfloat)r   ZNetZConcatZassertFalser   ZHasBlobZFillRandomNetworkInputsÚ
assertTrueÚassertEqualZ	FetchBlobÚshapeÚassertRaisesÚRuntimeError)ÚselfÚnetZ	input_dim© r   ú=/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/filler_test.pyr   	   s    
zTestFiller.test_fillerN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s   r   N)Zcaffe2.pythonr   r   r   ZTestCaser   r   r   r   r   Ú<module>   s   