U
    ‰dV  ã                   @   s:   d dl mZ d dlZd dlZd dlZG dd„ dejƒZdS )é    )Ú	workspaceNc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestDBc                 C   s2   t  ¡ \}| _t |¡ dd„ tddƒD ƒ| _d S )Nc                 S   s,   g | ]$}d   |¡ d¡d  |¡ d¡f‘qS )zkey{}Úasciizvalue{})ÚformatÚencode)Ú.0Úi© r	   ú9/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/db_test.pyÚ
<listcomp>   s   ýþz TestDB.setUp.<locals>.<listcomp>é   é
   )ÚtempfileÚmkstempÚ	file_nameÚosÚcloseÚrangeÚdata)ÚselfÚhandler	   r	   r
   ÚsetUp   s
    
ûzTestDB.setUpc                 C   s¨   t j d| jt jjj¡}| jD ]\}}| ¡ }| ||¡ ~q~t j d| jt jjj	¡}| 
¡ }g }| ¡ rŒ| | ¡ | ¡ f¡ | ¡  qd~| ¡  |  || j¡ d S )NZminidb)r   ÚCZ	create_dbr   ZModeÚwriter   Znew_transactionÚputÚreadZ
new_cursorZvalidÚappendÚkeyÚvalueÚnextr   ÚassertEqual)r   Údbr   r   ZtransactionÚcursorr   r	   r	   r
   Ú
testSimple   s.      ÿ  ÿ
zTestDB.testSimpleN)Ú__name__Ú
__module__Ú__qualname__r   r#   r	   r	   r	   r
   r      s   r   )Zcaffe2.pythonr   r   r   ZunittestZTestCaser   r	   r	   r	   r
   Ú<module>   s   