o i&I@sddlZddlmZddlZddlZddlZddlmZddlZddl Z ddl Z ddl Z ddl Z zddlm Z WneyGddlmZ YnwddlZddlZddlZddlZddlmZddlmZddlmZdd lmZd d lmZmZmZd d lmZej d dZ!Gddde j"Z#d-ddZ$ddZ%ddZ&ddZ'ddZ(ddZ)ddZ*Gdd d Z+d!d"Z, #  $d.d%d&Z-d'dd(d)d*Z.d+d,Z/dS)/N StrictVersion)Path) nullcontext) ExitStack)cbook)ft2font)pyplot)ticker)comparable_formatscompare_imagesmake_test_filename)ImageComparisonFailurec cstjj}zUtt dVWdn1swYWdn1s.wYWtjjtjj|t ddSWtjjtjj|t ddStjjtjj|t dw)Nall) matplotlibunitsregistrycopywarningscatch_warnings rc_contextclearupdatepltclose)orig_units_registryrX/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/matplotlib/testing/decorators.py _cleanup_cms$     rc@s(eZdZdZeddZeddZdS)CleanupTestCasezAA wrapper for unittest.TestCase that includes cleanup operations.cCst|_dSN)r __enter___cmclsrrr setUpClass,szCleanupTestCase.setUpClasscCs|jddddSr!)r#__exit__r$rrr tearDownClass0szCleanupTestCase.tearDownClassN)__name__ __module__ __qualname____doc__ classmethodr&r(rrrrr *s  r cs,fdd}tr|}ddg|S|S)z A decorator to ensure that any global state is reset before running a test. Parameters ---------- style : str, dict, or list, optional The style(s) to apply. Defaults to ``["classic", "_classic_test_patch"]``. csBtrtfdd}|Stfdd}|S)Nc ?s~t1tj|i|EdHWdn1s wYWddSWddS1s8wYdSr!rrstylecontextargskwargsfuncr/rrwrapped_callableIsPz7cleanup..make_cleanup..wrapped_callablec svt.tj|i|Wdn1swYWddSWddS1s4wYdSr!r.r1r4rrr6NsP)inspectisgeneratorfunction functoolswraps)r5r6r/)r5r make_cleanupGs  zcleanup..make_cleanupclassic_classic_test_patch)callable)r/r<resultrr;rcleanup5s rAcCsV|durdSt|tr||f}dd|D}ttj}|d|ko(|dkSS)NTcSsg|]}t|qSrr).0xrrr dsz*check_freetype_version..rr ) isinstancestrrr__freetype_version__)verfoundrrrcheck_freetype_version^s   rJcCs0ddl}d|tjf}|jjt| |tddS)NrzAMismatched version of freetype. Test requires '%s', you have '%s'F)reasonraisesstrict)pytestrrGmarkxfailrJr)required_freetype_versionrNrKrrr_checked_on_freetype_versionjsrRc Cs|dt}|D]6}|d|j||j||j||j|z|j ||j |Wq t yCYq wdS)N) suptitler NullFormatterget_axes set_titlexaxisset_major_formatterset_minor_formatteryaxiszaxisAttributeError)figurenull_formatteraxrrrremove_ticks_and_titlests         racCsFd}t|||dd}|r!dD] }tj||||<qtd|dS)NT) in_decorator)actualexpecteddiffzEimages not close (RMS %(rms).3f): %(actual)s %(expected)s %(diff)s)r ospathrelpathr)rdrctol__tracebackhide__errkeyrrr_raise_on_image_differencesrmcCs$ddl}|jj|tvd|dS)Nrz&Cannot compare {} files on this system)rK)rNrOskipifr format) extensionrNrrr_skip_if_format_is_uncomparables rqcCsdddl}t|tr|}g}nt|tr|j\}g|j}n|j\}|jg}|j|g|t |dS)Nr)marks) rNrErFtuplevaluesrrr2rOparamrq)rprNnamerrrrr$_mark_skip_if_format_is_uncomparables   rwc@s.eZdZdZddZddZdddd Zd S) _ImageComparisonBasez Image comparison base class This class provides *just* the comparison-related functionality and avoids any code that would be specific to any testing framework. cCs,||_t|\|_|_||_||_||_dSr!)r5_image_directories baseline_dir result_dirri remove_textsavefig_kwargs)selfr5rir|r}rrr__init__s  z_ImageComparisonBase.__init__c Cs|j|}|d|}|dkr|s|d}t|j|jd}z6tt t |Wdn1s9wYz t ||WW|StyYt ||YW|Swtyp}z td|d||d}~ww)N.epsz.pdfrdzMissing baseline image z0 because the following file cannot be accessed: )rz with_suffixexistsrr{rv contextlibsuppressOSErrorrfremovesymlinkshutilcopyfiler)r~baselinerp baseline_pathorig_expected_pathexpected_fnamerkrrr copy_baselines8      z"_ImageComparisonBase.copy_baselineF_lockc Csd}t|}t|}|jrt||j|d|}|j} |dkr2| ddddd|r9t |nt } | |j |fi| |||} t| ||jWddS1s`wYdS)NTrpdfmetadata)CreatorProducer CreationDate)r get_fignumsr^r|rar{rr}r setdefaultr _lock_pathrsavefigrrmri) r~idxrrprrjfignumfig actual_pathr3lock expected_pathrrrcompares&    "z_ImageComparisonBase.compareN)r)r*r+r,rrrrrrrrxs rxc s:ddltttjjf dd}|S)z Decorate function with image comparison for pytest. This function creates a decorator that wraps a figure-generating function with image comparison code. rNc sttjdj tt fdd}tj }dj vr@|t dg7}dj vrN|t dg7}j |d}||_ tdg|j}||_|S)Nrpc sd}djvr ||d<djvr||d<td}tj|i|tdd|jdD}dur<}n|d}t t t |ksYJd t t t |t |D] \}} |j|| ||d q]dS) NTrprequest)rir|r}css|] }|jddkVqdS)rrpN)r2)rBmarkerrrr s  zO_pytest_image_comparison..decorator..wrapper.. parametrizebaseline_imagesz9Test generated {} images but there are {} baseline imagesr) parametersrxrtestingset_font_settings_for_testinganynode iter_markersgetfixturevaluelenrrro enumerater) rprr2r3rjimg needs_lockour_baseline_imagesrr)rr5old_sigr|r}rirrwrappers2    z<_pytest_image_comparison..decorator..wrapperrr pytestmark)r7 signaturer9r:rOrr/rRlistrrt Parameterreplace __signature__getattrr)r5rrnew_sig new_marks KEYWORD_ONLYr extensionsfreetype_versionrNr|r}r/ri)r5rr decorators"    !   z+_pytest_image_comparison..decorator)rNmaprwr7rr)rrrirr|r}r/rrrr_pytest_image_comparisons  8rFr=r>c Cs|dur-gtddd|D}|r-|durtdt|dkr$td|}dd|D}|dur5gd }|durmsz#image_comparison..z[When including extensions directly in 'baseline_images', 'extensions' cannot be set as wellr zaWhen including extensions directly in 'baseline_images', all baselines must share the same suffixcSsg|]}t|jqSr)rstemrrrrrDys z$image_comparison..pngrsvglgQ?)rrrirr|r}r/)filter ValueErrorrdictsysmaxsizer)rrrirr| savefig_kwargr/ baseline_extsrrrimage_comparison.s8=    rr)rrics2ttjtjdtjjfdd}|S)a Decorator for test cases that generate and compare two figures. The decorated function must take two keyword arguments, *fig_test* and *fig_ref*, and draw the test and reference images on them. After the function returns, the figures are saved and compared. This decorator should be preferred over `image_comparison` when possible in order to keep the size of the test suite from ballooning. Parameters ---------- extensions : list, default: ["png", "pdf", "svg"] The extensions to test. tol : float The RMS threshold above which the test is considered failed. Examples -------- Check that calling `.Axes.plot` with a single argument plots it against ``[0, 1, 2, ...]``:: @check_figures_equal() def test_plot(fig_test, fig_ref): fig_test.subplots().plot([1, 3, 5]) fig_ref.subplots().plot([0, 1, 2], [1, 3, 5]) z_-[]()csddl}t\}tddhjstd|jdfdd}dd j D}djvrG|t dg7}d jvrU|t d g7}j |d }||_ t d g|j}||_|S) Nrfig_testfig_refzwThe decorated function must have at least the parameters 'fig_ref' and 'fig_test', but your function has the signature extc sdjvr ||d<djvr||d<dfdd|jjD}zBtd}td}|||d||d |}|d |}||||t||d Wt|t|dSt|t|w) NrrrSc3s|] }|vr|VqdSr!r)rBc) ALLOWED_CHARSrrrs zJcheck_figures_equal..decorator..wrapper..test reference)rrrz -expected.)ri) rjoinrrvrr^rrmr) rrr2r3 file_namerrtest_image_pathref_image_path)rr5rr{rirrrs(        z7check_figures_equal..decorator..wrappercSsg|] }|jdvr|qS)>rr)rv)rBrurrrrDs  z:check_figures_equal..decorator..rrr)rNryr7rissubsetrrrOrrtrrrrr)r5rN_rrrrrrrri)r5rr{rrs*      z&check_figures_equal..decorator)setstringdigits ascii_lettersr7rr)rrirrrrcheck_figures_equals6rcCsLttj|jj}|jd|j}td|j}|jddd||fS)a= Compute the baseline and result image directories for testing *func*. For test module ``foo.bar.test_baz``, the baseline directory is at ``foo/bar/baseline_images/test_baz`` and the result directory at ``$(pwd)/result_images/test_baz``. The result directory is created if it doesn't exist. r result_imagesT)parentsexist_ok) rrmodulesr*__file__parentrresolvemkdir)r5 module_pathrzr{rrrrys  ryr!)NrNFNr)0rdistutils.versionrr9r7rfpathlibrrrrunittestrr ImportErrorrrmplmatplotlib.stylematplotlib.unitsmatplotlib.testingrrr rr rr r r exceptionsrcontextmanagerrTestCaser rArJrRrarmrqrwrxrrrryrrrrsZ          )   ;I Z X