U
    d                     @   s,   d dl m  m  mZ dd Zdd ZdS )    Nc                 C   sV   | j dddddd | j ddtjd	d
d | j dddddd | j dddddd d S )Nz-Gz--generate-serialized
store_truegeneratez@generate output files (default=false, compares to current files))actiondesthelpz-Oz--outputoutputz'output directory (default: %(default)s))defaultr   r   z-Dz--disable-serialized-checkdisablez!disable checking serialized testsz-Cz--disable-gen-coveragedisable_coveragez)disable generating coverage markdown file)Z	addoptionserialDATA_DIR)parser r   H/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/operator_test/conftest.pypytest_addoption	   s8    r   c                 C   sv   | j ddd}| j dtjd}| j ddd}| j ddd}tjd| tjd| tjd	| tjd
| d S )Nr   F)r   r   r	   r
   Zshould_generate_output
output_dirZdisable_serialized_checkZdisable_gen_coverage)Z	getoptionr   r   Z_output_context__setattr__)configr   r   r	   r
   r   r   r   pytest_configure(   s    r   )Z2caffe2.python.serialized_test.serialized_test_utilpythonZserialized_testZserialized_test_utilr   r   r   r   r   r   r   <module>   s   