U
    ‰df  ã                   @   s*   d dl Z d dlmZ G dd„ de jƒZdS )é    N)Útaskc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestTaskc                 C   sd   t  ¡ dft  ¡ dft  ¡ dft  g ¡dft  ¡ dft  ¡ dfg}|D ]\}}|  | ¡ |¡ qFd S )Nz!Cluster(nodes=[], node_kwargs={})zNode(name=local, kwargs={})z8TaskGroup(tasks=[], workspace_type=None, remote_nets=[])z!TaskOutput(names=[], values=None)z-Task(name=local/task, node=local, outputs=[])z)SetupNets(init_nets=None, exit_nets=None))	r   ZClusterÚNodeÚ	TaskGroupZ
TaskOutputZTaskZ	SetupNetsÚassertEqualÚ__repr__)ÚselfZcasesÚobjZwant© r
   ú;/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/task_test.pyÚtestRepr   s    

þ

÷zTestTask.testReprc                 C   s    t  ¡ }| ¡ }|  |j¡ d S )N)r   r   r   ZassertFalseZ_already_used)r   Z
task_groupÚ_reprr
   r
   r   ÚtestEffectlessRepr   s    zTestTask.testEffectlessReprN)Ú__name__Ú
__module__Ú__qualname__r   r   r
   r
   r
   r   r      s   r   )ZunittestZcaffe2.pythonr   ZTestCaser   r
   r
   r
   r   Ú<module>   s   