o :a @sbddlmZddlZddlZddlmZddlmZmZddlm Z ej dgddd Z 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}|dur:|dus8JdS|dus@J|dt|dd ksLJdS) 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)NZdummyr r r F/usr/lib/python3/dist-packages/matplotlib/tests/test_compare_images.py9sz2test_image_comparison_expect_rms..ZexpectedT)tolZ in_decoratorNZrmsg-C6?)abs)maprrrshutilcopyfilerr) Zim1Zim2rZ expect_rmsZ baseline_dirZ result_dirZ result_im1Z result_im2resultsr r r test_image_comparison_expect_rms s- r) ZpathlibrrZpytestrZmatplotlib.testing.comparerrZmatplotlib.testing.decoratorsrZmarkZ parametrizerr r r rs   !