o
    tBhQ                     @   s  d dl Z d dlZd dlmZ d dlZd dlZd dlmZ d dlmZm	Z	 d dl
Z
d dlZd dlmZmZ d dlmZ d dlmZ d dlZd dlmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z% dd Z&d	d
 Z'dd Z(dd Z)dd Z*dddie+dfdddde+dfdddde+dfdddie+dfdddde+dfdddde+dfgZ,g dZ-e
j.j/de,e-ddd Z0dd Z1d d! Z2d"d# Z3d$d% Z4d&d' Z5d(d) Z6e
j./d*e6d+d,d- Z7e
j./d.e6d/d0d1 Z8e
j./d2d3d4e9fd5d6e:d7d7fd8d9e9fd7ge9fgd:d; Z;d<d= Z<d>d? Z=e
j.j>ej?d@kdAdBdCdD Z@e
j.j>ej?d@ko:eA  dEdBdFdG ZBdS )H    N)Path)mock)cyclerCycler)_api_c_internal_utils)validate_boolvalidate_colorvalidate_colorlist_validate_color_or_linecolorvalidate_cyclervalidate_floatvalidate_fontweightvalidate_hatchvalidate_hist_binsvalidate_intvalidate_markeveryvalidate_stringlist_validate_linestyle_listify_validatorc                 C   s  t jddd t jddd t jd }t jd }t| d	 }|d
 t jd| id t jd | ks7J W d    n1 sAw   Y  t jd |ksOJ t j|d t jd dks_J W d    n1 siw   Y  t jd |kswJ t j|ddid t jd dksJ W d    n1 sw   Y  t jd |ksJ t ddidd }|  |  t | t jd dksJ d S )NtextF)usetexlines   )	linewidthtext.usetexlines.linewidthtest_rcparams.rczlines.linewidth: 33rc)fname!   ,   )r    r   c                   S   s   t jd dks	J d S )Nr   r"   mplrcParams r&   r&   u/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/matplotlib/tests/test_rcparams.pyfunc;   s   ztest_rcparams.<locals>.func)r$   r   r%   r   
write_text
rc_contextrc_file)tmpdirr   r   rcpathr(   r&   r&   r'   test_rcparams!   s0   




r.   c                  C   s~   t g ddddd} d }|t| ksJ d }|t| ks$J dd	gt| d
ks1J dgt| dks=J d S )N)zApple ChanceryTextilezZapf Chancerycursivez
sans-serifnormal   )font.cursivefont.familyzfont.weight	font.sizea  
RcParams({'font.cursive': ['Apple Chancery',
                           'Textile',
                           'Zapf Chancery',
                           'cursive'],
          'font.family': ['sans-serif'],
          'font.size': 12.0,
          'font.weight': 'normal'})z
font.cursive: ['Apple Chancery', 'Textile', 'Zapf Chancery', 'cursive']
font.family: ['sans-serif']
font.size: 12.0
font.weight: normalr3   r5   zi[vz]r4   family)r$   RcParamslstripreprstrsortedfind_alllist)r   expected_reprexpected_strr&   r&   r'   test_RcParams_classG   s   
	r@   c               	   C   s   t ddi} ddi}tt- tjtdd | | W d    n1 s(w   Y  W d    d S W d    d S 1 s@w   Y  d S )Nfigure.figsize)      @*   rB   rC      validatematch)r$   r7   pytestraises
ValueErrorwarnsUserWarningupdate)r   bad_dictr&   r&   r'   test_rcparams_updateh   s   PrP   c                	   C   s|   t t/ t jtdd tddi W d    n1 sw   Y  W d    d S W d    d S 1 s7w   Y  d S )NrF   rG   rA   rD   )rI   rJ   rK   rL   rM   r$   r7   r&   r&   r&   r'   test_rcparams_initq   s
   PrQ   c               	   C   sJ  t  G t  tj } | D ]	}| | tj|< qW d    n1 s%w   Y  t  ttj W d    n1 s?w   Y  W d    n1 sNw   Y  tt	 t
d  W d    n1 sgw   Y  tt	* t  dtjd< W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )NTzsvg.fonttype)r   'suppress_matplotlib_deprecation_warningr$   r*   r%   copydeepcopyrI   rJ   rK   r   )_copykeyr&   r&   r'   test_Bug_2543w   s,   





"rW   facecolorrinherit)rY   zaxes.facecolorgedge)rY   zaxes.edgecolor)zsame facecolorzinherited facecolordifferent facecolorzsame edgecolorzinherited edgecolorr^   zcolor_type, param_dict, target)idsc                 C   s   | d|d|  d< d|  d}t|' t \}}|jtddd | }t|j	| |ks5J W d    d S 1 s@w   Y  d S )NrY   zlegend.get_   test)label)
popr$   r*   pltsubplotsplotrangelegendgetattrlegendPatch)
color_type
param_dicttargetget_func_axlegr&   r&   r'   test_legend_colors   s   "rs   c                  C   4   dt jd< t jddgddg} |  dksJ d S )NrZ   zlines.markerfacecolorrE      )r$   r%   r   Line2Dget_markerfacecolorlnr&   r&   r'   test_mfc_rcparams      
rz   c                  C   rt   )NrZ   zlines.markeredgecolorrE   ru   )r$   r%   r   rv   get_markeredgecolorrx   r&   r&   r'   test_mec_rcparams   r{   r}   c                  C   s4   dt jd< t \} }|d}| dksJ d S )NrZ   zaxes.titlecolorTitle)r$   r%   re   rf   	set_title	get_color)rp   rq   titler&   r&   r'   test_axes_titlecolor_rcparams   s   

