U md @shddlmZddlZddlZddlmZddlmZddlmZej dddd d d d d gddZ dS))PathN)approx)compare_images)_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 Cs||tttdd\}}||}t|||t||||dd} |dkrb| dkstn$| dk snt| dt|ddkstdS) 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)Ndummyr r r ]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/matplotlib/tests/test_compare_images.py=z2test_image_comparison_expect_rms..T)tolZ in_decoratorNZrmsg-C6?)abs) chdirmaprrshutilcopyfilerAssertionErrorr) Zim1Zim2rZ expect_rmsZtmp_pathZ monkeypatchZ baseline_dirZ result_dirZ result_im2resultsr r r test_image_comparison_expect_rms s0  r) pathlibrrZpytestrZmatplotlib.testing.comparerZmatplotlib.testing.decoratorsrmarkZ parametrizerr r r rs"     !