U
    dx                     @   sJ   d dl mZmZ d dlZedg G dd dejZedkrFe  dS )    )core	workspaceNpythonc                   @   s   e Zd Zdd ZdS )BlobDeallocationTestc                 C   s   t d}|jg dgdgdddgd}|jg dgdgd	d
dgd}|j||gddgdd t  t| t  t| | d d S )Nnetx   abc)shapevaluesydefZconcated_r   )ZaxisT)r   ZNetZGivenTensorStringFillZConcatr   ZResetWorkspaceZ
RunNetOnce
assertTrue)selfr   r   r    r   M/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/test/blob_deallocation_test.pytest   s    


zBlobDeallocationTest.testN)__name__
__module____qualname__r   r   r   r   r   r      s   r   __main__)	Zcaffe2.pythonr   r   ZunittestZ
GlobalInitZTestCaser   r   mainr   r   r   r   <module>   s
   