a
    ==icì  ã                   @   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¡}| g d¢ddg¡ |  t d¡¡ d}t ||||ggg d¢g¡ |  t d¡¡ |  t 	d¡j
|¡ |  t¡& t ||ggdgg¡ W d   ƒ n1 s¨0    Y  d S )	NÚtest_filler)ÚX0ZX1ZX2Z
concat_outZ
split_infor   )é   é   )Úfloatr
   r
   r
   )r   ZNetZConcatZassertFalser   ZHasBlobZFillRandomNetworkInputsÚ
assertTrueÚassertEqualZ	FetchBlobÚshapeÚassertRaisesÚRuntimeError)ÚselfÚnetZ	input_dim© r   új/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/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   