o
    jĎim                     @   s   d Z ddlZddlZddlmZ ddlZddlmZ ddlm	Z	 ej
dddd	 Zej
ddd
d Zej
dddd Zej
dddd Zej
dddd Zej
dddd Zej
dddd Zej
dddd Zej
dddd Zej
dddd Zddee fddZdS ) z
Testing the package pybboxes.
Default image/bbox selected from the following source
https://albumentations.ai/docs/getting_started/bounding_boxes_augmentation/
    N)Optional)EXPECTED_OUTPUTS)	load_jsonpackage)scopec                   C      dS )N)i  i   r   r   r   S/home/jeff/fluffinator/venv/lib/python3.10/site-packages/tests/pybboxes/conftest.py
image_size      r
   c                   C   r   )Ni*  r   r   r   r   r	   	bbox_area   r   r   c                   C      g dS )N)皙?      ?g      ?g?r   r   r   r   r	   albumentations_bbox      r   c                   C   r   )N)   r   r   r   r   r   r	   unnormalized_bbox_shift_amount   r   r   c                   C   r   )N)g?gQ?r   r   r   r   r	   normalized_bbox_shift_amount$   r   r   c                   C   r   )Ng      ?r   r   r   r   r	   scale_factor)   r   r   c                   C   r   )N)b   Y  iB  u   r   r   r   r   r	   	coco_bbox.   r   r   c                   C   r   )N)r   r   皙?333333?r   r   r   r   r	   fiftyone_bbox3   r   r   c                   C   r   )N)r   r   i  i  r   r   r   r   r	   voc_bbox8   r   r   c                   C   r   )N)gfffff?gfffff?r   r   r   r   r   r   r	   	yolo_bbox=   r   r   prefixc                    s    fdd} d u rd |S )Nc                    sP   t jt| dd}t jt | d}| jdd}t	|| | _
| S )Nz.py z.jsonoutput_)ospathbasenameinspectgetfilereplacejoinr   __name__r   output)fnargskwargsmodule_namer#   	test_namer   r   r	   wrapperC   s
   z$get_expected_output.<locals>.wrapperr    r   )r   r1   r   r0   r	   get_expected_outputB   s   r2   )N)__doc__r%   r"   typingr   pytesttests.pybboxesr   tests.utilsr   fixturer
   r   r   r   r   r   r   r   r   r   strr2   r   r   r   r	   <module>   s8    



