r   c                 C   sl   t | d }|jddd tjddd t|dd}W d    n1 s&w   Y  |d	d
ks4J d S )Nr   ztimezone: UTCz	UTF-32-BE)encodingzlocale.getpreferredencoding)return_valueTFtimezoneUTC)r   r)   r   patchr$   rc_params_from_fileget)r,   r-   r   r&   r&   r'   test_Issue_1713   s   r   c                  C   s   dD ]} | t jd< qd S )N)
pngjpegtiffrawrgbappmsgibmppbmsvgzanimation.frame_formatr#   )fmtr&   r&   r'   test_animation_frame_formats   s   r   c              !   c   s   t g dd dD dd dD R dd ddd	g fD d
tdg fdddgfddgfddgfddgfddgddgfdddgftddgddgftddgddgff	t tfdtfdtjftddgtjffd
t	t
dddd dddgddgdtdfD dd dD d
t	tdddd dddgddgdtdfD dd dD d
td td!d"ftd#d$d%gtd#d$d%gfd&td!d"td'g d( fd)td!d"td*g d+ fd,td!d"td#d$d%g ftd-d$d%gtd#d$d%gftdd.gd/td'dd.gffd0tfd1tfd2tfd3tfd4tfd5tfd6tfd7tfd8tfd9tfd:tfd;tfd<tfd=tfd>tftd?g d@tftd!g dAtffd
tdBdCtfdDtfdEtffd
tdFg dGfg dGg dGfdHdIgfg dJdKdLgftg dMtg dNgtg dMg dNgftg dMg dNgtg dMg dNgffdOtffd
tdPdQtfdRtfdStfdTtfdUtffd
tdVdWtfdXtffd
tdYdZd[d\g d+fg d+g d+ftd]td]ffdtffd
td^d_d`dadbg d+g d+ftdtd dd ftdddctdddcffd+tfg ddtfg detfg dftfdgtfdhtfditfdjtfdktfdltfdmtfdntfdotfdptfdltfdtft tffd
tdqdrdsdtdudvdwdxdydzd{d|d}d~ddgdddgffg ddg dffdddgfdddgffd	ddgfd	ddgfffdtfdtfddtfg d+tfdtfdddgftfd ddgftfdg d+ftfddgdftff	d
f}|D ],}|d }| r|d D ]\}}|||fV  qq|d D ]\}}|||fV  qĐqd S )Nc                 s       | ]}|d fV  qdS )TNr&   .0rp   r&   r&   r'   	<genexpr>       z/generate_validator_testcases.<locals>.<genexpr>)tyyesontrue1rE   Tc                 s   r   )FNr&   r   r&   r&   r'   r      r   )fnnoofffalse0r   Fc                 s       | ]}|t fV  qd S NrK   r   r&   r&   r'   r          aardvarkru   )	validatorsuccessfail za,babz
aardvark, zaardvark, ,)r   r   rE   )rE   ru   )r   c                 s       | ]	}|d dgfV  qdS )rE   ru   Nr&   r   r&   r&   r'   r          z1, 2      ?      @c                 s   r   r   r   r   r&   r&   r'   r      r   )r   r   rE   rE   ru   ra   c                 s   r   )r   r   Nr&   r   r&   r&   r'   r      r   z1.5, 2.5)r   r   c                 s   r   r   r   r   r&   r&   r'   r      r   )r   r   r   r   Nzcycler("color", "rgb")rY   rgb	linestyle---zW(cycler("color", ["r", "g", "b"]) +
                          cycler("mew", [2, 3, 5]))markeredgewidth)ru   ra      zcycler(c='rgb', lw=[1, 2, 3])r   r   z5cycler('c', 'rgb') * cycler('linestyle', ['-', '--'])lsr   )mew   zcycler("bleh, [])zCycler("linewidth", [1, 2, 3])z1cycler('c', [j.__class__(j) for j in ['r', 'b']])z2cycler('c', [j. __class__(j) for j in ['r', 'b']])z2cycler('c', [j.	__class__(j) for j in ['r', 'b']])z2cycler('c', [j.__class__(j) for j in ['r', 'b']])z9cycler('c', [j.__class__(j).lower() for j in ['r', 'b']])z1 + 2zos.system("echo Gotcha")z	import osz9def badjuju(a): return a; badjuju(cycler("color", "rgb"))zcycler("waka", [1, 2, 3])zcycler(c=[1, 2, 3])zcycler(lw=['a', 'b', 'c'])waka)rE   ra   r   )C1rZ   r\   ))--|r   )\oOr   )/+*/.xr   r   r   z--_   Xzr,g,b)rZ   r\   r   zr, ,rZ   )r   r\   bluer\   r   )rE   r   r   r   rE   r   fish)
Nonenoner   r   )AABBCCz#AABBCC)AABBCC00z	#AABBCC00)tab:bluer   )C12r   )z	(0, 1, 0))              ?r   )r   r   )z(0, 1, 0, 1))r   r   r   r   )r   rE   r   rE   r   ztab:verybluez(0, 1)z(0, 1, 0, 1, 0)z(0, 1, none)z(0, 1, "0.5")))	linecolorr   )markerfacecolorr   )mfcr   )markeredgecolorr   )mecr   linemarker)autor   )fdr   )10
   z1, 2, 3   )NNrE   rE   皙?r   )r   r   )r   r   ra   )rE   ru   333333?)r   ru   ra   )rE   ru   r   )r   皙?r   )r   ru   ra   )rE   r   r   )rE   r   )r   rE   abc)rE   r   )r   r   r   )r   r   )r   r   )solidr   )r   r   )dashedr   )-.r   )dashdotr   ):r   )dottedr   r   ) r   r   r   )DoTtEdr   )z1, 3)r   )rE   ra   gGz?i  r   g     |@)rE   ru   ra   r   )r   g       @g      @g      @s   dottedr   zutf-16r   r   r   )r   r   iternparraysetrK   r   MatplotlibDeprecationWarningr   r   r   r   r   r   r
   r	   r   r   aranger   slice	TypeErrorobjectr   encode)validvalidation_testsvalidator_dictr   argrn   
error_typer&   r&   r'   generate_validator_testcases   s  






	
6
	
	



  Vr  zvalidator, arg, targetTc                 C   sX   | |}t |tjrtj|| d S t |ts ||ksJ d S t|t|ks*J d S r   )
isinstancer   ndarraytestingassert_equalr   r=   )r   r   rn   resr&   r&   r'   test_validator_valid  s   
r  zvalidator, arg, exception_typeFc                 C   s8   t | | | W d    d S 1 sw   Y  d S r   )rI   rJ   )r   r   exception_typer&   r&   r'   test_validator_invalid  s   
"r
  zweight, parsed_weight)boldr  BOLD)d   r  )100r  r  )g4@   z20.6c                 C   sT   |t u r tt  t|  W d    d S 1 sw   Y  d S t| |ks(J d S r   )rK   rI   rJ   r   )weightparsed_weightr&   r&   r'   test_validate_fontweight  s
   
"r  c                  C   s2   dd t jD } | D ]}tt j| tsJ q
d S )Nc                 S   s   g | ]}d |v r|qS )keymapr&   )r   kr&   r&   r'   
<listcomp>  s    z test_keymaps.<locals>.<listcomp>)r$   r%   r  r=   )key_listr  r&   r&   r'   test_keymaps  s   r  c                
   C   s   t jddidI t jd du sJ tt! t jdddd W d    n1 s,w   Y  W d    n1 s;w   Y  t jd du sIJ W d    d S 1 sTw   Y  d S )Nr   Fr   T)r   z	test.blah)r$   r*   r%   rI   rJ   KeyErrorr&   r&   r&   r'   test_rcparams_reset_after_fail  s   "r  linuxz
Linux only)reasonc                 C   sj   i t jdddt| d}ttj tjtj	ddg|dtj
d W d    d S 1 s.w   Y  d S )Nr   )DISPLAYWAYLAND_DISPLAY
MPLBACKENDMPLCONFIGDIR-cz^import matplotlib;matplotlib.use('tkagg');import matplotlib.pyplot;matplotlib.pyplot.plot(42);T)envcheckstderr)osenvironr:   rI   rJ   
subprocessCalledProcessErrorrunsys
executableDEVNULL)r,   r!  r&   r&   r'   test_backend_fallback_headless  s   "r,  headlessc                 C   sR   t d i tjdt| d}tjtjddg|dd}|	 
 dks'J d S )	Ntkinterr   )r  r  r   a  import matplotlib as mpl; sentinel = mpl.rcsetup._auto_backend_sentinel; assert mpl.RcParams({'backend': sentinel})['backend'] == sentinel; assert dict.__getitem__(mpl.rcParams, 'backend') == sentinel; import matplotlib.pyplot; print(matplotlib.get_backend())T)r!  universal_newlinesagg)rI   importorskipr$  r%  r:   r&  check_outputr)  r*  striplower)r,   r!  backendr&   r&   r'   test_backend_fallback_headful  s   
r6  )CrS   r$  pathlibr   r&  r)  unittestr   r   r   rI   
matplotlibr$   r   r   matplotlib.pyplotpyplotre   matplotlib.colorscolorsmcolorsnumpyr   matplotlib.rcsetupr   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r.   r@   rP   rQ   rW   to_rgbalegend_color_testslegend_color_test_idsmarkparametrizers   rz   r}   r   r   r   r  r  r
  rK   r   r  r  r  skipifplatformr,  display_is_validr6  r&   r&   r&   r'   <module>   s    @&!	





	 a



