U \ a @slddlmZddlZddlZddlmZddlmZmZddlm Z ej dddd d d d d gddZ dS))PathN)approx)compare_imagesmake_test_filename)_image_directorieszim1, im2, tol, expect_rms) basn3p02.pngbasn3p02-minorchange.png N)rrrgBs@)rzbasn3p02-1px-offset.pngrg%̴V@)rzbasn3p02-half-1px-offset.pngrgO@)rzbasn3p02-scrambled.pngrgD/Xe@) all127.png all128.pngr)r r rr c Cstttdd\}}t||d}t|||||}t|||t|||dd}|dkrr|dkstn$|dk s~t|dt|dd kstdS) 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. cSsdS)NdummyrrrH/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/tests/test_compare_images.py9z2test_image_comparison_expect_rms..expectedT)tolZ in_decoratorNZrmsg-C6?)abs) maprrrshutilcopyfilerAssertionErrorr) Zim1Zim2rZ expect_rmsZ baseline_dirZ result_dirZ result_im1Z result_im2resultsrrr test_image_comparison_expect_rms s- r) pathlibrrZpytestrZmatplotlib.testing.comparerrZmatplotlib.testing.decoratorsrmarkZ parametrizerrrrrs"    !