o
    j                     @   s   d dl Zd dlZd dlmZ d dlmZ ddlmZ ej	
dejejejgej	
ddd	gej	
d
dd	gdd Zej	
ddd	gej	
d
dd	gdd Zdd Zdd Zdd Zdd Zdd ZdS )    N)_supported_float_type)optical_flow_ilk   )_sin_flow_gendtypegaussianTF	prefilterc           	      C   s   t jd}|jdd}t|\}}|j| dd}t| }t|||||d}|jt| ks/J t	|| 
 dk s;J | |kr`tt t||||| d W d    d S 1 sYw   Y  d S d S )Nr      r
   sizeF)copy)r   r   r         ?)nprandomdefault_rngnormalr   Zastyper   r   r   absmeanpytestraises
ValueError)	r   r   r   rngimage0gt_flowimage1Zfloat_dtypeflow r   d/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/skimage/registration/tests/test_ilk.pytest_2d_motion	   s"   

"r   c                 C   sV   t jd}|jdd}t|dd\}}t||d| |d}t||  dk s)J d S )	N{   )2   7   <   r      )Znpics   )Zradiusr   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   test_3d_motion"   s   
r&   c                  C   8   t jd} | jdd}t||}t |dksJ d S )Nr   r	   r   r   r   r   r   r   allr   imgr   r   r   r   test_no_motion_2d2      
r,   c                  C   r'   )Nr   )@   r.   r.   r   r(   r*   r   r   r   test_no_motion_3d;   r-   r/   c                  C   sx   t jd} | jdd}t|\}}t||dd}|jdks J t||dd}|jdks.J t||  dk s:J d S )Nr   r	   r   float64r   float32gMbP?)	r   r   r   r   r   r   r   r   r   )r   r   r   r   Zflow_f64Zflow_f32r   r   r   test_optical_flow_dtypeD   s   r3   c                  C   sb   t jd} | jdd}| jdd}tt t||\}}W d    d S 1 s*w   Y  d S )Nr   r	   r   )   r
   r   r   r   r   r   r   r   r   )r   ZI0ZI1uvr   r   r   test_incompatible_shapesY   s   "r8   c                  C   sZ   t jd} | jdd}tt t||dd\}}W d    d S 1 s&w   Y  d S )Nr   r	   r   intr1   r5   )r   r+   r6   r7   r   r   r   test_wrong_dtypea   s
   "r:   )numpyr   r   Zskimage._shared.utilsr   Zskimage.registrationr   Z	test_tvl1r   markZparametrizeZfloat16r2   r0   r   r&   r,   r/   r3   r8   r:   r   r   r   r   <module>   s"    		