U
    ‰dG
  ã                   @   s˜   d dl Z d dlmZ d dlmZmZ d dlZd dl	m
Z
mZ d dlm  mZ d dlm  mZ e  ejj d¡G dd„ dejƒƒZedkr”e  ¡  dS )é    N)ÚgivenÚsettings)ÚcoreÚ	workspacezNo MKLDNN support.c                   @   s¾   e Zd Zef e dd¡e dd¡e dd¡e dd¡dœej—Žeddddd„ ƒƒZ	ef e dd¡e dd¡e dd¡e dd¡ej
e d	d
¡dd
ddœej—Žeddddd„ ƒƒZdS )Ú	ShapeTesté   é€   )ÚnÚcÚhÚwé
   N)Zmax_examplesÚdeadlinec                 C   sî   t jddgdg|d d}t jddgdg|d d}tj ||||¡ tj¡d	 }	t d|	|d ¡ t d|	|d ¡ t 	|¡ t 	|¡ t 
d¡}
t 
d¡}tj|
|ddd
sêt| ¡ ƒ t|
 ¡ ƒ tt t ||
 ¡¡ƒ |  d¡ d S )NÚShapeÚX0ÚY0r   )Údevice_optionÚX1ÚY1r   ç      à?©ZatolZrtolF)r   ÚCreateOperatorÚnpÚrandomÚrandÚastypeÚfloat32r   ÚFeedBlobÚRunOperatorOnceÚ	FetchBlobÚallcloseÚprintÚflattenÚmaxÚabsÚ
assertTrue)Úselfr	   r
   r   r   ÚgcÚdcÚop0Úop1ÚXr   r   © r,   úE/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/ideep/shape_op_test.pyÚ
test_shape   s0    üü



zShapeTest.test_shaper   é   )Zmin_sizeÚmax_size)r	   r
   r   r   Úaxesc                 C   s  t t|ƒƒ ¡ }tjddgdg||d d}tjddgdg||d d}	tj ||||¡ tj	¡d	 }
t
 d|
|d ¡ t
 d|
|d ¡ t
 |¡ t
 |	¡ t
 d¡}t
 d¡}tj||ddd
sþt| ¡ ƒ t| ¡ ƒ tt t || ¡¡ƒ |  d¡ d S )Nr   r   r   r   )r1   r   r   r   r   r   r   F)ÚlistÚsetÚsortr   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   )r&   r	   r
   r   r   r1   r'   r(   r)   r*   r+   r   r   r,   r,   r-   Útest_shape_with_axes2   s6    ûû



zShapeTest.test_shape_with_axes)Ú__name__Ú
__module__Ú__qualname__r   ÚstZintegersÚmuZgcsr   r.   Úlistsr5   r,   r,   r,   r-   r      s&   


ýü



üû
r   Ú__main__)ZunittestZhypothesis.strategiesZ
strategiesr9   Z
hypothesisr   r   Znumpyr   Zcaffe2.pythonr   r   Z"caffe2.python.hypothesis_test_utilÚpythonZhypothesis_test_utilÚhuZcaffe2.python.ideep_test_utilZideep_test_utilr:   ZskipIfÚCZ
use_mkldnnZHypothesisTestCaser   r6   Úmainr,   r,   r,   r-   Ú<module>   s   H