o
    h                     @   s^   d dl mZ 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g ddd	 ZdS )
    )PathN)approx)compare_images)_image_directorieszim1, im2, tol, expect_rms))basn3p02.pngbasn3p02-minorchange.png
   N)r   r   r   gBs@)r   zbasn3p02-1px-offset.pngr   g%̴V@)r   zbasn3p02-half-1px-offset.pngr   g     O@)r   zbasn3p02-scrambled.pngr   gD/Xe@)
all127.png
all128.pngr      )r
   r	   r   r   c           
      C   s   | | tttdd \}}||  }t|| | t||  ||dd}	|du r2|	du s0J dS |	dus8J |	d t|ddksDJ dS )	a  
    Compare two images, expecting a particular RMS error.

    im1 and im2 are filenames relative to the baseline_dir directory.

    tol is the tolerance to pass to compare_images.

    expect_rms is the expected RMS value, or None. If None, the test will
    succeed if compare_images succeeds. Otherwise, the test will succeed if
    compare_images fails and returns an RMS error almost equal to this value.
    c                   S   s   dS )Ndummy r   r   r   x/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/matplotlib/tests/test_compare_images.py<lambda>=   s    z2test_image_comparison_expect_rms.<locals>.<lambda>T)tolin_decoratorNrmsg-C6?)abs)chdirmapr   r   shutilcopyfiler   r   )
im1im2r   
expect_rmstmp_pathmonkeypatchbaseline_dir
result_dir
result_im2resultsr   r   r    test_image_comparison_expect_rms   s   
0r!   )pathlibr   r   pytestr   matplotlib.testing.comparer   matplotlib.testing.decoratorsr   markparametrizer!   r   r   r   r   <module>   s    !