o
    h                    @   s$=  d dl Z d dlmZmZ d dlZd dlmZ d dlmZ d dl	Z	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Zd dlZd dlmZ d dlmZ d dl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#m$Z$ d dl%m&Z& d dl'm(Z( d dl)m*Z* d dl+m,Z- d dl.m/Z0 d dl1m2Z3 d dl4m5Z6 d dl7m8Z8 d dl9m:Z: d dl;m<Z= d dl>m?Z@ d dlAmBZC d dlDmEZF d dlGmHZI d dlJmKZKmLZLmMZM d dlNmOZOmPZPmQZQ d dlRmSZS ePdgddd ZTdd ZUdd ZVeP dd ZWeP dd ZXeP dd  ZYd!d" ZZePdgdd#d$ Z[ePdgdd%d& Z\ePdgdd'd( Z]d)d* Z^ePdgdd+d, Z_eOg d-e` d.krHd nd/d0d1d2 Zad3d4 Zbd5d6 ZceOd7gd8d9 Zdd:d; Zeejfgd<d=d>d? Zhejfgd<d=ePdgd@dAdBdC ZieOdDge` d.krd ndEd0dFdG ZjdHdI ZkdJdK ZldLdM ZmdNdO ZnePdgddPdQ ZoeOdRgdSdTdUdV ZpejqrdWdXdY ZsejqrdWdZd[ ZtejqrdWd\d] ZuePdgdd^d_ ZveOd`gdSdadbdcdd ZweOdegdSdadbdfdg Zxdhdi Zydjdk Zzdldm Z{eOdngdSdTdodp Z|eOdqgdSdTdrds Z}dtdu Z~dvdw Zdxdy Zdzd{ ZeOd|gd}d~iddd Zdd ZePdgddd Zdd ZeOddgdd ZeOdgdaddd ZePdgddd Zdd ZeOdgdgdSdaddd ZeOdgdd ZeOdgdd0dd ZeOddgdSdTdd Zdd Zdd ZeOdgdSdTdd Zdd ZeOdgdaddd Zdd ZeOdgdadSddd Zdd ZePdgddd Zdd ZeOdgdd ZejqrdWejfgdg dePdgddd ZeOddgdSdadbdd ZeOdgdae` d.krd nddōddǄ ZddɄ ZeOdgdSdTdd̄ ZejfjgdedΡddfedddedΡdfeddddedΡfgg dҢdӍddՄ ZejfjgdedΡddfedddedΡdfeddddedΡfgg dעdӍddل ZeOdgdSe` d.krd ndd܍ddބ ZeOdgdadSddd ZeOdgdSdTdd ZeOdgdd ZeOdgdSdTdd Zdd ZeOdgdSdTdd ZeOdgdgddd ZeOdgddgdSddd ZejfgdddgePdgddd ZePdgddd ZeOd gdaddd ZeOdgdaddd Zdd Zdd	 Zd
d ZePdgddd ZePdgddd Zdd Zdd ZejfgdddSgePdgddd ZeOdge` d.krd ndd0dd ZeOdgdSdWdbdd ZeOdgdSdTdd  Zd!d" ZeOd#gdSdTd$d% ZeOd&gdSd'd܍d(d) ZeOd*gdSd+d܍d,d- ZeOd.gdSdTd/d0 ZeOd1gdSdTd2d3 ZeOd4gdWdSdd5d6 ZeOd7gdSdTd8d9 ZeOd:gdSdTd;d< ZeOd=d=gd>d? ZÐd@dA ZĐdBdC ZeOdDgdEdF ZeOdGgdHdI ZePdgddJdK ZePdgddLdM ZePdgddNdO ZePdgddPdQ ZːdRdS Z̐dTdU ZeOdVgdWdX ZΐdYdZ Zϐd[d\ ZАd]d^ Zѐd_d` ZҐdadb ZӐdcdd ZԐdedf ZejqrdWeP dgdh Zejfgdidjdkdjdlgdjfg dmg dng dog dodlfg dmg dng dpg dpdlfg dmg dndqg drdqfgdsdt Zאdudv Zؐdwdx ZeOdygdSdTdzd{ ZePdgdd|d} Zېd~d ZeOdgdSdTdd Zݐdd ZeOdgdSdTdd ZeOdgdSdTdd ZeOdgdd Zdd Zdd Zejfjgdej!jdd dd gg ddӍdd Zejfgdg dkfg gdkfg g gdfgdd Zejfgddej2jjfdejj(jfdej2jjfgdd Zdd ZePdgddd ZePdgddd ZePdgddd ZePdgddd Zdd Zdd Zdd Zdd Zdd ZeOdgdSdTdd ZeOdgdd ZePdgddd Zdd ZeOdgdSdadbdd ZeOdgdae` d.krd nddōdŐdƄ ZeOdǐdgdSdadbdȐdɄ ZeOdgdSdadbdːd̄ Zd͐d΄ ZG dϐdЄ dЃZd2ddќdҐdӄZedԐdՃZ ejfgde e deg dآgdٍfedڐdۍe deg dܢgdٍfedڐdאdٍe deg dܢgdٍfedאdݍe deg dآgdٍfedאdgdݍe dאdgeg dآg dߢgdٍfgdd Z[[ ejfgde dfedאdۍdfedڐdэdfedڐdgdэdڐdgfedڐddfeddэdfeddэdfedڐdސddfedאdڐdސddfedڐdݍdfedڐdސddfgdd Zdd ZdZdddgddgfg dg dgg dg dgfgZejfgdedd Zejfgdedd Z	dd Z
dd Zd d Zdd Zdd ZeOddge` d.k	rd nd/d0dd ZeOd	gdSdTd
d ZeP dd Zi dd i fddZeOdgd}didWddd ZeOdgd}didWddd ZeOdgd}didWddd ZeOdgd}didWddd ZeOdgdSd}didWd d!d"d# ZeOd$gd}didWd d%d&d' ZeOd(gdSd}didWd)d*d+ ZeOd,gdSd}d-idWd)d.d/ ZeOd0gdSd}didWd)d1d2 ZeOd3gdSd}didWd)d4d5 ZeOd6gdSd}didWd)d7d8 ZeOd9gdSd}didWd)d:d; ZeOd<gdSd}didWd)d=d> ZeOd?gdSd}didWd)d@dA Z eP dBdC Z!eOdDgdSd}didWd)dEdF Z"eOdGgdSd}didWd)dHdI Z#eOdJgdSd}didWd)dKdL Z$eOdMgdSd}didWd)dNdO Z%eOdPgdSd}didWd)dQdR Z&eOdSgdSd}didWd)dTdU Z'eOdVgdSd}didWd)dWdX Z(eOdYgdSd}didWd)dZd[ Z)eOd\gdSd}didWd)d]d^ Z*d_d` Z+dadb Z,eOdcgd}didWdddde Z-eOdfgd}didWddgdh Z.didj Z/eOdkdkgdldWdmdndo Z0ePdgddpdq Z1eOdrgd}didWddsdt Z2eOdugdSdWdbdvdw Z3eOdxgdSd}didWd)dydz Z4eOd{d|gdWdd}d~ Z5dd Z6eOdgd}d-idSdkdWddd Z7eOdgdSd}didWd)dd Z8eOdgdSd}didWd)dd Z9dd Z:dd Z;dd Z<dd Z=eOdgdSd}didWd)dd Z>eOddgdd Z?eOdgdd Z@eOdgdd ZAeOdgdd ZBeOdgdd ZCeOdgdd ZDeOdgdd ZEeOdgdd ZFeOdgdd ZGeOdgdd ZHeOdgdd ZIeOdgdd ZJeOdgdd ZKeOdgdd ZLeOdgdSdadbdd ZMdÐdĄ ZNdŐdƄ ZOdǐdȄ ZPdɐdʄ ZQePdgddːd̄ ZRePdgdd͐d΄ ZSdϐdЄ ZTdѐd҄ ZUdӐdԄ ZVeOg dբd֐dׄ ZWeOdgdgdSddِdڄ ZXdېd܄ ZYeP dݐdބ ZZdߐd Z[eOdgdd Z\dd Z]dd Z^dd Z_ePdgddd Z`dd Zadd Zbdd ZceP dd Zdejfgdg deg ddkgdd ZeePdgddd ZfeOddgdd ZgeOdgdd ZheOd gdSdTdd ZieOdgdd ZjeOdgdd ZkeOd	gdadSdd
d Zldd Zmdd Zndd ZoeOdgdadSddd ZpeOdgdd ZqeOdgdd ZreOdgdd ZseOdgdSdTdd  Ztd!d" Zud#d$ Zvd%d& Zwd'd( ZxeOd)gd*d+ Zyejfgd,d-g d.id/g d.id0g d1id2g d3id4g d5id6g d.ifePdgdd7d8 Zzejfgd9ddڐd:dސd;d:dfddڐd:dސd;d<dfddڐdސd=dڐdfddڐdސd=d>dfddڐd:d?d:d@fddڐd:d?d<dfdd:dސdAd:dfdd:dސdAd<dfddڐdBdڐd@fddڐdBd>dfdd:dCd:d@fdd:dCd<dDfddސdEdDdfddސdEdFdfdGdidDd@fdGdidFdDfgdHdI Z{dJdK Z|dLdM Z}dNdO Z~dPdQ ZeOdRgdSdTdSdT ZeOdUgdSdTdVdW ZeOdXgdSdTdYdZ Zd[d\ Zd]d^ ZeOd_gdSdTd`d_ ZeOdadagdSdTdbdc ZeOddgdSdTdedf Zejfgdgg dhdidj Zdkdl ZeOdmgdSdTdndo Zdpdq Zejfgdrg gg d dkggd dkgg ggejfgdsg dtdudv ZePdgddwdx ZeOdygdSdTdzd{ ZeOd|ge` d.krd nd}d0d~d ZeOdgdSdTdd ZeOddge` d.krd ndd0dd ZeP dd ZeP dd ZeP dd ZeP dd Zdd ZeOg ddgddd Zdd ZeOg ddgddd Zdd Zejfgdrdkddejdgejg ddgePdgddd ZeOdgdgdaddd ZeOddgdSdd܍dd ZeOdgdSdTdd Zdd ZeOg ddSddWddd ZeOg ddSddWddd ZeOg ddSddWddd Zdd ZePdgddd Zdd ZeOg ddSdd܍dd ZeOg dâdSdTdĐdń ZdƐdǄ ZePdgddȐdɄ ZeOdgdSe` d.krd ndd܍d̐d̈́ ZeOdΐdgdSdTdϐdЄ Zejfgdg dҢdӐdԄ ZdՐdք Zdאd؄ ZePdgddِdڄ Zejqrdadېd܄ Zdݐdބ Zejfgde=je=jdge=je=jdggdd Zdd Zdd Zdd Zdd Zdd Zejfgdedi dfedi dfedkdgfi dfee ddidffdd Zdd Zejfgd<d=dd ZÐdd ZĐdd ZeOd gdd0dd ZeOg ddgdadddd ZeOdgdaddōd	d
 ZeOdgdaddōdd ZeOdgdaddōdd ZeOdgdaddōdd ZeOdgdadEdōdd ZeOdgdd0dd ZeOdgdaddōdd Zΐdd Zϐdd  ZАd!d" Zѐd#d$ ZҐd%d& ZӐd'd( ZeP d)d* ZeP d+d, ZeOd-ge` d.krd nd.d0d/d0 Zאd1d2 ZePdgdd3d4 Zِd5d6 Zڐd7d8 Zېd9d: Zܐd;d< ZeOd=gd>d? Zސd@dA ZߐdBdC ZeOdDgd}dEiddFdG ZdHdI ZdJdK ZdLdM ZejfgdedNi dOfedPi dOfee djdQidOfee d:dQidOfeddkdkd=dRfeddjdkidRfedSi dTffdUdV ZdWdX ZdYdZ Zd[d\ Zd]d^ Zd_d` Zdadb Zdcdd Zdedf ZeOdggd}diddhdi ZeOdjgd}diddkdl Zdmdn Zdodp Zdqdr Zdsdt Zejfgdudvejjfdwejjfededfejjfg dxg dyfejjfejdzejdzfejjfgejfgdred{d|ejdϐddϡgd}d~ Zdd Zdd Zdd Zdd Z dd Zdd Zejfgd,e dd ZeOdgdSdTdd Zdd Zdd ZejfgddSdgdd ZeOdgdSdd܍dd ZeOdgdSdae` d.krAd ndddd Z	ejqrdWdd Z
ejdjd:gddd Zejg dddd Zdd Zdd Zdd Zdd ZejqrdWdd ZejqrdWdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZdÐdĄ ZdŐdƄ ZdǐdȄ Zdɐdʄ Zdːd̄ Z d͐d΄ Z!dϐdЄ Z"eOdgdd0dӐdԄ Z#eOdgd֐dׄ Z$eOdgdd0dِdڄ Z%eOdgdSdTdܐd݄ Z&dސd߄ Z'dd Z(dd Z)dd Z*ejfjgdg dg ddӍdd Z+dd Z,dd Z-dd Z.dd Z/dd Z0dd Z1dd Z2dd Z3dd Z4ejqrdWdd Z5dd  Z6dd Z7dd Z8ejfgdedkggdg idfedkggdg idfedkggd	g id
fedkggdg idfedkggdg idfedkgi dfedkggddkdgidfedkggddkdgidfedkggd	dkdgidfedkggddkdgidfedkggddkdgidfedkggdgdkdgidffdd Z9dd Z:dd Z;ejfgdddgejfgdddgd d! Z<d"d# Z=d$d% Z>d&d' Z?ePdgdd(d) Z@ePdgdd*d+ ZAd,d- ZBd.d/ ZCd0d1 ZDeOd2gdSdadbd3d4 ZEd5d6 ZFd7d8 ZGejfgd9ddSgejfgd:ddSgd;d< ZHd=d> ZId?d@ ZJdAdB ZKdCdD ZLdEdF ZMdGdH ZNeOdIgdadSe` d.krd nd.dJdKdL ZOG dMdN dNeFjPZQeOdOgdae` d.krd ndPdōdQdR ZRdSdT ZSdUdV ZTdWdX ZUdYdZ ZVd[d\ ZWeOd]gdSdadbd^d_ ZXd`da ZYdbdc ZZddde Z[dfdg Z\dhdi Z]djdk Z^ePdgddldm Z_dndo Z`dpdq Zadrds Zbdtdu Zcdvdw Zddxdy Zedzd{ Zfd|d} Zgd~d Zhdd Zidd Zjdd Zkdd Zldd Zmdd Zndd Zodd Zpejfgdd=ejfgdddd ZqePdgddd ZrejqrdWdd Zsdd Ztdd Zudd Zvdd Zwejfgddd e-jxD g d ejqrdWdd Zydd Zzdd Z{ePdgddd Z|ePdgddd Z}dd Z~ejfgdg ddd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZeOdgdĐdń ZdƐdǄ ZdȐdɄ Zejfgdʐdːd̐d̐jgd͐d΄ ZdϐdЄ Zdѐd҄ ZdӐdԄ ZdՐdք Zdאd؄ Zdِdڄ ZejqrdWdېd܄ Zdݐdބ Zdߐd Zejfgddejfgdrddedσigdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZePdgddd ZeOdgdaddd Zdd Zdd Zdd  Zdd ZeOdgdSdae` d.krd ndPddd ZejqrdWdd ZePdgddd	 Zd
d Zdd Zdd Zdd Zdd ZeOdgdWddd ZePdgdejfgdd=dd Zejfjejddυ dko"ejjdkdddd Zd d! ZeSeP d"d# ZePdgdd$d% ZePdgdd&d' ZeOd(ge` d.krid nd)d0d*d+ Zd,d- Zd.d/ ZePdgdd0d1 ZdS (3      N)
namedtupledequeDecimal)partial)product)SimpleNamespace)ma)cycler)
rc_contextpatheffects)Figure)Axes)Line2D)PathCollection)
HammerAxes)	PolarAxes)assert_allcloseassert_array_equalassert_array_almost_equal)image_comparisoncheck_figures_equalremove_ticks_and_titles)needs_usetexpng
extensionsc                 C   s   |   }|d d S NF)subplotsset_visiblefig_testfig_refax r$   n/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/matplotlib/tests/test_axes.pytest_invisible_axes5   s   r&   c                  C   sD   t  \} }|d |d | dksJ | dks J d S )Nzx labelzy label)pltr   
set_xlabel
set_ylabel
get_xlabel
get_ylabelfigr#   r$   r$   r%   test_get_labels;   s
   

r.   c                  C   sH   t  \} }|d |d |d |d t|dks"J d S )NlabeltitlexyzH<Axes: label='label', title={'center': 'title'}, xlabel='x', ylabel='y'>)r'   r   	set_label	set_titler(   r)   reprr,   r$   r$   r%   	test_reprC   s   



r6   c                 C   s   |   }|jddgddgddgdd}|  |jddd |jdd	d | |}|jd
dd |  }|jddgddgddgdd}|  |jddd	d |jddd	d ||}|jd
dd	d d S )N      scattercr/   Y LabeltoplocX LabelrightZ Labelr2   har1   rD   r   r9   legendr)   r(   colorbarr3   r!   r"   r#   sccbarr$   r$   r%   test_label_loc_verticalN   s   

rL   c                 C   s   |   }|jddgddgddgdd}|  |jddd |jdd	d | j|d
d}|jdd	d |  }|jddgddgddgdd}|  |jddd	d |jddd	d |j|d
d}|jddd	d d S )Nr7   r8   r9   r:   r<   bottomr>   r@   left
horizontalorientationrB   r   rC   rE   rF   rI   r$   r$   r%   test_label_loc_horizontala   s   rR   c                 C   s   t ddd5 |  }|jddgddgddgdd}|  |d |d	 | j|d
d}|d W d    n1 s@w   Y  | }|jddgddgddgdd}|  |jdddd |jd	ddd |j|d
d}|jdddd d S )NrA   r=   )zxaxis.labellocationzyaxis.labellocationr7   r8   r9   r:   r<   r@   rO   rP   rB   rC   rE   )	
matplotlibr   r   r9   rG   r)   r(   rH   r3   rI   r$   r$   r%   test_label_loc_rct   s$   



rT   c                  C   s   t  \} }|jddd |jddd |jj dksJ |jddd |jj dks/J |jddd |jj dks@J |jddd |jddd |jj dksXJ |jddd |jj dksiJ |jddd |jj dkszJ d S )Nz
Test labelrN   r>   centerrA   r=   rM   )r'   r   r(   xaxisr/   get_horizontalalignmentr)   yaxisr,   r$   r$   r%   test_label_shift   s   rY   c           	      C   s   t jd d}t jdd| }|d }|  }|j||d | }t j||ddt || }t 	| |d }||d | ||  }|
|dg| |jdddd d S )	N!N,   r   r7   maxlagsfullmoder2   xminxmax)nprandomseednormalcumsumr   acorr	correlatedotarangevlinesaxhline	r!   r"   Nxr1   r]   ax_testax_refnorm_auto_corrlagsr$   r$   r%   
test_acorr   s   ru   c           	      C   s   t jd d}t j|d  }t |t j}|d }|  }|j	||d | }t j
||ddt || }t | |d }||d | ||  }||dg| |jdddd	 d S )
NrZ   3   
   r7   r\   r^   r_   r   ra   )rd   re   rf   randrh   ceilastypeint64r   ri   rj   rk   rl   rm   rn   ro   r$   r$   r%   test_acorr_integers   s   r|   c                 C   s  t jd t d}d|d d< t j| |d}| d}|d | |d j|dd	d
 |d}|d j|ddd |d j	
  |d jg t |d d d dR ddi |d jd|d  |d  d d d d |D ]}|j	d qxd S )NrZ   i   r   r[   )    r}   r8   r7   .lower)markerorigingray_rnearest)cmapinterpolation
markersizerw   )aspectxlimylimboth)rd   re   rf   onesshufflereshaper   spyimshowrV   tick_topplotnonzerosetget_xlimget_ylimset_ticks_position)r!   r"   aaxs_testaxs_refr#   r$   r$   r%   test_spy   s$   



. r   c               	   C   sl   t  \} }ddidddfD ]$}tt |jtddfi | W d    n1 s.w   Y  qd S )Nr   r   osolid)r   	linestyle   )r'   r   pytestraises	TypeErrorr   rd   eye)r-   r#   unsupported_kwr$   r$   r%   test_spy_invalid_kwargs   s   r   c                 C   sT   t jd tjdd}|  | | }|| |j	
  |j	d d S )Nmpl20r}   r   )mplstyleuserd   re   rx   add_subplotmatshowr   rV   r   r   )r!   r"   r   rr   r$   r$   r%   test_matshow   s   

r   )formatter_ticker_001formatter_ticker_002formatter_ticker_003formatter_ticker_004formatter_ticker_005x86_64gX9v?)tolc                     sV  dd l m  m     dtjd<  fddtdD }  fddtdD } fddtdD }t 	 }|
d	 t 	 }|
d	 |j| |d
dd t 	 }|
d	 |j| |d
dd |
d t 	 }|j| |d
dd |j| |ddd |
d t 	 }|j| |d
dd |j| |ddd |
d |  d S )Nr      zlines.markeredgewidthc                    s   g | ]}| j  qS r$   )sec.0r1   unitsr$   r%   
<listcomp>       z)test_formatter_ticker.<locals>.<listcomp>rw   c                       g | ]}d | d  j  qS )      ?      ?kmr   r2   r   r$   r%   r         c                    r   )      ?      ?r   r   r   r$   r%   r     r   zx-label 001bluer   )colorxunitszx-label 003greenhourzx-label 004zx-label 005)matplotlib.testing.jpl_unitstesting	jpl_unitsregisterrS   rcParamsranger'   figurer   r(   r   autoscale_view)xdataydata1ydata2r#   r$   r   r%   test_formatter_ticker   s0   






r   c                  C   s   dd } t   }|jjsJ |jjsJ |jjsJ |jjs"J |j|  |jjr.J |jjs4J |jjs:J |jjs@J t	| }t
|j tj	sPJ |j j|jks[J d S )Nc                 S      dS )N r$   r1   posr$   r$   r%   	_formfunc     z4test_funcformatter_auto_formatter.<locals>._formfunc)r'   r   r   rV   isDefault_majfmtisDefault_minfmtrX   set_major_formattermtickerFuncFormatter
isinstanceget_major_formatterfunc)r   r#   targ_funcformatterr$   r$   r%   !test_funcformatter_auto_formatter  s    

r   c                  C   s   d} t   }|jjsJ |jjsJ |jjsJ |jjs J |j|  |jjs,J |jjs2J |jjs8J |jjr>J t	| }t
|j tj	sNJ |j j|jksYJ d S )Nz	{x}_{pos})r'   r   r   rV   r   r   rX   set_minor_formatterr   StrMethodFormatterr   get_minor_formatterfmt)formstrr#   targ_strformatterr$   r$   r%   &test_strmethodformatter_auto_formatter8  s    

r   twin_axis_locators_formattersc                  C   s
  t jddddd} t t j|  d }t|}tg d}t }|ddd}|dd	gddg |j	
| |j	| |j	td
 |j	tg d |j
t  |jtg d |jtd |jtg d |  |  d S )Nr   r7      T)numendpoint       @皙?皙?333333?r   d   z%08.2lf)tricksmindjedi)   #   7   K   z%05.2lf)r;   3pr   )rd   linspacesinpir'   FixedLocatorr   r   r   rX   set_major_locatorset_minor_locatorr   FormatStrFormatterr   FixedFormatterrV   LinearLocatortwinytwinx)valslocsmajlminlr-   ax1r$   r$   r%   "test_twin_axis_locators_formattersQ  s"   
r  c                  C   s   t  \} }| }| }t   |j rJ |j r J |  |  |j r/J |j r6J |j	 s=J |j sDJ |j rKJ |j	 rRJ |j sYJ |j s`J |j	 sgJ d S N)
r'   r   r
  r	  drawrV   get_visiblepatchclarX   )r-   r#   ax2ax3r$   r$   r%   test_twinx_claj  s"   r  twinr1   r2   c                 C   s   |  d}d|  }ddg}ddg}t  }| }||| t||jd us)J t|| }t||jd us9J t||jt||ju sGJ d S )Naxisr  01r   b)r   r   r   getattrr   )r  	axis_name	twin_funcr   r  r-   r  r  r$   r$   r%   test_twin_units  s   

 r"  RQ?)r   r   c                 C   sF  d| }d| d}t dd}| ddd}t|| }t||d ||| | ddd}t||d t|| }||| dD ]Q}|dd|}	t|	|d |	|| tjj}
|tj	j
|
d	d	gd	dgd	dgddgddgdd	gdd	gd	d	gg|
j|
jgd
 |	jdtjd tjd dd qGt|  t| d S )Nr  set_scaler7   r   r8   logr7   r8   r      nonezaxes.edgecoloraxes.linewidth
projecting)	transform	facecolor	edgecolor	linewidthcapstyle)rd   rl   r   r  r   rS   pathPath
add_artistpatches	PathPatchMOVETOLINETO	transAxesr   r   r   )r!   r"   r  r!  	set_scaler1   rq   ax_twinirr   r2  r$   r$   r%   test_twin_logscale  sD   
r<  ztwin_autoscale.pngg;On?c                  C   s   t g d} d|  }t g d}d| }t }|jdddd}|j| |dd	d
 t|}|j||ddd |dd |dd d S )Nr   r   r7   r   r   r7   r8   r8   r   r   r7   r7   F)autoscalex_onautoscaley_onr   rw   r   lwr--r   rC  r   )rd   arrayr'   r   add_axesr   r
  margins)r1   r2   x2y2r-   r#   r  r$   r$   r%   test_twinx_axis_scales  s   
rK  c                  C   st   t  \} }| }|d | }| sJ | rJ | }|d | }| s2J | r8J d S r   )r'   r   r
  set_autoscalex_onget_autoscalex_onr	  set_autoscaley_onget_autoscaley_on)r-   r#   ax_x_onax_x_offax_y_onax_y_offr$   r$   r%   #test_twin_inherit_autoscale_setting  s   

rT  c                  C   s  t d} |  }| rJ | rJ tjd}|| | r&J | s,J |  t	ddtj
 d}||t| | rIJ | rOJ |  || t   | rbJ | shJ | jD ]}|  qkt d}t jd|d}|jd | sJ ||t| |  | rJ |  || ||t| |  | sJ t |  d S )	Nr   )r   r   r8   r         shareyT)r'   r   gcaxaxis_invertedyaxis_invertedrd   re   r   r  r   r  r   cos	autoscaleaxesremovesubplotrX   set_invertedclose)r-   r#   imgr1   ax0r  r$   r$   r%   test_inverted_cla  sB   






re  c            	         s  t jtdd G  fdddt} W d    n1 sw   Y  t jtdd G  fdddt}W d    n1 s=w   Y  G dd d| }G  fd	d
d
t}G  fdddt}G dd d|}t }| |||||fD ]}d ||g d} sJ d |   sJ qqd S )NzOverriding `Axes.cla`matchc                          e Zd Z fddZdS )z(test_subclass_clear_cla.<locals>.ClaAxesc                       d d S NTr$   selfcalledr$   r%   r       z,test_subclass_clear_cla.<locals>.ClaAxes.claN)__name__
__module____qualname__r  r$   rm  r$   r%   ClaAxes      rs  c                          e Zd Z fddZ  ZS )z-test_subclass_clear_cla.<locals>.ClaSuperAxesc                       dt    d S rj  )superr  rk  	__class__rn  r$   r%   r  %     z1test_subclass_clear_cla.<locals>.ClaSuperAxes.cla)rp  rq  rr  r  __classcell__r$   rm  ry  r%   ClaSuperAxes$      r}  c                   @      e Zd ZdS )z+test_subclass_clear_cla.<locals>.SubClaAxesNrp  rq  rr  r$   r$   r$   r%   
SubClaAxes*      r  c                       rh  )z*test_subclass_clear_cla.<locals>.ClearAxesc                    ri  rj  r$   rk  rm  r$   r%   clear.  ro  z0test_subclass_clear_cla.<locals>.ClearAxes.clearN)rp  rq  rr  r  r$   rm  r$   r%   	ClearAxes-  rt  r  c                       ru  )z/test_subclass_clear_cla.<locals>.ClearSuperAxesc                    rv  rj  )rw  r  rk  rx  r$   r%   r  3  rz  z5test_subclass_clear_cla.<locals>.ClearSuperAxes.clear)rp  rq  rr  r  r{  r$   rm  r|  r%   ClearSuperAxes2  r~  r  c                   @   r  )z-test_subclass_clear_cla.<locals>.SubClearAxesNr  r$   r$   r$   r%   SubClearAxes8  r  r  Fr?  )r   warnsPendingDeprecationWarningr   r   r  )	rs  r}  r  r  r  r  r-   
axes_classr#   r$   rm  r%   test_subclass_clear_cla  s2   
r  c                  C   s*   t  D ]} d| jvrd| jvsJ qd S )Nr  r  )r   __subclasses__rr  __dict__)klassr$   r$   r%   !test_cla_not_redefined_internallyI  s
   
r  c                 C   st   t ddd |  }|ddgddg W d    n1 s!w   Y  | }|ddgddg |  d S )NT)zxtick.minor.visiblezytick.minor.visibler   r7   )rS   r   r   r   minorticks_onr!   r"   rq   rr   r$   r$   r%    test_minorticks_on_rcParams_bothQ  s   
r  autoscale_tiny_rangeTremove_textc                  C   sL   t dd\} }t|jD ]\}}dd|  }|ddgdd| g qd S )Nr8   rw   ir   r7   )r'   r   	enumerateflatr   )r-   axsr;  r#   y1r$   r$   r%   test_autoscale_tiny_range\  s
   r  defaultc                  C   s   t dd\} }|g d |jdddd |jdddd t| d t| d	 | s3J | s9J |	 s?J |jd d
 t| d t| d	 | sYJ | s_J |	 seJ d S )Nr7   r7   r8   r   r(  Tr1   F)enabler  tightr2   )g333333ÿg333333	@)r         @)r  )
r'   r   r   r]  r   r   r   rM  rO  get_autoscale_onr,   r$   r$   r%   test_autoscale_tighte  s   r  c                  C   s   t jdtd} tjdddd\}\}}|| |  || |  |jdd |jdd t  | d | d f}t	|
 | t	| | t	|
 | t	| | d	 | d f d S )
Nr   dtyper8   r7   Tsharex)r  r   r   )rd   rl   floatr'   r   loglogsemilogxr]  r  r   r   r   )r1   r-   r  r  limsr$   r$   r%   test_autoscale_log_shared}  s   r  c                  C   s   t  \} }|jddgddggdd t| d t| d d|_|  d	d|j  d
d|j  f}d	d|j	  d
d|j	  f}t| | t| | d|_|  t| d t| d d S )Nr   r7   r8   r   r   r   )      r   Fr  r   T)
r'   r   r   r   r   r   use_sticky_edgesr]  _xmargin_ymargin)r-   r#   r   r   r$   r$   r%   test_use_sticky_edges  s   r  c                 C   sZ   t ddd}| d}| jd|d}|| |d}|jd|d}|| d S )Nr   r   r   rU  rV  r  )rd   rl   r   r   
pcolormesh)r!   r"   Zrd  r  r$   r$   r%   test_sticky_shared_axes  s   


r  zsticky_tolerance.pngr   )r  r   c                  C   s   t dd\} }d}|jd jd|dd |jd jd|dd |jd jd| dd |jd jd| dd |jd jd| dd	 |jd jd| d
d	 |jd jd|dd	 |jd jd|d
d	 d S )Nr8   r   r   gffff&@)r1   heightrM   r7   gffff@gffff&)r2   widthrN   gffffr   )r'   r   r  barbarh)r-   r  r  r$   r$   r%   test_sticky_tolerance  s   r  zsticky_tolerance_cf.pngc                  C   s:   t  \} }ddg }}ddgddgg}|||| d S )NgGZP@g    `P@r   r7   r8   r   )r'   r   contourf)r-   r#   r1   r2   datar$   r$   r%   test_sticky_tolerance_contourf  s   r  c                   C   s<   t jtdd t  W d    d S 1 sw   Y  d S )Nz0 were givenrf  )r   r   r   r'   stemr$   r$   r$   r%   test_nargs_stem  s   
"r  c                  C   sN   t jtdd t } | dgdgd W d    d S 1 s w   Y  d S )Nz3 were givenrf  FirstSecondr   )r   r   r   r'   r`  rG   r#   r$   r$   r%   test_nargs_legend  s   "r  c                  C   sX   t jtdd t } | ddgg dg dg W d    d S 1 s%w   Y  d S )Nz2 were givenrf  r   r7   r   r8   r7   r8   r   )r   r   r   r'   r`  
pcolorfastr  r$   r$   r%   test_nargs_pcolorfast  s   "r  offset_pointsc                  C   sf   t ddd} t dt j |  }t }|jdddd}|j| |d	d
d\}|jdddddd d S )N              @{Gz?r   F)r   r   r   )autoscale_onr   r   r   purplerC  r   z	local maxr   r7   r  r   r   zoffset points)xyxycoordsxytext
textcoords)	rd   rl   r\  r  r'   r   r   r   annotate)tsr-   r#   liner$   r$   r%   test_basic_annotate  s   

r  zarrow_simple.pngc            
      C   s   d} d}d}t | ||}tdd\}}tt|j|D ];\}\}}|dd |dd |\} }}dtj	 | d }	|j
ddt|	t|	|	d	 | |||	d
 |	d
 d
 qd S )N)TF)r^   rN   rA   r   r(  r8      r   r   rw   )r  length_includes_headshapehead_starts_at_zero
head_widthhead_length)r   r'   r   r  zipr  set_xlimset_ylimrd   r  arrowr  r\  )
r  r  r  kwargsr-   r  r;  r#   kwargthetar$   r$   r%   test_arrow_simple  s&   
r  c                  C   s$   t  \} }|jdddddd d S )Nr   )r  )r'   r   r  _r#   r$   r$   r%   test_arrow_empty  s   r  c                  C   s@   t  \} }|dddd | dksJ | dksJ d S )Nr7   )皙?皙@)r'   r   r  r   r   r  r$   r$   r%   test_arrow_in_view  s   r  c                  C   sN   t  \} }|jdddd}|jd u sJ |jdddi d}|jd us%J d S )Nfoor  r8   r   )r  )r  
arrowprops)r'   r   r  arrow_patch)r-   r#   annr$   r$   r%   test_annotate_default_arrow  s
   r  c                  C   sp   t  \} }t|jj}ttjj}t|	 t|	 ks"J t
| | D ]
\}}||ks5J q+dS )z?Check that the signature of Axes.annotate() matches Annotation.N)r'   r   inspect	signaturer  
parametersmtext
Annotationlistkeysr  values)r-   r#   annotate_paramsannotation_paramsp1p2r$   r$   r%   test_annotate_signature%  s   r  zfill_units.pngdpi<   )savefig_kwargc                  C   s  dd l m  m}  |   | jdtdddd}d| j }| dd}tj	d	d
dd}t
|}tdd\}\\}}\}	}
|j|g|gddd g d}||| g dd |j|g|gddd ||||| || gg dd |	j|g|gddd |	|| d| j d| j d| j d| j gd |
j|g|gddd |
j|||| || gd| j d| j d| j d| j gdd |  d S )Nr   ET  r(     dt      $@g     @z
2009-04-27z
2009-04-29datetime64[D]r  r8   degred)yunitsr   r?  )r  r  g     V@r  r  Z   r   r-  )r   r   r   r   Epochdatetimer  Durationrd   rl   mdatesdate2numr'   r   r   fillautofmt_xdate)r   r  valuedayr	  dtnr-   r  r  r  ax4indr$   r$   r%   test_fill_units/  s4   


""r  c                   C   s   t jtdd tjdgdgddd W d    n1 sw   Y  t jtdd tjdgdgdd	d
 W d    n1 s?w   Y  t jtdd tjdgdgddd W d    n1 saw   Y  tjdgdgddd d S )Nzmarker .* redundantly definedrf  r   r   r1   r   z linestyle .* redundantly defined---r   zcolor .* redundantly definedrr   r   r)  )r   r   )r   r  UserWarningr'   r   errorbarr$   r$   r$   r%    test_plot_format_kwarg_redundantQ  s   r&  c                 C   sz   g d}t |}| }|  }|j||t |t |d^}}|ddg |j||t |t |ddgd d S )Nr  xerryerrr8   )r(  r)  dashes)rd   r  rY  r%  abs
set_dashes)r!   r"   r1   r2   rr   rq   r  r  r$   r$   r%   test_errorbar_dashes\  s   
"(r-  c                  C   s>   dd t dD } t \}}|j|  |  |  d d S )Nc                 S   s   i | ]}||qS r$   r$   )r   iir$   r$   r%   
<dictcomp>k  s    z/test_errorbar_mapview_kwarg.<locals>.<dictcomp>rw   r1   r2   r(  )r   r'   r   r%  r  r  )Dr-   r#   r$   r$   r%   test_errorbar_mapview_kwargj  s    r2  single_pointc                  C   s   dt jd< dt jd< td\} \}}|dgdgd |dgdgd dgdgd}td\} \}}|jd	d	d|d
 |jddd|d
 d S )Nr   lines.markerT	axes.gridr8   r   r7   r   r  r   r  r  )rS   r   r'   r   r   )r-   r  r  r  r$   r$   r%   test_single_pointp  s   

r8  zsingle_date.png)r   c                  C   s   dt jd< ttd} dg}dg}t dd\}}tt	j
 |d j||  |d	d
d W d    n1 s9w   Y  |d j||d	d
d d S )N)   zaxes.formatter.limitsz
0000-12-31g    X&Ag(\bPr8   r7   r   r   r"  r#  )r'   r   r  r  rd   
datetime64r   r   r  r   MatplotlibDeprecationWarning	plot_dater   )r	  time1data1r-   r#   r$   r$   r%   test_single_date  s   
r@  c                 C   s|   t dd}t dddd}| d}|d | |d | |d}|d jdd	d
 |d |  d S )Nrw   r7   r   r   r   r   r7   r8   r7   gQQ?)r   	   r   r   )rd   rl   r   r   r   r   ravel)r!   r"   rowcolr  r$   r$   r%   test_shaped_data  s   

rJ  c                  C   sZ   t jddgdtfdtfgd} t d}|d jdd| d |d	 jddd
| d d S )Nr7   r7   r8   r8   r   twosr  r8   r   r7  r7   r"  )rd   rF  r  r'   r   r   r   )ptsr  r$   r$   r%   test_structured_data  s   rO  aitoff_proj)r   r  r   c                  C   sz   t t j t jd} t t j d t jd d}t | |\}}tjdtddd\}}|  |j|j	|j	ddd	 d
S )z^
    Test aitoff projection ref.:
    https://github.com/matplotlib/matplotlib/pull/14451
       r8   )   @aitoff
projection)figsize
subplot_kwr   r(  )r   N)
rd   r   r  meshgridr'   r   dictgridr   r  )r1   r2   XYr-   r#   r$   r$   r%   test_aitoff_proj  s   
r^  axvspan_epochc                  C      dd l m  m}  |   | jdtdddd}| jdtdddd}| d| jd}t	
 }|j||d	d
d ||d|  |d|   d S Nr   r  r  r7      r     r   r         ?)r-  alphar  )r   r   r   r   r  r  r  r  convertr'   rY  axvspanr  r   t0tfr	  r#   r$   r$   r%   test_axvspan_epoch      rk  axhspan_epoch{Gz?c                  C   r`  ra  )r   r   r   r   r  r  r  r  rf  r'   rY  axhspanr  rh  r$   r$   r%   test_axhspan_epoch  rl  rp  zhexbin_extent.pngc                  C   sl   t  \} }tdd d}|\}}|j||g dd ||d}t  \} }|jddg d|d d S )	N  )r8     r   r   333333?ffffff?extentr  r1   r2   )rw  r  )r'   r   rd   rl   r   hexbinr-   r#   r  r1   r2   r$   r$   r%   test_hexbin_extent  s   
rz  c                  C   s   t  \} }tdd d}|\}}tjtdd |j||dd W d    n1 s.w   Y  tjtdd |j||dd W d    d S 1 sNw   Y  d S )	NrQ  )r8   rw   z)In extent, xmax must be greater than xminrf  r7   r   r   r7   rv  z)In extent, ymax must be greater than ymin)r   r7   r7   r   )	r'   r   rd   rl   r   r   r   
ValueErrorrx  ry  r$   r$   r%   test_hexbin_bad_extents  s   "r}  c                  C   sz   t  \} }|jtjdtjddddd}t|tjj	s!J t|j
tjjs+J |j
jdks3J |j
jdks;J d S )Nrw   r&  r8   r   normvminvmax)r'   r   rx  rd   re   rx   r   rS   collectionsPolyCollectionr  colorsLogNormr  r  )r-   r#   hexr$   r$   r%   test_hexbin_string_norm  s   $r  zhexbin_empty.pngc                  C   sV   t  \} }|g g  |jg g dd |jg g g tjd |jg g tdd d S )Nr&  bins)Creduce_C_functionrw   )r'   r   rx  rd   maxrl   r,   r$   r$   r%   test_hexbin_empty  s
   r  c                  C   s`   t  \} }tdd d}|\}}|j||g ddd}tddd}||d	 s.J d S )
N   r8   r   rs  r   )rw  picker  ,  r  r   )r'   r   rd   rl   r   rx  r   contains)r-   r#   r  r1   r2   hbmouse_eventr$   r$   r%   test_hexbin_pickable  s   r  zhexbin_log.pngc                  C   s   dt jd< tjd d} tj| }dd|  dtj|   }td|d	 }t  \}}|j||d
d
dtj	d}t 
| | jdksIJ d S )NFpcolormesh.snaprZ   順 r         @r  r8   r   r&  T)yscaler  	marginalsr  )i&-  r8   )r'   r   rd   re   rf   standard_normalpowerr   rx  sumrH   get_offsetsr  )nr1   r2   r-   r#   hr$   r$   r%   test_hexbin_log  s   

r  c                  C   sz   t ddd} t \}}|j| | dddd}t j| t ddgddgddgddgddgddgddgd	dgg d S )
Nr7   r     r&  r8   )xscaler  gridsizer   r   r   )	rd   	geomspacer'   r   rx  r   assert_almost_equalr  rF  )r1   r-   r#   r  r$   r$   r%   test_hexbin_log_offsets-  s"   r  zhexbin_linear.pngr   r  c                  C   s^   t jd d} t j| }dd|  dt j|   }t \}}|j||ddt jd d S )	NrZ   r  r   r  r  rw   r   T)r  r  r  )rd   re   rf   r  r'   r   rx  r  )r  r1   r2   r-   r#   r$   r$   r%   test_hexbin_linear?  s   
r  c                  C   sH   t dd\} }t \}}|j| |dddd}| dks"J d S )Nr  r  r&  r8   r   )r  r  r  )rd   rl   r   r'   r   rx  get_clim)r1   r2   r-   r#   r  r$   r$   r%   test_hexbin_log_climL  s   r  c           
   	   C   s   g d}t | \}}dgt| }g d}d}|  }| }	|	j||||dd |	d |j||dgt| dd d||d	 |d d S )
N)r   r   r  )   r   )r   r  r7   )      $rw   r  rw   )r:  r:  )rw  r  mincntr   c                 S   s   t | S r  )r  )vr$   r$   r%   <lambda>r      z>test_hexbin_mincnt_behavior_upon_C_parameter.<locals>.<lambda>)r  r  r  rw  r  )r  lenr   rx  set_facecolor)
r!   r"   
datapointsr\  r]  r  rw  r  rq   rr   r$   r$   r%   ,test_hexbin_mincnt_behavior_upon_C_parameterS  s.   
r  c                  C   s   t  \} }|  |g dg d | dksJ | dks$J t   t  \} }|  |g dg d | dksDJ | dksLJ t  \} }|d |	dd	 | d
kseJ d S )N)r  r8   r(  )r7   r8   r  r   )r(  r  r  )r  r(  )r   r  r&  rw   r7   )rw   r7   )
r'   r   invert_xaxisr   r   r   rb  invert_yaxis
set_yscaler  r,   r$   r$   r%   test_inverted_limitsz  s   
r  nonfinite_limitsc                  C   st   t dt jd} t jdd t | }W d    n1 sw   Y  t j| t| d < t \}}|	| | d S )Nr  r  ignoredivider8   )
rd   rl   eerrstater&  nanr  r'   r   r   r1   r2   r-   r#   r$   r$   r%   test_nonfinite_limits  s   r  plot_fun)r9   r   fill_betweenc                 C   s^   t jdddd}| }| }t|| g g  ||fD ]}t|| |tt|dd qd S )Nz
2010-01-01z
2011-01-01r  r  C0r#  )rd   rl   r   r  r   r  )r  r!   r"   r1   rq   rr   r#   r$   r$   r%   test_limits_empty_data  s   r  r   c                  C   s   dt jd< d} t| | f\}}|| d 8 }|| d 8 }t|d |d  ||  }t \}}|| d|i}t \}}|jd|d d S )Nr   image.interpolationr   r8   r"  r7  )rS   r   rd   indicessqrtr'   r   r   )Nr1   r2   r"  r-   r#   r  r$   r$   r%   test_imshow  s   

r  imshow_clipgףp=
?)r   r   c                  C   s   dt jd< d} t| | f\}}|| d 8 }|| d 8 }t|d |d  ||  }t \}}||| d g}t	|
 d | }|j||d d S )Nr   r  r   r8   r(  r   )	clip_path)rS   r   rd   r  r  r'   r   contourmtransformsTransformedPath	get_pathsget_transformr   )r  r1   r2   r"  r-   r#   r;   r  r$   r$   r%   test_imshow_clip  s   
r  c                  C   sf   ddgddgg} t  }tjtdd |j| tddd	d
d W d   dS 1 s,w   Y  dS )4Parameters vmin, vmax should error if norm is given.r7   r8   r   r(  LPassing a Normalize instance simultaneously with vmin/vmax is not supported.rf  rw   r   r   r~  N)r'   r^  r   r   r|  r   mcolors	Normalize)r   r#   r$   r$   r%   test_imshow_norm_vminvmax  s   "r  polycollection_joinstylec                  C   sf   t  \} }tddgddgddgddgg}tjj|gdd}|| |dd |	dd d S )Nr7   r8   (   )
linewidthsr   r   )
r'   r   rd   rF  r   r  r  add_collection
set_xbound
set_ybound)r-   r#   vertsr;   r$   r$   r%   test_polycollection_joinstyle  s   "
r  z	x, y1, y2rL  r   r  r8   )
2d_x_input2d_y1_input2d_y2_input)idsc                 C   J   t  \}}tt || || W d    d S 1 sw   Y  d S r  )r'   r   r   r   r|  r  r1   r  rJ  r-   r#   r$   r$   r%   test_fill_between_input     "r  z	y, x1, x2)
2d_y_input2d_x1_input2d_x2_inputc                 C   r  r  )r'   r   r   r   r|  fill_betweenx)r2   x1rI  r-   r#   r$   r$   r%   test_fill_betweenx_input  r  r  fill_between_interpolateg~jt?)r  r   c               	   C   s  t ddd} t dt j |  }dt dt j |   }tjdddd\}\}}|j| || |d	d
 |j| ||||kdddd |j| ||||kddd t j	|d}t jj
|d< |j| || |d	d
 |j| ||||kddd |j| ||||kddd d S )Nr  r8   rn  333333?r(  r7   Tr  blackr#  white/)wherer-  hatchinterpolater  )r  r-  r  r   r   r   )rd   rl   r  r  r'   r   r   r  r	   masked_greatermasked)r1   r  rJ  r-   r  r  r$   r$   r%   test_fill_between_interpolate  s(   
r  #fill_between_interpolate_decreasingc               	   C   s   t g d} t g d}t g d}tjdd\}}||| d ||| d |j| ||||k dd	d
d |j| ||||kdd	d
d |dd |dd d S )N)gfffff@i  i  )g"@r:  r  )g@gffffff@gffffff@)rE  rE  rW  ztab:redkr   T皙?r  r-  r  re  r  r   r      X  )rd   rF  r'   r   r   r  r  r  )r   r  profr-   r#   r$   r$   r%   (test_fill_between_interpolate_decreasing-  s   r  fill_between_interpolate_nanc                  C   s   t d} t ddt jdddddddg
}t ddddddddt jt jg
}t \}}|j| |d	d
 |j| |dd
 |j| ||||kdddd |j| ||||kdddd d S )Nrw   rR              r}   r   r  r;   r  r   Tr   r  r  )rd   rl   asarrayr  r'   r   r   r  r  r$   r$   r%   !test_fill_between_interpolate_nanB  s   
 "
r  z
symlog.pdfc                  C   sX   t g d} t g d}t \}}|| | |d |d |dd d S )N)r   r7   r8   r(  r  rE  r  r
  )i@B i  r  r   r   r   r   r   symloglinearr   i )rd   rF  r'   r   r   r  
set_xscaler  r  r$   r$   r%   test_symlogV  s   

r  zsymlog2.pdfc                  C   sn   t ddd} tdd\}}t|g dD ]\}}|| |  |jd|d |d	 q|d
 dd d S )N2   MbP?r   r7   )      4@r   r   r   r  r  )	linthreshTr   皙r   )	rd   rl   r'   r   r  r   r  r[  r  )r1   r-   r  r#   r  r$   r$   r%   test_symlog2b  s   r  c               	   C   s   g d} g d}t | |\}}t ||}t| tt| t| ||d dd df  t||t|d dd df  d S )N)            r  r  r   r   r   )r  g      r        r        пr   rd  r         ?r         ?r   r   )rd   rY  hypotr'   pcolorr  )r1   r2   r\  r]  r  r$   r$   r%   test_pcolorargs_5205o  s   
&r#  r  c                  C   s,  dt jd< d} tdd| }tdd| d }t||\}}t|t| }t|t| }|d }t||d }||  t	| }t
t|d	t| k |}t d
d\}	\}
}}|
j|||d dd df d	dd |j|||d dd df dddgd |j|||dd d S )NFr  r  r  r   r8   皙?r   r   r7   r   r   r  rC  
edgecolorsr  wgouraudshading)r'   r   rd   r   rY  r\  r  r!  minptpr	   masked_wherer+  r  r   r  )r  r1   r2   r\  r]  QxQzr  Zmr  r  r  r  r$   r$   r%   test_pcolormesh}  s   
 $(r1  pcolormesh_smallepsc                  C   sh  d} t dd| }t dd| d }t ||\}}t |t | }t |t | }|d }t ||d }||  t | }t	t 
|dt | k |}t	|dt | k |}	tdd\}
\\}}\}}|j|||d d	d d	f dd
d |j|||d d	d d	f dddgd |j|||dd |j|||	dd |
jD ]}|  qd S )Nr   r  r   r8   r$  r   r   r  r   r  r%  r  r'  r(  r)  )rd   r   rY  r\  r  r!  r+  r,  r	   r-  r+  r  r'   r   r  r^  set_axis_off)r  r1   r2   r\  r]  r.  r/  r  r0  Zm2r-   r  r  r  r  r#   r$   r$   r%   test_pcolormesh_small  s&    $(

r6  pcolormesh_alphapdf)r   r  c               
   C   s  dt jd< d} ttdd| tdd| d \}}|}|t| }t||d }||  t| }t	j
d d	}|td	}d
d
ttd	  |d d df< t|}t dd\}	\\}
}\}}|
|||fD ]}|tjdddg ddd qq|
j|||d dd df |dddd |j||||dddd |j|||d dd df |ddd |j||||ddd d S )NFr  r  r  r   r8   r   viridis   r   r   )r   r  )ru  r   r   r   r   )r-  zorderr   r  r  r7   )r   re  r*  r;  r(  )r   r*  r;  )r'   r   rd   rY  r   r  r!  r+  r,  r   	colormaps	resampledrl   r  ListedColormapr   	add_patchmpatches	Rectangler  )r  r\  r]  r.  Qyr  virr  r   r-   r  r  r  r  r#   r$   r$   r%   test_pcolormesh_alpha  s2   
$

&rD  z
dims,alphar  )r(  r   c                 C   sN   |   }tjdd|ftdd }|| |  }|j|d ddd|d	 d S )
Nr   r  r  r8   ).r   grayr   r7   )r   r  r  re  )r   rd   r   r  r  )r!   r"   dimsre  r#   r;   r$   r$   r%   test_pcolormesh_rgba  s
   
rG  c                 C   s`   t d}t d}t ||\}}|| }|  }|j|dd | }|j|||dd d S )Nr(  r:  r   r)  )rd   rl   rY  r   r  )r!   r"   r1   r2   r\  r]  r  r#   r$   r$   r%   test_pcolormesh_nearest_noargs  s   

rH  zpcolormesh_datetime_axis.pngc                     s  dt jd< t  } | jdddd tddd t fd	d
tdD }td}t	tdtd\}}|| }t 
d t |d d |d d |d dd df  t 
d t ||| tj|tj ddd}tj|d d tjf ddd}t 
d t |d dd df |d dd df |d dd df  t 
d t ||| |  D ]}| D ]}|d |d qqd S )NFr  r  \(\?333333?hspacer=   rM     r7   c                       g | ]
} t j|d  qS daysr  	timedeltar   dbaser$   r%   r         z1test_pcolormesh_datetime_axis.<locals>.<listcomp>rb  rQ     r      r   r        rA   r   )r'   r   r   subplots_adjustr  rd   rF  r   rl   rY  r`  r  repeatnewaxisget_axesget_xticklabelsset_haset_rotationr-   r1   r2   z1z2zr#   r/   r$   rV  r%   test_pcolormesh_datetime_axis  s0   


.

>

ri  zpcolor_datetime_axis.pngc                     s  t  } | jdddd tddd t fddtd	D }td	}ttd
td
\}}|| }t 	d t 
|d d |d d |d dd df  t 	d t 
||| tj|tj d	dd}tj|d d tjf d	dd}t 	d t 
|d dd df |d dd df |d dd df  t 	d t 
||| |  D ]}| D ]}|d |d qqd S )Nr  rI  rJ  rK  rM  r7   c                    rN  rO  rR  rT  rV  r$   r%   r     rX  z-test_pcolor_datetime_axis.<locals>.<listcomp>rb  rQ  rY  r   rZ  r   r[  r\  r]  rA   r   )r'   r   r^  r  rd   rF  r   rl   rY  r`  r"  r_  r`  ra  rb  rc  rd  re  r$   rV  r%   test_pcolor_datetime_axis  s.   

.

>

rj  c               	   C   s  d} t dd| }t dd| d }t ||\}}t ||d }t \}}tt |	||| W d    n1 s?w   Y  tt |	|||j
 W d    n1 s\w   Y  tt |j	|||d dd df dd W d    n1 sw   Y  tt |j	|||d dd df dd W d    n1 sw   Y  t j|d	< tt |	|||d dd df  W d    n1 sw   Y  t jd
d t jj||d	k d}W d    n1 sw   Y  tt |	|||d dd df  W d    n	1 sw   Y  g d}ddg}t ||\}}t |j}tjtdd |j	|||dd W d    n	1 sMw   Y  t ddd}t ddd}t ||\}}t |j}t jd t j|j}t j|j}	tjtdd9}
|j	|| ||	 |dd t|
d	ksJ |j	|d|  |d|	  |dd t|
dksJ W d    d S 1 sw   Y  d S )Nr  r  r   r8   r   r   r(  r)  r   r  invalidmask)ig  r   r7   r  rw   z.are not monotonically increasing or decreasingrf  autor7   r   rZ   )rd   r   rY  r!  r'   r   r   r   r   r  Tr  r|  r  r	   rF  zerosr  r  r$  re   rf   r  )r  r1   r2   r\  r]  r  r  r#   noise_Xnoise_Yrecordr$   r$   r%   test_pcolorargs   sj   $$
  "$ru  c                  C   s~   t jdd/ t ddd} t ddd}t jt|t| }t \}}|| || W d   dS 1 s8w   Y  dS )z
    Test that underflow errors don't crop up in pcolormesh.  Probably
    a numpy bug (https://github.com/numpy/numpy/issues/25810).
    raise)underr   r   r   r  N)	rd   r  rl   re   randnr  r'   r   r  r1   r2   rh  r-   r#   r$   r$   r%   test_pcolormesh_underflow_errorX  s   "rz  c                  C   s&  t ddd} t g d| j}|jjdu sJ t jj| |d}t	
| t ddd	} t ddd	}t | |\}}t dt j | t dt j |  }t jd	td
}d|d< t ||j}|jjdu skJ t jj||d}t	
||| t jj||d}	t jj||d}
t	|	|
| d S )Nr  r8   r   FTFFrm  r   r7   rw   r  Tr   )rd   rl   r   broadcast_tor  flags	writeabler	   rF  r'   r  r   rY  r  r  r\  rq  boolr"  )r1   xmaskmasked_xr2   r\  r]  r  rn  masked_Zmasked_Xmasked_Yr$   r$   r%   test_pcolorargs_with_read_onlye  s$   
(r  c                 C   s   |   }tdd}tdd}tjd tjdd}|j|||dd |  }|d d	 t|d  }|d d	 t|d  }|j|||d
d d S )Nr   rw   r   rZ   r8   rE  r  r)  r   r   r   rd   rl   re   rf   rx  r  diffr!   r"   r#   r1   r2   r  rI  rJ  r$   r$   r%   test_pcolornearest|  s   r  c                 C   s   |   }dd tdD }tdd}tjd tjdd}|j|||d	d
 |  }dd tdD }|d d t|d  }|j|||dd
 d S )Nc                 S   s   g | ]
}t j |d  qS )  r  fromtimestampr   r$   r$   r%   r     rX  z+test_pcolornearestunits.<locals>.<listcomp>rw   r   r   rZ   r8   rE  r  r)  c                 S   s    g | ]}t j |d  d qS )r   r  r  r   r$   r$   r%   r     s     r   r   )	r   r   rd   rl   re   rf   rx  r  r  r  r$   r$   r%   test_pcolornearestunits  s   r  c                  C   s   t  \} }tdd}tdd}tjd tjdd}tjt	dd |j
|||dd W d    d S 1 s;w   Y  d S )	Nr   rE  r   rZ   zDimensions of Crf  r  r)  )r'   r   rd   rl   re   rf   rx  r   r   r   r  r-   r#   r1   r2   r  r$   r$   r%   test_pcolorflaterror  s   "r  c                  C   sv   t  \} }ttdtd\}}|| }tjtdd |j|||dd W d    d S 1 s4w   Y  d S )Nr   r   z.*one smaller than Xrf  r  r)  )	r'   r   rd   rY  rl   r   r   r   r  r  r$   r$   r%   test_samesizepcolorflaterror  s   "r  snapFc           	      C   s   |   }tdd}tdd}tjd tjdd}|j||||d |  }|d d t|d	  }|d d t|d	  }|j||||d d S )
Nr   rw   r(  rZ   r   rE  )r  r   r8   r  )	r!   r"   r  r#   r1   r2   r  rI  rJ  r$   r$   r%   test_pcolorauto  s   r  	canonicalc                  C   s   t  \} }|g d d S )Nr  r'   r   r   r,   r$   r$   r%   test_canonical  s   r  zarc_angles.pngc            
      C   sJ  d} d}d}d}t dd\}}t|jD ]\}}|d d }|d }	|tj|| |dd	 |tj|| ||	|d
 ||t	
t	|	 |d  |d |t	
t	| |d  g|t	t	|	 |d  |d |t	t	| |d  g || | || | | d9 } |d9 }|d d |d d f}|d9 }qd S )Nr8   r7   )r   r   r   h  rE  -   r   re  )theta1theta2r   rw   )r'   r   r  r  r?  r@  EllipseArcr   rd   r\  deg2radr  r  r  )
r'  r  centrer%  r-   r  r;  r#   r  r  r$   r$   r%   test_arc_angles  s4   
r  arc_ellipsec               	   C   sH  d\} }d\}}d}t t d}|d t | }|d t | }t |}t t |t | gt |t |gg}	t |	||g\}}|| 7 }||7 }t }
|
j	ddd}|j
||d	d
d
ddd tj| |f|||dddd}|| |
j	ddd}|j
||d	dddd tj| |f|||dddd}|| d S )N)gRQ?gp=
ף?)r   r   r  r   rU  ro  )r   r   yellowr7   )re  r-  r.  r/  r;  r8   F)angler/  r  r;  rV  equalr   )re  r-  r.  r;  )rd   r  rl   r\  r  rF  rk   r'   r   r   r  r@  r  r?  )xcenterycenterr  r  r  r  r1   r2   rthetaRr-   r#   e1e2r$   r$   r%   test_arc_ellipse  s:   

r  c                  C   sb   t  \} }td}|jg dg d|d |jg dg d|d |jg dg d|d d S )Nr   r  r   r8   r7   r  )r(  r   r8   )r   r(  r   )r'   r   mmarkersMarkerStyler   r9   r%  )fixr#   mr$   r$   r%   test_marker_as_markerstyle  s
   
r  	markeveryc                  C      t ddd} t | t | d d  }t \}}|j| |ddd |j| |dd d	d
 |j| |dddd
 |j| |dddd
 |  d S )Nr   rw   r   r   r   r  r/   rU  mark allr  r/   r  mark every 10+r   rQ  mark every 5 starting at 10rd   r   r  r  r'   r   r   rG   r  r$   r$   r%   test_markevery  s   r  markevery_lineg{Gzt?c                  C   r  )Nr   rw   r   r   z-or  r  z-dr  r  z-sr  z-+r  r  r  r  r$   r$   r%   test_markevery_line%  s   r  markevery_linear_scalesr  c            
      C   s   d ddg dddgt dddd	d
dddg} d}tjt| | d |}d}tddd|  d| }t|d | }t| D ])\}}|| }|| }	t	
|||	f  t	dt|  t	j||ddd|d q@d S )NrR  r   rR  r:  r
  r   r   r   r   r  r   r   r   r   r  r   ?r   r7   )\(?rw   r8   r   markevery=%sr   r  r(  lsmsr  )slicerS   gridspecGridSpecr  rd   r   r  r  r'   r`  r0   strr   
casescolsgsdeltar1   r2   r;  caserH  rI  r$   r$   r%   test_markevery_linear_scales7  s(   
r  markevery_linear_scales_zoomedc            
      C   s   d ddg dddgt dddd	d
dddg} d}tjt| | d |}d}tddd|  d| }t|d | }t| D ]3\}}|| }|| }	t	
|||	f  t	dt|  t	j||ddd|d t	d t	d q@d S )NrR  r  r  r   r   r   r  r   r   r   r   r  r  r7   r  rw   r8   r   r  r   r  r(  r  )r  g@)r$  g333333?)r  rS   r  r  r  rd   r   r  r  r'   r`  r0   r  r   r   r   r  r$   r$   r%   #test_markevery_linear_scales_zoomedP  s,   

r  markevery_log_scalesc            
      C   s   d ddg dddgt dddd	d
dddg} d}tjt| | d |}d}tddd|  d| }t|d | }t| D ]3\}}|| }|| }	t	
|||	f  t	dt|  t	d t	d t	j||ddd|d q@d S )NrR  r  r  r   r   r   r  r   r   r   r   r  r  r7   r  rw   r8   r   r  r&  r   r  r(  r  )r  rS   r  r  r  rd   r   r  r  r'   r`  r0   r  r  r  r   r  r$   r$   r%   test_markevery_log_scalesk  s,   


r  markevery_polarc            	      C   s   d ddg dddgt dddd	d
dddg} d}tjt| | d |}tddd}dtj | }t| D ]+\}}|| }|| }t	j
|||f dd t	dt|  t	j||ddd|d q6d S )NrR  r  r  r   r   r   r  r   r   r   r   r  r  r7   r  r8   Tpolarr  r   r  r(  r  )r  rS   r  r  r  rd   r   r  r  r'   r`  r0   r  r   )	r  r  r  r"  r  r;  r  rH  rI  r$   r$   r%   test_markevery_polar  s&   
r  markevery_linear_scales_nansc            
      C   s   d ddg dddgt dddd	d
dddg} d}tjt| | d |}d}tddd|  d| }t|d | }tj |d d<  |dd < |dd< t	| D ])\}}|| }|| }	t
|||	f  t
dt|  t
j||ddd|d qSd S )NrR  r  r  r   r   r   r  r   r   r   r   r  r  r7   r  rw   r8   r   r  F   r  r   r  r(  r  )r  rS   r  r  r  rd   r   r  r  r  r'   r`  r0   r  r   r  r$   r$   r%   !test_markevery_linear_scales_nans  s*   
&r  marker_edgesc                  C   sz   t ddd} t \}}|j| t | ddddd |j| d t | ddddd |j| d	 t | ddd
dd d S )Nr   r7   rw   zy.      >@r"  )r  mewmecr   r   r8   r  )rd   r   r'   r   r   r  r1   r-   r#   r$   r$   r%   test_marker_edges  s
    $r  zbar_tick_label_single.pngc                  C   sP   t  } | jddddd ddd}t  \}} t  } | jdddd|d	 d S )
Nr   r7   edger  )align
tick_labelr6  r   r  )r  r  r  )r'   rY  r  r   )r#   r  r-   r$   r$   r%   test_bar_tick_label_single  s   
r  c                  C   s&   t  \} }|ddgtjdg d S )Nr   r7   r(  )r'   r   r  rd   r  r,   r$   r$   r%   test_nan_bar_values  s   r  c                  C   s   t  \} }|g g  d S r  )r'   r   r  r,   r$   r$   r%   test_bar_ticklabel_fail     r  zbar_tick_label_multiple.pngc                  C   0   t  } | jddgddgddgddgdd	 d S )
Nr7         @r8   r   r   r   r  rU   r  r  r  )r'   rY  r  r  r$   r$   r%   test_bar_tick_label_multiple     
r  z/bar_tick_label_multiple_old_label_alignment.pngc                  C   s:   dt jd< t } | jddgddgddgdd	gdd
 d S )NrU   zytick.alignmentr7   r  r8   r   r   r   r  r  )rS   r   r'   rY  r  r  r$   r$   r%   *test_bar_tick_label_multiple_old_alignment  s
   

r  c                 C   s`   |   }g d}g d}dd |D }dd |D }|j||dd |  }|j||dd d S )Nr   g @g333333@rS  r$  r  gffffff
@g@c                 S      g | ]}t |qS r$   r   r   r$   r$   r%   r         z+test_bar_decimal_center.<locals>.<listcomp>c                 S   r  r$   r   r   r$   r$   r%   r     r  rU   r  r   r  r!   r"   r#   x0y0r1   r2   r$   r$   r%   test_bar_decimal_center  s   r  c                 C   sl   |   }g d}g d}dd |D }dd |D }|j||g ddd |  }|j||g ddd d S )	Nr  r  c                 S   r  r$   r   r   r$   r$   r%   r     r  z,test_barh_decimal_center.<locals>.<listcomp>c                 S   r  r$   r   r   r$   r$   r%   r     r  )r   r   r7   r7   rU   r  r  r   r  r  r$   r$   r%   test_barh_decimal_center  s   r   c                 C   ^   g d}g d}g d}dd |D }|   }|j|||dd |  }|j|||dd d S )Nr  r  ru  g333333?r7   r8   c                 S   r  r$   r   r   r;  r$   r$   r%   r   
  r  z*test_bar_decimal_width.<locals>.<listcomp>rU   r  r  r  )r!   r"   r1   r2   w0r'  r#   r$   r$   r%   test_bar_decimal_width     r  c                 C   r  )Nr  r  r  c                 S   r  r$   r   r  r$   r$   r%   r     r  z,test_barh_decimal_height.<locals>.<listcomp>rU   r  r  )r!   r"   r1   r2   h0r  r#   r$   r$   r%   test_barh_decimal_height  r  r	  c                  C   R   t  } | jddgddgdddd}|D ]}| dksJ | d	ks&J qd S )
Nr7   r8   r(  r   r)  r"  re  r   r.  r   r   r   r   r7   r   r   r   r'   rY  r  get_facecolorget_edgecolorr#   rectsrectr$   r$   r%   test_bar_color_none_alpha!     r  c                  C   r
  )
Nr7   r8   r(  r   r"  r)  r  r  r  r  r  r$   r$   r%   test_bar_edgecolor_none_alpha)  r  r  zbarh_tick_label.pngc                  C   r  )
Nr7   r  r8   r   r   r   r  rU   )r  r  r  )r'   rY  r  r  r$   r$   r%   test_barh_tick_label1  r  r  c                  C   s^  t  \} }|jtddddtjddd |jtddddtjddtjddd t  \} }|jtddddtjddd	 |jtddddtjddtjddd
 t  \} }|jtdddtdddgtddgtjddd	 |jtdddtdddgtddgdd dD d	 |tdddtjddfgd dS )z>Smoketest that bar can handle width and height in delta units.  r7   r   r   hoursr  r8   )r(  r  )r  )r  r)  r   c                 S   s   g | ]}t j|d qS )r  rR  r   r  r$   r$   r%   r   M      z&test_bar_timedelta.<locals>.<listcomp>r'  rw   rQ  N)	r'   r   r  r  rS  r  rd   rF  broken_barhr,   r$   r$   r%   test_bar_timedelta9  s<   







r   c                  C   s   t t dt dt dg} t t dt dt dg}t \}}|jg d||  | d t|j t	j
s?J t \}}|jg d||  | d	 t|j t	j
s\J d
S )z+test that tickers are correct for datetimesz
2012-01-01z
2012-02-01z
2012-01-15z
2012-02-07z
2012-02-13z
2012-02-12)r   r7   r   )r  rM   )r  rN   N)rd   rF  r;  r'   r   r  r   rX   r   r  AutoDateFormatterr  rV   )startstopr-   r#   r$   r$   r%   test_bar_datetime_startS  s   r$  c                 C   sB   t jdd}| jdd| jdddj}t  tj||d d S )Nr   r8   z1/1/2000r7   )years)periodsfreq	positions)	rd   re   rx   
date_range
DateOffsetyearr'   r   boxplot)pdr  r%  r$   r$   r%   test_boxplot_dates_pandasc  s   r/  c                  C   s   t jdd} td\}}|d j| g dg dd |d j| g dd	d |d
 j| g dd |d j| dg dd |d j| dd	d |d j| dd |d j| g dd |d j| d	d |d |  d S )Nr   r   rE  r   )r   r   r   r   )	capwidthswidthsr7   r   r8   r0  r   r(  r  r1  r:  rR  )rd   re   rx   r'   r   r-  r  r-   r  r$   r$   r%   test_boxplot_capwidthsl  s   r5  c                 C   s   ddl m}m} t }|d}tdddg}t|dk r2||d tj	dd	  t|dk st
d
}t
||\}}t|d t|d f}	t
|	d |	d  }
|	|
_z|  ||||
}|j  W |  d S |  w )Nr   )register_matplotlib_converters deregister_matplotlib_converterso     r7   r:  r   x   )secondsr   )pandas.plottingr6  r7  r'   r   r   r  r  appendrS  rd   rl   rY  r  r  canvasr  )r.  r6  r7  r-   r#   timesy_vals	time_axisy_axisr  z_dataimr$   r$   r%   test_pcolor_regression}  s"   

rE  c                 C   s   |  g dg dg dg dd}| |g d |d< |ddg dg }|j}|d }|d }t \}}|j||dd	d
 |j||ddd d S )N)r  r  r  r7   r7   r7   r  )r,  monthr  r  )r,  rG  r  dater  rw   rU   r  oranger(  rB  )		DataFrameto_datetimegroupbyr  indexr'   r   r  r   )r.  dfmonthlydatesforecastbaseliner-   r#   r$   r$   r%   test_bar_pandas  s   rS  c                 C   sD   | j g dg ddg dd}t \}}|j|jd|jd d S )N)r   r   r  r   r  rt  )r1   r  r  rM  r   r  )rJ  r'   r   r  r1   r  r.  rN  r-   r#   r$   r$   r%   test_bar_pandas_indexed  s
   rW  c                 C   sh   |   }|  }ddg}ddg}ddg}tdD ]}|j|| || d|| d q|j|||d d S )	Nr7   r8   r   r1   r   r  )r   r  r  )r   r   r  )r!   r"   rq   rr   r1   r2   hatchesr;  r$   r$   r%   test_bar_hatches  s    rZ  )r1   r  r/   expected_labelscontainer_labelr1   r7   
_nolegend_r   r  r;   )rw   rQ  r   ABr  )r  r]  r]  bars)r]  r]  r]  c           	      C   sJ   t  \}}|j| ||d}dd |D }||ksJ | |ks#J d S )Nr  c                 S      g | ]}|  qS r$   )	get_label)r   r  r$   r$   r%   r     r  z#test_bar_labels.<locals>.<listcomp>)r'   r   r  rd  )	r1   r  r/   r[  r\  r  r#   bar_container
bar_labelsr$   r$   r%   test_bar_labels  s
   rg  c                  C   s   t  \} }tt |jddgddgg dd W d    n1 s$w   Y  t  \} }tt |jddgddgdgd W d    d S 1 sMw   Y  d S )Nr1   r2   r7   r8   )r\  r]  r  r  r\  )r'   r   r   r   r|  r  r  r$   r$   r%   test_bar_labels_length  s   "rh  c                 C   s   | j ddgdd| j ddgddfD ]}t|| t|j| t| t|j q| dg di}t| t|| d S )Nr7   r8   float64r  Float64rI  r  )Seriesr'   r   rM  rJ  )r.  r1   rN  r$   r$   r%   test_pandas_minimal_plot  s   

rl  hist_logc                  C   sH   t dddd } t d|  d|  g}t \}}|j|ddd d S )Nr   r7   r  r   FT)r  r&  )rd   r   concatenater'   r   hist)data0r  r-   r#   r$   r$   r%   test_hist_log  s   rq  c                 C   s   |  dd}| dd}tg dD ]B\}}|d|f d |d|f jdd|d |d|f jdd|d |d|f d |d d |f D ]}|jddd|d	 qHqd S )
Nr8   r   )r  step
stepfilledr   r&  r7   histtypeT)r&  ru  )r   r  r  ro  )r!   r"   r   r   r;  ru  r#   r$   r$   r%   test_hist_log_2  s   rv  c                  C   s   t d\} }|d jdgddggddd |d d |d jg dddd |d d | j  |d  |d  ksCJ d S )Nr8   r   r7   
barstackedrt  r&  r   r   r7   )r'   r   ro  r  r>  r  r   r-   r  r$   r$   r%   test_hist_log_barstacked	  s   
 rz  zhist_bar_empty.pngc                  C      t  } | jg dd d S )Nr  rt  r'   rY  ro  r  r$   r$   r%   test_hist_bar_empty	     r}  c                  C   s   t jd t t jjddddddt j} tj| ddd	}|d
 }t	dt
|D ]}||d   }||  }|d d |d d ksIJ q+d S )NrZ   r   r   rr  sizer   r7   r   )r  re  r8   )rd   re   rf   cliprg   rz   float16r'   ro  r   r  get_corners)r  r  bcr"  rleftrrightr$   r$   r%   test_hist_float16	  s   r  zhist_step_empty.pngc                  C   r{  )Nrr  rt  r|  r  r$   r$   r%   test_hist_step_empty'	  r~  r  zhist_step_filled.pngc            
      C   s   t jd t jdd} d}ddiddidd ii gd }d	gd
 dgd
  }tjdd
d\}}t|||jD ]"\}}}|j| |f|dd| |	| d|  |j
dd q7|d j}	tdd |	D sjJ d S )Nr   rr  r   rw   r  TFr8   rr  r(  rs  nrowsncolsru  stackedr  r  rM   r  c                 s   s     | ]}|  | kV  qd S r  )r  r  r   r   r$   r$   r%   	<genexpr>>	  s    z(test_hist_step_filled.<locals>.<genexpr>)rd   re   rf   rx  r'   r   r  r  ro  r4   r  r4  all)
r1   n_binsr  typesr-   r  kg_typer#   r4  r$   r$   r%   test_hist_step_filled.	  s   
r  zhist_density.pngc                  C   s6   t jd t jd} t \}}|j| dd d S )NrZ   rq  Tdensity)rd   re   rf   r  r'   r   ro  r  r-   r#   r$   r$   r%   test_hist_densityA	  s   r  c                  C   sV   t jd} | d}g d}tj||dd\}}}t j||dd\}}t|| d S )Ni  r   )r  r   r  r   r7   r   T)r  r  )rd   re   RandomStaterx  r'   ro  	histogramr   )rngr  r  mpl_heightsr  
np_heightsr$   r$   r%   test_hist_unequal_bins_densityI	  s   
r  c                  C   sd   t  dddt  dddgt  dddt  dddgg} t \}}|j| dd |j| dd d S )N  r7   r8   T)r  F)r  r'   r   ro  r  r$   r$   r%   test_hist_datetime_datasetsT	  s   r  bins_preprocessc                 C      | S r  r$   r  r$   r$   r%   r  ^	      r  c                 C   s   t | dS )Nr;  )rd   r  r  r$   r$   r%   r  _	  s    )r  zdatetime.datetimeznp.datetime64c              	   C   s   t  dddt  dddt  dddt  dddgt  dddt  dddt  dddt  dddgg}t  dddt  dddt  dddg}t \}}|j|| |dd	\}}}tj|tj	| |j|| |d
d	\}}}tj|tj	| d S )Ni  r7   r   r  r8   r   r	  T)r  r  F)
r  r'   r   ro  rd   r   r   r   rP  r  )r  r  
date_edgesr-   r#   r  r  r$   r$   r%    test_hist_datetime_datasets_bins\	  s   r  zdata, expected_number_of_histsc                 C   sJ   t | \}}}t|}|jdkrd|ksJ d S |jd |ks#J d S )Nr7   r   )r'   ro  rd   r  ndimr  )r  expected_number_of_histshistsr  r$   r$   r%   test_hist_with_empty_inputs	  s
   

r  zhisttype, zorderr  rr  rs  c                 C   sH   t   }|jddg| d |jsJ |jD ]
}| |ks!J qd S )Nr7   r8   rt  )r'   r   r   ro  r4  
get_zorder)ru  r;  r#   r  r$   r$   r%   test_hist_zorder	  s   

r  c                  C   s\   t  \} }tjtdd |jg dg ddd dd W d    d S 1 s'w   Y  d S )Nzbaseline=None and fill=Truerf  )r(  r   r7   r   r8   )r7   r8   r   r(  r   r  r   T)r-  rR  r  )r'   r   r   r  r$  stairsr,   r$   r$   r%   "test_stairs_no_baseline_fill_warns	  s   "r  c                 C   s  dd l m} tg d}tg d}| dd }|d j||d d |d j||d dd	 |d || |d j||dd
 |d || |d   |d j||dd
 |d   ddd}|dd }|d j	|t
||d fddi| |d j	t
|d ||fddi| |d j	|t
||d fddi| |d |j|d |d gd|d gfi | |d |j|d |d gd|d gfi | |d dd  |d j	t
|d ||fddi| |d |jd|d g|d |d gfi | |d |jd|d g|d |d gfi | |d dd  |d j	|t
||d fddi| |d |j|d |d gd|d gfi | |d |j|d |d gd|d gfi | |d   |d j	t
|d ||fddi| |d |jd|d g|d |d gfi | |d |jd|d g|d |d gfi | |d   d S )Nr   )r     r}   %   0   r}   rb  r(  )	r   r   r  r  r        @      @       @      "@r   r8   rR  r7   rO   )rR  rQ   rP   r(  r   miterbuttsolid_joinstylesolid_capstyler   	drawstyle
steps-post)matplotlib.lineslinesrd   rF  r   flattenr  semilogyr  r   r=  add_liner   r  r  )r!   r"   mlinesr2   r1   	test_axesr   ref_axesr$   r$   r%   test_stairs	  s@   
(((22(22(22(22r  c                 C   s  g dg d}}d}|  dd }|d j||dd |d j||d	dd
 |d j|||dd |d j|||d	dd | dd }|d j|t||d ddd |d dd  |d j|t||d ddd |d dd  |d j|t||d t	t
|d | ddd |d |d  |d j|t||d t	t
|d | ddd |d |d  d S )N)r7   r8   r   r(  r8   r   r7   r8   r   r(  r   r  r8   r   Tr  r7   rO   )rQ   r  )rR  r  r   )rR  rQ   r  r   post)rr  rC  )r   r  r  r  rd   r=  r  r  r  r   r  )r!   r"   r  r  bsr  r  r$   r$   r%   test_stairs_fill	  s,   ""r  c                 C   s  d}|   }|g d}|| |g d |jtdd d |g dtdd  |g d |d td t| d td	dsQJ t| d	 tds`J |jd
d | jd
ksoJ |  }|jg dd
d}|| d S )N)r  r(  r  r  r(  r8   edges)r7   r8   r7   r   r7   r  r  )	r   r  r  set_datard   rl   allcloseget_datarR  )r!   r"   r   test_axr  ref_axr$   r$   r%   test_stairs_update	  s    
 r  c                 C   sn   t g d}t g d}|  }|j||d d ddd}| }|j|t ||d fdd	i| d S )
N)r   r8   r   r   rw   )g|?5^?gV-?g+?r   r  r  r  r  r   r  r  )rd   rF  r   r  r   r=  )r!   r"   r1   r2   r  r   r  r$   r$   r%   test_stairs_baseline_None	  s   
(r  c                  C   s>   t   } | g dg |  dksJ |  dksJ d S )N*   )'   r  rC  )r'   r   r   r  r   r   r  r$   r$   r%   test_stairs_empty
  s   r  c                   C   sL   t jtdd tddgdtjdg W d    d S 1 sw   Y  d S )NNan values in "edges"rf  r7   r8   r   )r   r   r|  r'   r  rd   r  r$   r$   r$   r%   test_stairs_invalid_nan
  s   "r  c                   C   sH   t jtdd tddgddg W d    d S 1 sw   Y  d S )NSize mismatchrf  r7   r8   r   )r   r   r|  r'   r  r$   r$   r$   r%   test_stairs_invalid_mismatch
  s   "r  c                  C   s\   t ddgg d} tjtdd | jdtjdgd W d    d S 1 s'w   Y  d S )Nr7   r8   r>  r  rf  r  )r'   r  r   r   r|  r  rd   r  r  r$   r$   r%   test_stairs_invalid_update
  s   "r  c                  C   sZ   t ddgg d} tjtdd | jtdd W d    d S 1 s&w   Y  d S )Nr7   r8   r>  r  rf  r   r  )r'   r  r   r   r|  r  rd   rl   r  r$   r$   r%   test_stairs_invalid_update2
  s   "r  ztest_stairs_options.pngc               
   C   s  t g dt g dt} }| }t j|d< t \}}|j|d | dddd |j|| d d d	dd
dd |j|| ddddd |j|d | d d dddd
dd |j|d d d d d | d d	ddd dd |j|d d d d d | dddddd |j|d d d d d | d t 	dd t
|d!dd"dd# |j|d d d d d d d$ | d d d d%ddd&d'd(d) |jd*d+ d S ),Nr7   r8   r   r(  r   r  r7   r   r   Tr`  )r   r  r/   r  rO   ra  )r   r  rQ   r/   rI  r   r8   r  )r   r  rC  r/   r   r1  )r  rC  rR  rQ   r/   r      E)r   r  rC  rR  r/   r  r  r  F)rR  r   r  rC  r/   r   r     r   G)rR  r   r  r/   r  r  r  r  //H)r   r  rC  rR  r  r/   r   r>   )rd   rF  rz   r  copyr  r'   r   r  r   r  rG   )r1   r2   ynr-   r#   r$   r$   r%   test_stairs_options"
  s2   $
&.4r  ztest_stairs_datetime.pngc               	   C   sF   t jdd\} }|tdttdtd t jdd d S )NT)constrained_layout$   z
2001-12-27z
2002-02-02r   )rotation)r'   r   r  rd   rl   r;  xticks)fr#   r$   r$   r%   test_stairs_datetime:
  s   r  c                 C   sD   |   }|jg dddd |  }|jg ddd}|d d S )Nr  r  T)r   r  r  )r   r  	set_color)r!   r"   r  r  str$   r$   r%   test_stairs_edge_handlingC
  s
   r  c                  C   sH   t ddd} t ddd}t | t |d d t jf  }| ||fS )Nr  r      r:  )rd   r   r\  r  r`  )r1   r2   rh  r$   r$   r%   contour_datO
  s   "
r  contour_hatchingc               
   C   s@   t  \} }}t \}}|j| ||dg dtjd ddd d S )Nr:  )r  \r  r  rE  r   r   )rY  r   extendre  )r  r'   r   r  r   r<  ry  r$   r$   r%   test_contour_hatchingV
  s   
r  contour_colorbargHzG?c            	   
   C   s   t  \} }}t \}}|j| ||tdddtjd dddd}|j| ||td	d
ddgddd}|j| ||tddddgdd}|j	||d}|
| |j
|dd d S )Ngg7A`?r   RdBug333333rt  r   )levelsr   r  r  r  gg^I+r2   r   r8   )r  r  
linestylesr  r  r;   )r  r  r  r  F)erase)r  r'   r   r  rd   rl   r   r<  r  rH   	add_lines)	r1   r2   rh  r-   r#   cscs1cs2rK   r$   r$   r%   test_contour_colorbar_
  s(   
r  hist2dc                  C   s   dt jd< tjd tjdd d } tjdd }t  \}}|j| |ddd	 | |d
}t  \}}|jddd|dd d S )NFr  r   r   r8   r   rw   Tr  
rasterizedr  r1   r2   )r  r  r  )r'   r   rd   re   rf   rx  r   r  )r1   r2   r-   r#   r  r$   r$   r%   test_hist2dv
  s   

r  hist2d_transposec                  C   sX   dt jd< tjd tdgd } tjdd }t  \}}|j| |ddd	 d S )
NFr  r   r   r   r8   rw   Tr  )	r'   r   rd   re   rf   rF  rx  r   r  r  r$   r$   r%   test_hist2d_transpose
  s   
r	  c                  C   s>   t jd\} }t  }|tfD ]
}|j| |dd qd S )Nr  Tr  )rd   re   r'   r   r   r  )r1   r2   r#   objr$   r$   r%   test_hist2d_density
  s
   r  c                "   @   s  e Zd Zedgddddd Zedgddd	d
 Zedgddgddd Zedgddd Zdd Z	e
jdddide iddiddigdd Zdd Zejd d!d" Zd#d$ Zd%d& Zedgdd'd( Zedgdd)d* Zd+d, Zedgdd-d. Zedgdd/d0 Zd1d2gd3fd4d5d6g d7d8fd9d:gd3fd:gd; d3fd<gd= d8fd<gd; d8fd<gd> d?fg d@d8fg dAd8fg dBd3fg dCgd8fg dCgd= d?fg dCgd; d8fg dCgd> d?fg dDgd8fg dDgd= d?fg dDgd; d8fg dDgd> d?fg dDgd= g dCg d8fg dDdEdFgd?fg dDdEdFdGgd8fg dDdEdFdGg dHgd?fg dDdEd:gd3fg dDdEdFd:gd3fg dDdEdFdGd:gd3fgZe
jdIedJdK Zejd edgddLdM ZdNdO ZdPdQ Z d8S )RTestScatterr9   r   Tr  c                 C   s   t g dt g dg dg dg dd}t \}}|j|d d |d	 d |d
 |d d |j|d d |d	 d |d |d d |jdd	d
d|d d S )Nr   r(  r8   r  r8   r   r8   r   r"  r2   r  limer
  r         )0.50.60.7z0.8)r1   r2   r;   r  c2r1   r   r2   r;   r  r;   r  r  )r;   r  r  )rd   rF  r'   r   r9   )rl  r  r-   r#   r$   r$   r%   test_scatter_plot
  s   **zTestScatter.test_scatter_plotzscatter_marker.pngr  c           
   
   C   s   t jdd\}\}}}|jg dg dg dg dg ddd	 |jg dg dg dg dg dtjd
ddd	 d\}}|| tj }tddtj d}tt	|| | t
|| | g}	|jg dg dg dg dg d|	d	 d S )Nr   r  r  r  )r7   r   r   r2   r  r  )r  r  r  r   )r  r"  gr  r  )r;   r  r&  r   r   r=   	fillstyler  r   r8   rb  )r'   r   r9   r  r  rd   r  r   column_stackr\  r  )
rl  r-   rd  r  r  rxryarear  r  r$   r$   r%   test_scatter_marker
  s2   
zTestScatter.test_scatter_marker
scatter_2Dr   )r  r   c                 C   sP   t d}t d}t ||\}}|| }t \}}|j|||ddd d S )Nr   r8   r  face)r;   r  r&  )rd   rl   rY  r'   r   r9   )rl  r1   r2   rh  r-   r#   r$   r$   r%   test_scatter_2D
  s   

zTestScatter.test_scatter_2Dr   c           
      C   s   t g d}t g d}t dd |D }t dd |D }g d}g d}| }	|	j||||d | }	|	j||||d d S )	Nr  r  c                 S   r  r$   r   r  r$   r$   r%   r   
  r  z4TestScatter.test_scatter_decimal.<locals>.<listcomp>c                 S   r  r$   r   r  r$   r$   r%   r   
  r  r  r  r  )rd   rF  r   r9   )
rl  r!   r"   r  r  r1   r2   r;   r  r#   r$   r$   r%   test_scatter_decimal
  s   z TestScatter.test_scatter_decimalc                 C   s   t t tjddgddgddgd W d    n1 sw   Y  t t tjg dg dg dd W d    d S 1 sBw   Y  d S )Nr7   r8   r   r   r#  r  )r   r   r|  r'   r9   rk  r$   r$   r%   test_scatter_color
  s   "zTestScatter.test_scatter_colorr  r   rE  r  r  r   r  c                 C   s   d}t jt|d tjg g fi | W d    n1 sw   Y  t jt|d tjddgddgfdg i| W d    n1 sEw   Y  tjg g fdg i| tjddgddgfdddgi| d S )	Nz"No data for colormapping provided rf  r7   r8   r   r(  r;   r   )r   r  Warningr'   r9   )rl  r  
warn_matchr$   r$   r%   test_scatter_color_warning
  s   "(z&TestScatter.test_scatter_color_warningc                 C   sv   t jg dg dg dtjdddg dd}| jd	ks J t| g d
g dg dg t| g d d S )Nr>  r7   r   r8   0.1z0.3r  r   r)  r  r$  r  ?r;   r   r  )r   r(  r   r   r   r7   r   r   r   r7   r   r   r   r7   )	r'   r9   r  r  get_facecolorsr  r   get_edgecolorsget_linewidthsrl  collr$   r$   r%   test_scatter_unfilled
  s   z!TestScatter.test_scatter_unfilledr  c                 C   sl   t jg dg dg ddg dd}t| |  t| g dg dg d	g t| g d d S )
Nr>  r,  r-  r1   r/  r1  r2  r3  r4  )r'   r9   r   r5  r6  r7  r8  r$   r$   r%   test_scatter_unfillable
  s   z#TestScatter.test_scatter_unfillablec                 C   s   t d}tjtdd t|||dd   W d    n1 s"w   Y  tjtdd t|dd  |dd  | W d    n1 sHw   Y  tjtdd t||d W d    d S 1 sgw   Y  d S )Nr(  zsame size as x and yrf  r7   zfloat array-liker  )rd   rl   r   r   r|  r'   r9   )rl  r1   r$   r$   r%   test_scatter_size_arg_size	  s   
 "z&TestScatter.test_scatter_size_arg_sizec                 C   sj   t jg ddtjtjgdd}t| dsJ t jg ddtjtjdgdd}t| ds3J d S )Nr  r7   r  r.  r  r{  )r'   r9   rd   r  r  
same_colorr  r8  r$   r$   r%   test_scatter_edgecolor_RGB  s   z&TestScatter.test_scatter_edgecolor_RGBc                 C   s   |  }tjd d}|dd |jtdtddtjdtjgg d|dd	 |  }tjd d}|jd
dgd
dgddgddg|d |jddgddgddgdd d S )Nr9  r:  r  r7   r(  r8   r  Tr;   r  r   plotnonfiniter   r   r;   r  r   )r  r   	r   r   r<  r=  set_badr9   r   rd   r  rl  r!   r"   r#   r   r$   r$   r%   test_scatter_invalid_color  s   $"z&TestScatter.test_scatter_invalid_colorc                 C   s   |  }tjd d}|dd |jtdtddtjdtjgg d|dd	 |  }|jd
dgd
dgddgddg|d d S )Nr9  r:  r  r7   r(  r8   r  Fr@  r   r   rB  rC  rE  r$   r$   r%   test_scatter_no_invalid_color+  s   (z)TestScatter.test_scatter_no_invalid_colorc              	   C   sb   g d}t  }tjtdd |j|||tddddd W d	   d	S 1 s*w   Y  d	S )
r  r  r  rf  r  rw   r   r   )r;   r  r  r  N)r'   r^  r   r   r|  r9   r  r  )rl  r1   r#   r$   r$   r%   test_scatter_norm_vminvmax7  s   "z&TestScatter.test_scatter_norm_vminvmaxc                 C   s:   |  }|jdddd |  }|jdgdgdgd d S )Nr7   r  )r   r9   )rl  r!   r"   r#   r$   r$   r%   test_scatter_single_pointA  s   z%TestScatter.test_scatter_single_pointc                 C   s^   t d}| }|j||dd|ddd | }|j|dd||ddd d S )Nrw   r8   r   r  )rd   rl   r   r9   r   )rl  r!   r"   r1   r#   r$   r$   r%   test_scatter_different_shapesH  s
   
 $z)TestScatter.test_scatter_different_shapes)r  Nrgby
conversion)r  N)r)  N)NN)r"  r  r  r)  N)jaunerL  rM  r(  r   r   r   r  )r  z0.4r  r  )r  r  r  C5)r  r   r  rN  r  r7   r   r   r   r  z0.0rN  r   r7   r   zc_case, re_keyc                 C   s   dd }d}ddd}|d urt jt|| dnt|tr)t|dkr)t jd	dnt }| t	j
jj|d
i ||d W d    d S 1 sGw   Y  d S )Nc                   S   r   Nr   r$   r$   r$   r$   r%   get_next_color  r   z2TestScatter.test_scatter_c.<locals>.get_next_colorr(  z!^'c' argument has [0-9]+ elementsz^'c' argument must be a color)r  rL  rf  r   (argument looks like a single numeric RGBr  r;   r&  r  xsizeget_next_color_func)r   r   r|  r   r  r  r  
contextlibnullcontextr   r^  r   _parse_scatter_color_args)rl  c_casere_keyrR  rU  REGEXPassert_contextr$   r$   r%   test_scatter_c~  s&   "zTestScatter.test_scatter_cc                 C   s   g dg}g dg}|  }|jtdtd|d |jtdd td|d |  }|jtdtd|d |jtdd td|d d S )N)r7   r   皙?)r7   r   r_  r   r   r#  r(  r8   r  )r   r9   rd   r   r   )rl  r!   r"   rgbrgbarr   rq   r$   r$   r%   test_scatter_single_color_c  s   

"z'TestScatter.test_scatter_single_color_cc              
   C   s   t d}t \}}tdD ] }|j|t d|d| dd|d d}| |d ks/J q|j|t dddddg tddd d}t| g tddt	j
d	  d S )
Nr   r   r  r1   r   r7   )r;   r   r  r  C3lines.linewidth)rd   rl   r'   r   r   r9   r^   r7  r   r   r   )rl  r1   r-   r#   r;  pcr$   r$   r%   test_scatter_linewidths  s   
z#TestScatter.test_scatter_linewidthsc                 C   s   t jtdd tjg dg dg ddd W d    n1 s!w   Y  t jtdd tjg dg dg dd	d
 W d    n1 sGw   Y  t jtdd tjg dg dg dd	d W d    d S 1 snw   Y  d S )NzGGot both 'linewidth' and 'linewidths', which are aliases of one anotherrf  r  )r   r  r   r   )r  r/  zGGot both 'edgecolor' and 'edgecolors', which are aliases of one another)#ffffffz#000000z#f0f0f0rg  )r&  r.  zGGot both 'facecolors' and 'facecolor', which are aliases of one another)
facecolorsr-  )r   r   r   r'   r9   rk  r$   r$   r%   &test_scatter_singular_plural_arguments  s*    "z2TestScatter.test_scatter_singular_plural_arguments)!rp  rq  rr  r   r  r#  r&  r   r'  r(  r   markparametrizer  r  r+  r:  r   r   contextr;  r<  r?  rF  rG  rH  rI  rJ  params_test_scatter_cr^  rb  rf  ri  r$   r$   r$   r%   r  
  s    











	
	











+


r  )r&  c                K   s   | |||fS r  r$   )r;   rU  r&  r  r$   r$   r%   _params     rn  _resultz	c, colorszparams, expected_resultr  r?  )r;   r  r"  r  r{  r#  r  )r   r   r   r7   c                 C   s@   dd }t jjj| d|i\}}}||jksJ t||j d S )Nc                   S   r   rQ  r$   r$   r$   r$   r%   rR    r   z5test_parse_scatter_color_args.<locals>.get_next_colorrV  )r   r^  r   rY  r;   r   r  )paramsexpected_resultrR  r;   r  _edgecolorsr$   r$   r%   test_parse_scatter_color_args  s   rt  zkwargs, expected_edgecolorsr=  r%  r)  )r.  r&  )r;   r.  r&  )r   r.  c                 C   sN   dd }|  dd }|  dd }tjjj||| d|d\}}}||ks%J d S )Nc                   S   r   rQ  r$   r$   r$   r$   r%   rR    r   z@test_parse_scatter_color_args_edgecolors.<locals>.get_next_colorr;   r&  r8   )rU  rV  )popr   r^  r   rY  )r  expected_edgecolorsrR  r;   r&  r  result_edgecolorsr$   r$   r%   (test_parse_scatter_color_args_edgecolors  s   
rx  c                  C   sj   dd } t jtdd  tg dg dg}tjjj|d i d| d W d    d S 1 s.w   Y  d S )	Nc                   S   r   rQ  r$   r$   r$   r$   r%   rR    r   z;test_parse_scatter_color_args_error.<locals>.get_next_colorz&RGBA values should be within 0-1 rangerf  )r   r   ru  )r   r  gffffff?r8   r  rU  rV  )	r   r   r|  rd   rF  r   r^  r   rY  )rR  r;   r$   r$   r%   #test_parse_scatter_color_args_error  s   
"rz  zYou passed both c and facecolor/facecolors for the markers. c has precedence over facecolor/facecolors. This behavior may change in the future.)r  r   r  r   r   r  r  rP  rx  )r7   r7   r   zc, facecolorc                 C   s   dd }t jttd tjjj| dd|id|d W d   n1 s$w   Y  t jttd tjjj| dd|id|d W d   dS 1 sJw   Y  dS )	z<Test the internal _parse_scatter_color_args method directly.c                   S   r   rQ  r$   r$   r$   r$   r%   rR     r   z=test_parse_c_facecolor_warning_direct.<locals>.get_next_colorrf  Nrh  r8   rT  r-  )r   r  r$  WARN_MSGr   r^  r   rY  )r;   r-  rR  r$   r$   r%   %test_parse_c_facecolor_warning_direct  s   

"r|  c                 C   s   t  \}}t| ttfrddgndg}|}tjttd |j	||| |d W d   n1 s1w   Y  tjttd |j	||| |d W d   dS 1 sRw   Y  dS )z:Test the warning through the actual scatter plot creation.r   r7   rf  )r;   rh  N)r;   r-  )
r'   r   r   r  tupler   r  r$  r{  r9   )r;   r-  r-   r#   r1   r2   r$   r$   r%   ,test_scatter_c_facecolor_warning_integration0  s   "r~  c                  C   sr   G dd d} |  }|  }t j|_tjd|d}t|tu s J t  tjd|d}t|tu s3J t  d S )Nc                   @   s   e Zd Zdd Zdd ZdS )z#test_as_mpl_axes_api.<locals>.Polarc                 S   s
   d| _ d S Nr   )theta_offsetrk  r$   r$   r%   __init__C     
z,test_as_mpl_axes_api.<locals>.Polar.__init__c                 S   s   t d| jifS )Nr  )r   r  rk  r$   r$   r%   _as_mpl_axesF  s   z0test_as_mpl_axes_api.<locals>.Polar._as_mpl_axesN)rp  rq  rr  r  r  r$   r$   r$   r%   PolarB  s    r  r?  rU  y   )	rd   r  r  r'   r^  typer   rb  r`  )r  prjprj2r#   r$   r$   r%   test_as_mpl_axes_api@  s   r  c                  C   sZ   t  \} }t  \}}t | |t  u sJ | t  u s!J t |  t | d S r  )r'   r   scarY  gcfrb  )fig1r  fig2r  r$   r$   r%   test_pyplot_axesY  s   

r  c                  C   s   t  \} }|ttdd |jddd |  |jddd dd |j	|j
fD \}}|g d	ks9J |g d
ksAJ d S )Nr   r   r&        @rV  r  c                 s   s"    | ]}d d |  D V  qdS )c                 S   s   g | ]}|  |j fqS r$   )get_loclabel1get_textr  r$   r$   r%   r   k  r   z-test_log_scales.<locals>.<genexpr>.<listcomp>N)_update_ticks)r   r  r$   r$   r%   r  j  s
    
z"test_log_scales.<locals>.<genexpr>))r   z$\mathdefault{9^{0}}$)r  z$\mathdefault{9^{1}}$)g     @T@z$\mathdefault{9^{2}}$r   r   r  r   r  r   r  r   )r  r   )r  r   )r  r   )g      2@r   )g      ;@r   )g      B@r   )g     F@r   )g      K@r   )g     O@r   )      R@r   ))gF]tE?z$\mathdefault{5.5^{-1}}$)r   z$\mathdefault{5.5^{0}}$)r  z$\mathdefault{5.5^{1}}$)gF]tE?r   )gtE]t?r   )gF]tE?r   )g]tE?r   r  r  r  r  )r'   r   r   rd   r&  r   r  r  r  rV   rX   )r-   r#   r  yticksr$   r$   r%   test_log_scalesd  s   

r  c                  C   sT   t  \} }|jddd |jtd | |   kr%dks(J  J d S )Nr&  )r  r  r7   r7   rw   )	r'   r   r   rV   r  r   MultipleLocatorr   r   r  r$   r$   r%   test_log_scales_no_data  s   (r  c                  C   s   t  \} }|d tjtdd |dd W d    n1 s#w   Y  |d tjtdd |dd W d    d S 1 sFw   Y  d S )Nr&  zAttempt to set non-positiverf  r   rw   )	r'   r   r  r   r  r$  r  r  r  r,   r$   r$   r%   test_log_scales_invalid  s   

"r  stackplot_test_imagec                  C   s   t  } tddd}d| }d| d }d| d }| ddd}||||| |d |d	 ||||d
}t  \} }|jdddd|g dd |d |d	 d S )Nr   rw   r   r   r7   r  r8   r   rw   r   r  )r1   r  rJ  y3r1   r  rJ  r  )r  C1C2)r  r  )	r'   r   rd   r   r   	stackplotr  r  r   )r-   r1   r  rJ  r  r#   r  r$   r$   r%   test_stackplot  s   


r  stackplot_test_baselinec                  C   s   t jd dd } | dd}d|dd d f< tdd\}}|d jtd|jd	d
 |d jtd|jdd
 |d jtd|jdd
 |d jtd|jdd
 d S )Nr   c                 S   s   t || f}t| D ]A}tdD ]:}ddt j   }dt j  d }ddt j   }|d d |f  |t t || | | d   7  < qq|S )Nr   r7   r   r8   r   rw   )rd   rq  r   re   exprl   )r  r  r   r;  jr1   r2   rh  r$   r$   r%   layers  s   <z'test_stackplot_baseline.<locals>.layersr   r   r  r8   r  zeror  r  symr7   r   wigglerK  weighted_wiggle)rd   re   rf   r'   r   r  r   rp  )r  rU  r-   r  r$   r$   r%   test_stackplot_baseline  s   

r  c           	      C   s   t ddd}d| }d| d }d| d }| }|j||||g dd	gd
 |d |d t t|}|  }|j|||dd	d |j|||| dd	d |j||| || | dd	d |d |d d S )Nr   rw   r   r   r7   r  r8   )r1   r  \\r  )r  r  r  r  r1   )r  r-  r  r  )	rd   r   r   r  r  r  rq  r  r  )	r"   r!   r1   r  rJ  r  rq   stack_baselinerr   r$   r$   r%   test_stackplot_hatching  s   

 
r  c                 C   r  r  r$   )r  r$   r$   r%   r    r  c                 C   s   t jd tjjt jjddddfi | }t \}}|	dddkr+|
d n|d |	d	d
s>tjd tjd< |j||fi | d S )N  r   r   )r  r(  )meansigmar  rQ   verticalr&  patch_artistFrd  boxplot.boxprops.linewidth)rd   re   rf   r   cbookboxplot_stats	lognormalr'   r   getr  r  r   bxp)stats_kwargstransform_stats
bxp_kwargslogstatsr-   r#   r$   r$   r%   _bxp_test_helper  s   
r  zbxp_baseline.pngr  )r  r   c                   C   s
   t   d S r  r  r$   r$   r$   r%   test_bxp_baseline  s   
r  zbxp_rangewhis.pngc                   C      t tddgdd d S )Nr   r   whisr  r  rZ  r$   r$   r$   r%   test_bxp_rangewhis     r  zbxp_percentilewhis.pngc                   C   r  )Nr   _   r  r  r  r$   r$   r$   r%   test_bxp_percentilewhis  r  r  zbxp_with_xlabels.pngc                  C   s   dd } t | d d S )Nc                 S   $   t | tdD ]\}}||d< q| S NABCDr/   r  r  statsr  r/   r$   r$   r%   r,       
z(test_bxp_with_xlabels.<locals>.transform)r  r  r,  r$   r$   r%   test_bxp_with_xlabels  s   r  zbxp_horizontal.pngr   )r  r  r   r   c                   C      t tddd d S )NrO   rP   r  r  r$   r$   r$   r%   test_bxp_horizontal  s   r  zbxp_with_ylabels.png)r  r   r   c                  C      dd } t | tddd d S )Nc                 S   r  r  r  r  r$   r$   r%   r,  &  r  z(test_bxp_with_ylabels.<locals>.transformrO   rP   r  r  r  r  r$   r$   r%   test_bxp_with_ylabels!     
r  zbxp_patchartist.png)r  r  r   c                   C   r  )NT)r  r  r  r$   r$   r$   r%   test_bxp_patchartist/     r  zbxp_custompatchartist.pngr   c                   C       t tdtdddddd d S )NTr  r   :)r-  r.  r  )r  boxpropsr  r  r$   r$   r$   r%   test_bxp_custompatchartist7     r  zbxp_customoutlier.pngc                   C      t ttdddddd d S )Nr)  rU  r  r   r   mfc)
flierpropsr  r  r$   r$   r$   r%   test_bxp_customoutlierA     r  zbxp_withmean_custompoint.pngc                   C   r  )NTr)  rU  r   r  )	showmeans	meanpropsr  r  r$   r$   r$   r%   test_bxp_showcustommeanJ  r  r  zbxp_custombox.pngc                   C   r  )Nr   r  r   r   r   rC  )r  r  r  r$   r$   r$   r%   test_bxp_customboxU  r  r  zbxp_custommedian.pngc                   C   r  )Nr   r  r   r  medianpropsr  r  r$   r$   r$   r%   test_bxp_custommedian^  r  r  zbxp_customcap.pngc                   C   r  )Nr   r  r   r  )cappropsr  r  r$   r$   r$   r%   test_bxp_customcapg  r  r  zbxp_customwhisker.pngc                   C   r  )Nr  r  r   r  )whiskerpropsr  r  r$   r$   r$   r%   test_bxp_customwhiskerp  r  r  c                 C   sJ   t d}ddi}i |ddi}|  j||d | j||d d S )Nr   r/  r  r  r  r  )rd   rl   r   r-  )r!   r"   r  medianprops_testmedianprops_refr$   r$   r%    test_boxplot_median_bound_by_boxy  s
   
r  zbxp_withnotch.pngc                   C   r  )NT)shownotchesr  r  r$   r$   r$   r%   test_bxp_shownotches  r  r  zbxp_nocaps.pngc                   C   r  )NF)showcapsr  r  r$   r$   r$   r%   test_bxp_nocaps  r  r  zbxp_nobox.pngc                   C   r  )NF)showboxr  r  r$   r$   r$   r%   test_bxp_nobox  r  r   zbxp_no_flier_stats.pngc                  C   r  )Nc                 S   s   | D ]}| dd  q| S )Nfliers)ru  )r  r  r$   r$   r%   r,    s   z*test_bxp_no_flier_stats.<locals>.transformF)
showfliersr  r  r  r$   r$   r%   test_bxp_no_flier_stats  r  r  zbxp_withmean_point.pngc                   C   s   t tdddd d S )NTFr  meanliner  r  r$   r$   r$   r%   test_bxp_showmean     r  zbxp_withmean_line.pngc                   C   s   t tdddd d S )NTr  r  r  r$   r$   r$   r%   test_bxp_showmeanasline  r  r  zbxp_scalarwidth.pngc                   C   r  )Nrd  r3  r  r  r$   r$   r$   r%   test_bxp_scalarwidth  r  r	  zbxp_customwidths.pngc                   C      t tg ddd d S )N)r   rd  g?g333333?r3  r  r  r$   r$   r$   r%   test_bxp_customwidths     r  zbxp_custompositions.pngc                   C   r
  )N)r7   r   r  r:  r(  r  r  r$   r$   r$   r%   test_bxp_custompositions  r  r  c                   C   B   t t ttdgdd W d    d S 1 sw   Y  d S )Nr7   r3  r  r   r   r|  r  rZ  r$   r$   r$   r%   test_bxp_bad_widths     "r  c                   C   sD   t t ttddgdd W d    d S 1 sw   Y  d S )Nr8   r   r(  r  r  r$   r$   r$   r%   test_bxp_bad_positions  s   "r  zbxp_custom_capwidths.pngc                   C   r
  )N)r  r   r   r   r2  r  r  r$   r$   r$   r%   test_bxp_custom_capwidths  r  r  zbxp_custom_capwidth.pngc                   C   r  )Nrt  r2  r  r  r$   r$   r$   r%   test_bxp_custom_capwidth  s   r  c                   C   r  )Nr7   r2  r  r  r$   r$   r$   r%   test_bxp_bad_capwidths  r  r  r-  g{Gz?)r   r   c                  C   s   t jd t ddd} t d| dg} t \}}|j| | gddd	 |d
 d| | gi}t \}}|jddd|d |d
 d S )Nr  r9  r:     r  '  r7   	bootstrapnotchr  r   r1   )r  r  r  	rd   re   rf   r   hstackr'   r   r-  r  )r1   r-   r#   r  r$   r$   r%   test_boxplot  s   
r  c                 C   sP   t ddd}|  }||dk }|| t j|d}| }|| d S )Nr   r7   r  r   )rd   r   r   r-  r	   masked_less)r!   r"   x_origr#   r1   r$   r$   r%   test_boxplot_masked   s   
r"  zboxplot_custom_capwidths.pngc                  C   sF   t ddd} t d| dg} t \}}|j| | gdddgd	 d S )
Nr9  r:  r  r  r  r7   r  r   )r  r0  )rd   r   r  r'   r   r-  r  r$   r$   r%   test_boxplot_custom_capwidths  s   r#  zboxplot_sym2.pngc                  C   s~   t jd t ddd} t d| dg} tdd\}\}}|j| | gd	d
d |d |j| | gd	dd |d d S )Nr  r9  r:  r  r  r  r7   r8   r  ^)r  r  r  r  r  )r1   r-   r  r  r$   r$   r%   test_boxplot_sym2  s   
r%  zboxplot_sym.pngc                  C   sJ   t ddd} t d| dg} t \}}|j| | gdd |d d S )	Nr9  r:  r  r  r  r  r  r  )rd   r   r  r'   r   r-  r  r  r$   r$   r%   test_boxplot_sym)  s
   r'  z$boxplot_autorange_false_whiskers.pngz#boxplot_autorange_true_whiskers.pngc                  C   s   t jd t d} t d| dg} t \}}|j| | gddd |d t \}}|j| | gddd	d
 |d d S )Nr  r  r   r8   r  r7   r  )r  r   T)r  r  	autorange)	rd   re   rf   r   r  r'   r   r-  r  )r1   r  r  r  r  r$   r$   r%   test_boxplot_autorange_whiskers6  s   

r)  c                 C   s\   t ddd}t d|dg}t| | ||g W d    | S 1 s'w   Y  | S )Nr9  r:  r  r  r  )rd   r   r  rS   r   r-  )r#   rc_dictr1   r$   r$   r%   _rc_test_bxp_helperI  s   
r+  boxplot_rc_parameters)r  r  r   r   c                  C   s  t jd td\} }i dddddgdd	d
dddddddddddddddddddddddddddddd }d!d"gdd#}dd$d$d$dddddddd%d&}|||g}t||D ]	\}}t|| qitjj	d'd( |d) 
 D v sJ d S )*Nr  r   zboxplot.notchTboxplot.whiskersr   r  zboxplot.bootstrapr  zboxplot.flierprops.colorr  boxplot.flierprops.markerr   z"boxplot.flierprops.markerfacecolorr  z"boxplot.flierprops.markeredgecolorzboxplot.flierprops.markersizezboxplot.flierprops.linestyler   zboxplot.flierprops.linewidthr   zboxplot.boxprops.colorr"  r  zboxplot.boxprops.linestylezboxplot.capprops.colorr;   zboxplot.capprops.linewidthzboxplot.capprops.linestylezboxplot.medianprops.colorr  )zboxplot.medianprops.linewidthzboxplot.medianprops.linestyler   r   )r-  zboxplot.patchartistF-.)r-  zboxplot.showcapszboxplot.showboxzboxplot.showflierszboxplot.showmeanszboxplot.meanlinezboxplot.meanprops.colorzboxplot.meanprops.linewidthzboxplot.meanprops.linestylezboxplot.whiskerprops.colorzboxplot.whiskerprops.linewidthzboxplot.whiskerprops.linestylec                 S   r  r$   )r  r  r$   r$   r%   r     r  z.test_boxplot_rc_parameters.<locals>.<listcomp>r7   )rd   re   rf   r'   r   r  r+  rS   r4  r5  get_children)r-   r#   rc_axis0rc_axis1rc_axis2	dict_listr  rc_axisr$   r$   r%   test_boxplot_rc_parametersQ  s|   	

r6  zboxplot_with_CIarray.pngc                  C   sv   t jd t ddd} t d| dg} t \}}t ddgd	d
gg}|j| | gdd dg|dd |	d d S )Nr  r9  r:  r  r  r  r  r  r        @r  r   r7   )r  usermediansconf_intervalsr  r  )
rd   re   rf   r   r  r'   r   rF  r-  r  )r1   r-   r#   CIsr$   r$   r%   test_boxplot_with_CIarray  s   r;  zboxplot_no_inverted_whisker.pngc                  C   sP   t jg dt jd} t }||  |d |jjddd |j	d d S )N)r   i(#  r  X   i^  i@ ix  i  r  r&  Fminorwhich)
rd   rF  ri  r'   r^  r-  r  rX   r[  rV   )r1   r  r$   r$   r%   test_boxplot_no_weird_whisker  s   


r@  c                  C   s   t ddd} t d| dg} t \}}tt |j| ddgd W d    n1 s.w   Y  tt |j| | gddgddggd W d    d S 1 sSw   Y  d S )	Nr9  r:  r  r  r  r7   r8   )r8  	rd   r   r  r'   r   r   r   r|  r-  r  r$   r$   r%   test_boxplot_bad_medians  s    "rB  c                  C   s   t ddd} t d| dg} t \}}tt |j| | gddggd W d    n1 s1w   Y  tt |j| | gddgdggd W d    d S 1 sUw   Y  d S )	Nr9  r:  r  r  r  r7   r8   )r9  rA  r  r$   r$   r%   test_boxplot_bad_ci  s   "rC  c                  C   sZ   t d} t \}}|| d d  dksJ |j| ddd d  dks+J d S )Nrw   boxesr   r8   r;  )rd   rl   r'   r   r-  r  )r1   r  r#   r$   r$   r%   test_boxplot_zorder  s   
&rF  c                  C   s   dt jd< dt jd< dt jd< t  \} }td}d|d	< |j|d
d}dD ]}|| D ]
}| dks7J q-q'|d d  dksEJ |d d  dksQJ d S )Nr  r4  r   r.  r$  zboxplot.meanprops.markerr   r  r   T)r  )whiskerscapsrD  mediansr   r  r   means)r'   r   r   rd   rl   r-  
get_marker)r-   r#   	test_data
bxp_handle	bxp_lines	each_liner$   r$   r%   test_boxplot_marker_behavior  s   



rP  z&boxplot_mod_artists_after_plotting.pngc                  C   sH   g d} t  \}}|j| dd}|D ]}|| D ]}|d qqd S )N)rJ  r  rD  rD  Q?gQ?gQr   r&  r   )r'   r   r-  r  )r1   r-   r#   bpkeyr
  r$   r$   r%   &test_boxplot_mod_artist_after_plotting  s   rT  zviolinplot_vert_baseline.pngc                  C   sp   t jd dd tdD } t }|j| tddddd d| i} t \}}|jdtdddd| d d S )	Niec                 S      g | ]	}t jjd dqS r   r  rd   re   rg   r   r  r$   r$   r%   r         z1test_vert_violinplot_baseline.<locals>.<listcomp>r(  Fr)  r  showextremashowmediansrU  )r)  r  r[  r\  r  )rd   re   rf   r   r'   r^  
violinplotr   )r  r#   r-   r$   r$   r%   test_vert_violinplot_baseline  s   
r^  zviolinplot_vert_showmeans.pngc                  C   sB   t  } tjd dd tdD }| j|tddddd d S )Niw5+c                 S   rU  rV  rW  rX  r$   r$   r%   r     rY  z2test_vert_violinplot_showmeans.<locals>.<listcomp>r(  TFrZ  r'   r^  rd   re   rf   r   r]  r#   r  r$   r$   r%   test_vert_violinplot_showmeans     
ra  zviolinplot_vert_showextrema.pngc                  C   sB   t  } tjd dd tdD }| j|tddddd d S )Nic                 S   rU  rV  rW  rX  r$   r$   r%   r     rY  z4test_vert_violinplot_showextrema.<locals>.<listcomp>r(  FTrZ  r_  r`  r$   r$   r%    test_vert_violinplot_showextrema  rb  rc  zviolinplot_vert_showmedians.pngc                  C   sB   t  } tjd dd tdD }| j|tddddd d S )Nib}&c                 S   rU  rV  rW  rX  r$   r$   r%   r     rY  z4test_vert_violinplot_showmedians.<locals>.<listcomp>r(  FTrZ  r_  r`  r$   r$   r%    test_vert_violinplot_showmedians  rb  rd  zviolinplot_vert_showall.pngc                  C   s\   t  } tjd dd tdD }| j|tddddddgdd	gd
dgddggd d S )NiOc                 S   rU  rV  rW  rX  r$   r$   r%   r     rY  z0test_vert_violinplot_showall.<locals>.<listcomp>r(  Tr   ?r   r  r   ru  r  rt  )r)  r  r[  r\  	quantilesr_  r`  r$   r$   r%   test_vert_violinplot_showall  s   
rg  z#violinplot_vert_custompoints_10.pngc                  C   D   t  } tjd dd tdD }| j|tdddddd d S )Ni$c                 S   rU  rV  rW  rX  r$   r$   r%   r   &  rY  z8test_vert_violinplot_custompoints_10.<locals>.<listcomp>r(  Frw   r)  r  r[  r\  pointsr_  r`  r$   r$   r%   $test_vert_violinplot_custompoints_10!     
rk  z$violinplot_vert_custompoints_200.pngc                  C   rh  )NiYqVc                 S   rU  rV  rW  rX  r$   r$   r%   r   0  rY  z9test_vert_violinplot_custompoints_200.<locals>.<listcomp>r(  Fr  ri  r_  r`  r$   r$   r%   %test_vert_violinplot_custompoints_200+  rl  rm  zviolinplot_horiz_baseline.pngc                  C   sD   t  } tjd dd tdD }| j|tdddddd d S )Ni\dc                 S   rU  rV  rW  rX  r$   r$   r%   r   :  rY  z2test_horiz_violinplot_baseline.<locals>.<listcomp>r(  rO   Fr)  rQ   r  r[  r\  r_  r`  r$   r$   r%   test_horiz_violinplot_baseline5  rl  ro  z violinplot_horiz_showmedians.pngc                  C   sD   t  } tjd dd tdD }| j|tdddddd d S )	Nilo/c                 S   rU  rV  rW  rX  r$   r$   r%   r   D  rY  z5test_horiz_violinplot_showmedians.<locals>.<listcomp>r(  rO   FTrn  r_  r`  r$   r$   r%   !test_horiz_violinplot_showmedians?  rl  rp  zviolinplot_horiz_showmeans.pngc                  C   sD   t  } tjd dd tdD }| j|tdddddd d S )	Ni&c                 S   rU  rV  rW  rX  r$   r$   r%   r   N  rY  z3test_horiz_violinplot_showmeans.<locals>.<listcomp>r(  rO   TFrn  r_  r`  r$   r$   r%   test_horiz_violinplot_showmeansI  rl  rq  z violinplot_horiz_showextrema.pngc                  C   sD   t  } tjd dd tdD }| j|tdddddd d S )	Nie!c                 S   rU  rV  rW  rX  r$   r$   r%   r   X  rY  z5test_horiz_violinplot_showextrema.<locals>.<listcomp>r(  rO   FTrn  r_  r`  r$   r$   r%   !test_horiz_violinplot_showextremaS  rl  rr  zviolinplot_horiz_showall.pngc                  C   s^   t  } tjd dd tdD }| j|tdddddddgd	d
gddgddggd d S )Ni"c                 S   rU  rV  rW  rX  r$   r$   r%   r   b  rY  z1test_horiz_violinplot_showall.<locals>.<listcomp>r(  rO   Tr   re  r   r  r   ru  r  rt  )r)  rQ   r  r[  r\  rf  r_  r`  r$   r$   r%   test_horiz_violinplot_showall]  s   
rs  z$violinplot_horiz_custompoints_10.pngc               	   C   F   t  } tjd dd tdD }| j|tddddddd d S )	Ni0c                 S   rU  rV  rW  rX  r$   r$   r%   r   m  rY  z9test_horiz_violinplot_custompoints_10.<locals>.<listcomp>r(  rO   Frw   r)  rQ   r  r[  r\  rj  r_  r`  r$   r$   r%   %test_horiz_violinplot_custompoints_10h     
rv  z%violinplot_horiz_custompoints_200.pngc               	   C   rt  )	Ni<9!c                 S   rU  rV  rW  rX  r$   r$   r%   r   w  rY  z:test_horiz_violinplot_custompoints_200.<locals>.<listcomp>r(  rO   Fr  ru  r_  r`  r$   r$   r%   &test_horiz_violinplot_custompoints_200r  rw  rx  zviolinplot_sides.pngc               
   C   s   t  } tjd tjjddg}tg dg dD ]\}}| j||gdddd|d	 qtg d
g dD ]\}}| j||gdddd|d	 q6d S )NrZ   r   r  )r   r  r   )r   lowhighrO   FT)r)  rQ   r  r[  r\  side)r(  r7        @r  )r'   r^  rd   re   rf   rg   r  r]  )r#   r  r   r{  r$   r$   r%   test_violinplot_sides|  s   r}  c                  C   sh   t  } tjd dd tdD }tt | j	|tdd W d    d S 1 s-w   Y  d S )Ni@ 3c                 S   rU  rV  rW  rX  r$   r$   r%   r     rY  z1test_violinplot_bad_positions.<locals>.<listcomp>r(  r   r(  
r'   r^  rd   re   rf   r   r   r   r|  r]  r`  r$   r$   r%   test_violinplot_bad_positions  s   "r  c                  C   sn   t  } tjd dd tdD }tt | j	|tdg dd W d    d S 1 s0w   Y  d S )NiA#c                 S   rU  rV  rW  rX  r$   r$   r%   r     rY  z.test_violinplot_bad_widths.<locals>.<listcomp>r(  r  )r)  r1  r~  r`  r$   r$   r%   test_violinplot_bad_widths  s   "r  c                  C   sn   t  } tjd tjjddg}tt | j	|ddgddggd W d    d S 1 s0w   Y  d S )	Nil r   r  r   r   r   ru  rf  
r'   r^  rd   re   rf   rg   r   r   r|  r]  r`  r$   r$   r%   test_violinplot_bad_quantiles  s   "r  c                  C   s   t  } tjd tjjddg}tt | j	|g dgd W d    n1 s,w   Y  tt | j	|g dgd W d    d S 1 sLw   Y  d S )Niq4r   r  )r   r   r   ?r  )皙r   r   r  r  r`  r$   r$   r%   $test_violinplot_outofrange_quantiles  s   "r  c                 C   sV   t jd t jjddg}|  }|j|g dd | }|j|g dgd d S )N+r   r  )r   r   re  r  )rd   re   rf   rg   r   r]  )r!   r"   r  r#   r$   r$   r%   %test_violinplot_single_list_quantiles  s   r  c                 C   s   t jd |jt jjddg dd}|jt jjddtdd}|t jjdd}|  |||g | |j|j|jg d S )	Nr  r:  r  )rE  rR  r:  r  r   r(  r   rU  rE  	ABCDEFGHIr  )	rd   re   rf   rk  rg   r  r   r]  r  )r!   r"   r.  s1s2s3r$   r$   r%   test_violinplot_pandas_series  s   r  c                  C   sz   t  \} }|dd | }tjd tjddd}tjddd}|j||gddgdd	 | }t	|| d S )
Nr   r(  rw   r   rQ  r7   r8   F)r)  manage_ticks)
r'   r   r  r   rd   re   rf   rg   r-  r   )r  r#   old_xlimr  rJ  new_xlimr$   r$   r%   test_manage_xticks  s   r  c                  C   s   t  \} }|jtjddgd |jtjddgd | j  | dks+J t	|
 ddgks7J dd | D dd	gksFJ d S )
Nr   r   r(  r   )r  r  c                 S   rc  r$   r  r  r$   r$   r%   r     r  z+test_boxplot_not_single.<locals>.<listcomp>r   5)r'   r   r-  rd   re   rx   r>  r  r   r  
get_xticksrb  r,   r$   r$   r%   test_boxplot_not_single  s   
"r  c                  C   s@   t ddgddg tjddi t } t j| ddd d S )Nr   r7   z	font.sizeP   raw)r  format)r'   r   rS   r   updateioBytesIOsavefigr  r$   r$   r%   test_tick_space_size_0  s   r  )errorbar_basicerrorbar_mixedr  c            
   	   C   s  t jdddt jd} t |  }ddt |   }d| }t }| }|j| |ddd |	d tj
d	d	d
d\}}|d }|j| ||dd |	d |jdd |d }|j| ||ddd |	d |d }|j| ||d	| g|d	| gdd |	d |d }|d t d|| }|| }|j| ||d	| g|ddd	d |	d |dd |d | |d }	t }| }|jd!d"dd|	d# |	d d S )$Nr   r(  r   r  r   r  r'  z&Simplest errorbars, 0.2 in x, 0.4 in yr8   T)r  r  r  r  r   r)  r   zVert. symmetricnbinsr  )r(  r   re  zHor. symmetric w/ alphar  z--or)  r(  r   zH, V asymmetricrK  r&  r  r  )r)  r(  r   ecolorcapthickzMixed sym., log yr
  zVariable errorbarsr  r1   r2   r(  r)  r  )rd   rl   
longdoubler  r  r'   r   rY  r%  r4   r   locator_paramsr  maximumr  suptitle)
r1   r2   r)  r(  r-   r#   r  ylower
yerr_lowerr  r$   r$   r%   test_errorbar  sD   


$




r  mixed_errorbar_polar_capsc                  C   s  t  } t jddd}g d}dgd }|j||ddd	d
 tjd d tjd g}ddg}|j||dtj dd	d
 dtj d d dtj d d dtj d g}dgd }g dg dg}g dg dg}	|j||||	d	d
 dg}
dg}|j|
|ddd	d
 dS )z
    Mix several polar errorbar use cases in a single test figure.

    It is advisable to position individual points off the grid. If there are
    problems with reproducibility of this test, consider removing grid.
    r8  r  rU  r  re  r   ffffff?r   r   r(  r)  r   r8   r   g?r  r  rJ  r(  r   r$  )r   r   r   )r   r   r   )r  r   r   )r   r  r    @@rw   N)r'   r   r`  r%  rd   r  )r-   r#   th_symr_symth_longr_longth_asymr_asymr(  r)  th_overr_overr$   r$   r%   test_mixed_errorbar_polar_caps'  s    	
0
r  c                  C   s   t  \} }td}d| }|j||d d\}}}|j|d| d d\}}}||d| \}t| tdks>J t| tdksLJ t| tdksZJ d S )Nrw   r8   r  r(  r  r  r  )	r'   r   rd   rl   r%  r   r  to_rgba	get_color)r  r#   r1   r2   r  r  r  ln1r$   r$   r%   test_errorbar_colorcycleJ  s   
 r  c              
      s   t ddd  fddtdD }|  }| }t|g dD ]\}}|j ||d dd	d
d |j ||d dd	|d
d q!d S )Nr   r(  r   c                    s   g | ]
}t   | qS r$   )rd   r  r   r  r1   r$   r%   r   \  rX  z.test_errorbar_cycle_ecolor.<locals>.<listcomp>)r  r  r  rc  rd  r  r   r  )r)  r   r   r  )r)  r   r   r   r  )rd   rl   r   r   r  r%  )r!   r"   r2   axtaxryir   r$   r  r%   test_errorbar_cycle_ecolorY  s   r  c                  C   s  t  } |  }tddd}t| }ddt|  }t|d| fj}d| }t	
t |j|||dd W d    n1 sEw   Y  t	
t |j|||dd W d    n1 scw   Y  t	
t |j||||dd	 W d    d S 1 sw   Y  d S )
Nr   r(  r   r   r8   r   r  )r(  r   r  )r'   r   rY  rd   rl   r  r  vstackrp  r   r   r|  r%  )r-   r#   r1   r2   yerr1r)  r(  r$   r$   r%   test_errorbar_shapeh  s    "r  errorbar_limitsc                  C   s\  t ddd} t |  }d}d}d}t \}}|j| ||||dd t | }d|g d	< |j| |d ||||d
d t | }d|g d< |j| |d ||||dd |j| |d dd|||||dd
 d}t | d}d|ddg< |}	|}
t | }t | }d|dg< d|dg< |j| |d dd|||	|
||ddddd |d |	d d S )Nr   r  r   r   dottedr   )r(  r)  r  r   T)r7   r   rE  r   )r(  r)  uplimsr  r   )r8   r(  rR  r   r  )r(  r)  lolimsr  r   r   r   rR  magenta)r   r  r(  r)  r  r  r  r   r   r   r  r  r)  r   cyan)r   r  r(  r)  xlolimsxuplimsr  r  r  r  capsizer   )r   r  zErrorbar upper and lower limits)
rd   rl   r  r'   r   r%  
zeros_like	full_liker  r4   )r1   r2   r(  r)  r  r-   r#   r  r  r  r  r$   r$   r%   test_errorbar_limitsz  sH   






r  c                  C   sd   t d} t d}tj| |dddd\}}}|d u sJ |D ]}t | tdks/J qd S )Nr   r7   r)  r  r  )rd   rl   r'   r%  r  r  r  r  )r1   r2   plotliner  barlineserrbarr$   r$   r%   test_errorbar_nonefmt  s   

r  c                  C   s   t d} t d}t \}}|j| |ddd}t|jdks"J t|jdks+J t|jdks4J |	  |jr=J |jrBJ |jrGJ d S )Nr   r7   r'  r8   )
rd   rl   r'   r   r%  r  
containersr  r  r_  )r1   r2   r-   r#   ecr$   r$   r%   test_errorbar_remove  s   



r  c                  C   s`   t d} t d}tj| |ddddddddd	d
d\}}}| dks&J | dks.J d S )Nr   r7   Noner  r^   z	steps-midroundr  r  bevel)
r(  r)  r  r   r  r  dash_capstyledash_joinstyler  r  )rd   rl   r'   r%  get_fillstyleget_drawstyle)r1   r2   r  r  r$   r$   r%   "test_errorbar_line_specific_kwargs  s   

r  c              	   C   s  |  }|jg dg dddddd |jg dg ddd	d
ddd |jg dg ddddddd |dd tg dg dg dg dd}tjd|d |   }|jg dg ddd |jg dg ddd	d |jg dg dddd |dd d S )N)r8   r(  rw   r>  r   r   r  r  )r1   r2   r)  r  r   r  r8   r   r(  z	tab:greenr  r2   )r1   r2   r)  r   r  r   r  r(  r   r  ztab:bluer/  r   r;   )r1   r2   r)  r   r  r   r  r7   r  )r   r  r/  )r  r  r   )r  r2   r;   r  r  r"  )r  r   r  r   r^  )
prop_cycler1   r2   r)  )r1   r2   r)  r   )r1   r2   r)  r   )r   r%  r  r
   r'   rc)r!   r"   r#   _cycler$   r$   r%   test_errorbar_with_prop_cycle  s(   r  c                  C   s8  t ddd} | d|   }|d }t  }tjtdd |j| ||dd W d    n1 s2w   Y  tjtdd |j| ||d	d W d    n1 sRw   Y  tjtd
d |j| ||ddgd W d    n1 stw   Y  tjtdd |j| ||dd W d    d S 1 sw   Y  d S )Nr   r7   r   r  znot a tuple of two integersrf  r  )
errorevery)r0  r   znot a valid NumPy fancy indexFTznot a recognized valuefoobar)	rd   r   r'   r   r   r   r   r|  r%  )r1   r2   r)  r#   r$   r$   r%   test_errorbar_every_invalid  s    "r  c                  C   sX  t   } tjtdd | jdgdgdgdggdgdggd W d    n1 s+w   Y  tjtdd | jdgdgdgdggd W d    n1 sPw   Y  tjtdd | jdgdgdgdggd	 W d    n1 suw   Y  tjtdd! td
}dd |D }| j||t	j
ddd	 W d    d S 1 sw   Y  d S )Nz''xerr' must not contain negative valuesrf  r   r  r7   r1   r2   r(  r)  r0  z''yerr' must not contain negative valuesr  r   c                 S   s"   g | ]}t  d d|d d qS )r9  rE  r8   r7   r  r  r$   r$   r%   r        " z/test_xerr_yerr_not_negative.<locals>.<listcomp>r  rP  )r'   r   r   r   r   r|  r%  rd   rl   r  rS  )r#   r1   r2   r$   r$   r%   test_xerr_yerr_not_negative  s4   (

"r  c                  C   s   t   } tjtdd | jdgdgd gdggd gdggd W d    n1 s+w   Y  tjtdd | jdgdgd gdggd W d    n1 sPw   Y  tjtdd | jdgdgd gdggd W d    d S 1 svw   Y  d S )	Nz'xerr' must not contain Nonerf  r   r7   r  r0  z'yerr' must not contain Noner  )r'   r   r   r   r   r|  r%  r  r$   r$   r%   test_xerr_yerr_not_none  s    ("r  c           	   
   C   s6  t ddd}|d|  }|d }| }|  }tdg dD ]8\}}|d7 }|j||||dfd|d	 |j|||d
d |j||d d ||d d ||d d d|dd q |j||d |dddd |j|dd d |dd d d dd
d |j||d |ddd |j||d |tdd dtdd ddddd |j|dd d |dd d d ddd
d |j|dd d |dd d d |dd d dddd |j||d |g dd g dd dddd |j|dd d |dd d d ddd
d |j|dd d |dd d d |dd d dddd d S )Nr   r7   r   r  rgbk)r   r   r8   r:  rn  r(  )r  r  r;   r  )r;   r;  r)  )r  r;   r   r   )r7   r(  r   )r  r  r   rE  )r  r   r   r8   r   r  )r  r  r  r;   r   r{  r   r  )rd   r   r   r  r%  r   r  )	r!   r"   r1   r2   r)  rr   rq   r   shiftr$   r$   r%   test_errorbar_every   sB   (*
,,
,,
r  
elinewidthr  c                 C   s"   t jg dg dg d| d d S )Nr  )r)  r  )r'   r%  )r  r$   r$   r%   test_errorbar_linewidth_typeL  s   "r  c                 C   s   |   }td}tddtjtjdg}tddtjtjdg}|||| |  }|ddgddgddg |jdgdgdgdd	 d S )
Nr   r7   r8   r   r(  r  r   r  r   )r   r   rd   rF  r  r%  )r!   r"   r#   xsysesr$   r$   r%   test_errorbar_nanS  s   r  hist_stacked_stepfilledc                  C   sj   t ddd} t ddd}t \}}|j| |fddd	 d
| |fi}t \}}|jd
dd|d d S )Nr7   r   rQ  r   rw   r  rs  Tr  r1   )ru  r  r  rd   r   r'   r   ro  )d1d2r-   r#   r  r$   r$   r%   test_hist_stacked_stepfilled_  s   r  hist_offsetc                  C   sH   t ddd} t ddd}t \}}|j| dd |j|d	d d S )
Nr   rw   r  r7   r   rQ  r   r  r   r   r  r  r-   r#   r$   r$   r%   test_hist_offsetm  s
   r  zhist_step.pngc                  C   sD   t ddd} t \}}|j| dd |dd |dd	 d S )
Nr7   r   rQ  rr  rt  r   rw   r   r   )rd   r   r'   r   ro  r  r  r  r-   r#   r$   r$   r%   test_hist_stepw  s
   r  zhist_step_horiz.pngc                  C   @   t ddd} t ddd}t \}}|j| |fddd	 d S )
Nr   rw   r  r7   r   rQ  rr  rO   )ru  rQ   r   r  r$   r$   r%   test_hist_step_horiz     r
  hist_stacked_weightsc                  C   sb   t ddd} t ddd}t ddd}t d	d
d}t \}}|j| |f||fddd d S )Nr   rw   r  r7   r   rQ  r  r7  r_  r   rs  T)weightsru  r  r   )r  r  w1w2r-   r#   r$   r$   r%   test_hist_stacked_weighted  s   r  zstem.pngc                  C   sH   t ddt j d} t \}}|j| t | ddddd |  d S )	Nr   r8   r   C2-.zk+C1-. )linefmt	markerfmtbasefmtr/   )rd   r   r  r'   r   r  r\  rG   r  r$   r$   r%   	test_stem  s   r  c                  C   s   dd } t  \}}g d}g d}| ||g d|fd | |||||fd | |j||dd||fd | |||d||fd | |j||dd	d
||fd | |j|ddg d|fd | ||dg d|fd dS )z5Test that stem() correctly identifies x and y values.c                 S   s8   t t| j \}}||d ksJ ||d ksJ d S Nr   r7   )mapr  
markerliner  )stem_containerexpectedr1   r2   r$   r$   r%   _assert_equal  s   z%test_stem_args.<locals>._assert_equalr7   r   r   rE  rR  r:  r>  )r  rD  r  zb--)r  r  Nr'   r   r  r  r-   r#   r1   r2   r$   r$   r%   test_stem_args  s    r#  c                  C   s  		ddd} t  \}}g d}g d}| |||ddd | |j||d	d
dd	d | |j||dd
dd	d | |||ddddd | |||ddddd | |j||dd	d
ddd	d | |j||ddd
dddd | |j||ddd
ddd	d | |j||ddd
dddd | |j||ddd
dddd | |j||dddddd | |j||dd	dddd	d | |j||dddddd	d dS )zATest that stem(..., markerfmt=...) produces the intended markers.Nc                 S   sl   |durt | j |sJ |durt | j |sJ |dur+| j |ks+J | j dks4J dS )zl
        Check that the given StemContainer has the properties listed as
        keyword-arguments.
        Nr  )r  r>  	stemlinesr  r  rK  get_linestyle)r  	linecolormarkercolorr   r$   r$   r%   r    s   z*test_stem_markerfmt.<locals>._assert_equalr  r  r  r   )r'  r   r1   )r  r   r"  )r&  r'  r   r  gxr  r  r   r   )r  r  )NNNr!  r"  r$   r$   r%   test_stem_markerfmt  sb   

r)  c                  C   s@   t dd\} }tjdtjdg}ddg}||| d S )Nr7   z2013-9-28 11:00:00z2013-9-28 12:00:00r   r  )r'   r   dateutilparserparser  )r-   r#   r  r  r$   r$   r%   test_stem_dates  s   

r-  zstem_orientation.pngc                  C   s@   t ddt j d} t \}}|j| t | ddddd d S )	Nr   r8   r  r  kxr  rO   )r  r  r  rQ   )rd   r   r  r'   r   r  r\  r  r$   r$   r%   test_stem_orientation   s   
r/  hist_stacked_stepfilled_alphac                  C   sB   t ddd} t ddd}t \}}|j| |fddd	d
 d S )Nr7   r   rQ  r   rw   r  rs  Tr   )ru  r  re  r   r  r$   r$   r%   "test_hist_stacked_stepfilled_alpha
  s   r1  hist_stacked_stepc                  C   r	  )
Nr7   r   rQ  r   rw   r  rr  Tr  r   r  r$   r$   r%   test_hist_stacked_step  r  r3  hist_stacked_normedc                  C   s@   t ddd} t ddd}t \}}|j| |fddd d S )	Nr7   r   rQ  r   rw   r  T)r  r  r   r  r$   r$   r%   test_hist_stacked_density  r  r5  zhist_step_bottom.pngc                  C   s4   t ddd} t \}}|j| t ddd d S )Nr7   r   rQ  rw   rs  )rM   ru  )rd   r   r'   r   ro  rl   r  r$   r$   r%   test_hist_step_bottom%  s   r6  c               	   C   s   g d} g d}ddgddgddgddgddgddgddgddgg}ddgddgddgddgddgg}d|fd|| ffD ]\}}t j|| |d	\}}\}t| | q=d S )
Nr   r7   r8   r   r   r   r7   r7   r7   r8   r   r8   r7   r   rr  rs  r  ru  r'   ro  r   get_xyr  r  r=   rM   ru  r  r  polygonr$   r$   r%   test_hist_step_geometry-  s   4"r>  c               	   C   s   g d} g d}ddgddgddgddgddgddgddgdd	gg}dd	gddgddgddgddgg}d
|fd|| ffD ]\}}t j|| g d|d\}}\}t| | q=d S )Nr7  r8  r   r7   r   r   r8   r  r   rr  rs  r7   r8   r   )r  rM   ru  r:  r<  r$   r$   r%   test_hist_step_bottom_geometry8  s   4"r@  c               
   C   sZ  g d} g d}g d}ddgddgddgddgddgddgddgddggddgddgddgddgddgddgddgddggg}ddgddgddgddgddggddgddgddgddgddggg}d	d
 t ||D }d|fd|ffD ]6\}}tj||g| d|d\}}}	t|	dksJ |	d \}
t|
 |d  |	d \}
t|
 |d  qtd S )Nr7  r8  r>  r   r8   r7   r   r(  c                 S      g | ]\}}|| qS r$   r$   r   r  r  r$   r$   r%   r   P  r  z3test_hist_stacked_step_geometry.<locals>.<listcomp>rr  rs  T)r  r  ru  r  r'   ro  r  r   r;  r  data_1data_2topsbottomscombinedru  r  r  r4  r=  r$   r$   r%   test_hist_stacked_step_geometryD  s(   22  

rJ  c               
   C   s`  g d} g d}g d}ddgddgddgddgddgdd	gdd	gdd
ggddgddgddgddgddgddgddgdd	ggg}dd
gddgddgddgddggdd	gddgddgddgddggg}dd t ||D }d|fd|ffD ]9\}}tj||g| dg d|d\}}}	t|	dksJ |	d \}
t|
 |d  |	d \}
t|
 |d  qtd S )Nr7  r8  r>  r   r7   r   r   r8   r  r   r(  r  r7  c                 S   rA  r$   r$   rB  r$   r$   r%   r   h  r  z:test_hist_stacked_step_bottom_geometry.<locals>.<listcomp>rr  rs  Tr?  )r  r  rM   ru  rC  rD  r$   r$   r%   &test_hist_stacked_step_bottom_geometry\  s(   22  

rK  hist_stacked_barc                  C   sn   g dg dg dg dg dg dg} g d}g d}t  \}}|j| d	d
d||d |jdddd d S )N)r   r   r   r   r  @  i  r  rQ  r  6  r  )rQ     r  r  r   i  )r:  r:  r:  r  r  r     )r  r  r  r  r  r  r   r   r   r   rw   r  )+  rQ  rQ  r   r   r   r   r   r   r   r   r   r   r   )r   r   r   r   r  r  r  r  r  r  r  r  ))gXxn?r   r  )r  r   g `p?)r  r   gǟ?)r  gˢ`?r   )gC?r  r   )gK?r  r   )r   rI  z yellowr  r  rw   rw  mid)r  ru  r  r   r/   zupper right)r   r   r7   )r?   bbox_to_anchorr  )r'   r   ro  rG   )rU  r  labelsr-   r#   r$   r$   r%   test_hist_stacked_bart  s   rU  r  r-  r  r.  r  )r  r  r~   r   )r  r   r  r/  )r7   r   r8   r   c                 C   s   t jd dd dD }| d\}}|d\}}d||fd||ffD ]D\}}	}
|	j|fd|d	|\}}}tt| \}}tt	||D ]\}\}}|
j|f||d	||id
t
|| d i qJq%d S )NrZ   c                 S   s   g | ]}t j|qS r$   )rd   re   rx  r  r$   r$   r%   r     r  z/test_hist_vectorized_params.<locals>.<listcomp>)rQ  r  r   r8   rs  rr  rw   r9  r;  )rd   re   rf   r   ro  nextiteritemsr  r  r  )r!   r"   r  r  axt1axt2axr1axr2ru  r  r  r  r  kwr  r;  r1   r  r$   r$   r%   test_hist_vectorized_params  s   
r^  zkwargs, patch_face, patch_edger2   )ru  r   r-  r.  )r2   r   )ru  r   r.  )r"  r   )ru  r   r-  r  )ru  r-  r.  )ru  r   )ru  r-  r  )ru  r.  )r  r   ru  c                    sD   t   jg dfi | \}}}t fdd|D s J d S )Nr  c                 3   s,    | ]}t | | g gV  qd S r  )r  r>  r  r  r  
patch_edge
patch_facer$   r%   r    s    z,test_hist_color_semantics.<locals>.<genexpr>)r'   r   r   ro  r  )r  ra  r`  r  r4  r$   r_  r%   test_hist_color_semantics  s   $rb  c                  C   sD   t ddg} tjddgddggdd| d |  ddgks J d S )Nrw   rQ  r   r7   r8   rw  )ru  rM   )rd   rF  r'   ro  tolistr  r$   r$   r%   %test_hist_barstacked_bottom_unchanged  s   rd  c                  C   s,   t  \} }|jg tdtdgdd d S )Nrw   rr  rt  )r'   r   ro  r   r,   r$   r$   r%   test_hist_emptydata  s    re  c                  C   sB   t  \} }|jg dgg dd | \}}|dgksJ d S )Nr  )r  unusedzalso unusedr  r  )r'   r   ro  get_legend_handles_labels)r-   r#   r  rT  r$   r$   r%   test_hist_unused_labels  s   rh  c                  C   s   t  \} }|jddgdd\}}}|d  dksJ |jddgdgd\}}}|d  dks3J |jddgd d\}}}|d  dksIJ |jddgdd\}}}|d  dks_J |jddgdd\}}}|d  dksuJ d S )Nr   r7   r  r  r]  00)r'   r   ro  rd  )r-   r#   r  rb  r$   r$   r%   test_hist_labels  s   rj  transparent_markersc                  C   s:   t jd t jd} t \}}|j| dddd d S )Nr   r  r1  r)  r   )r  r   rd   re   rf   r'   r   r   r  r$   r$   r%   test_transparent_markers  s   rm  rgba_markersc               	   C   s   t jdd\} }ddg}ddg}d dg}tdd	d
}t|D ]?\}}t|D ]6\}}	t|D ]-\}
}|| j|d |
d df||	|d| || j|d |
d df|	|d| q.q&q|D ]	}|g d q`d S )Nr8   r  r{  rO  r?  )r   r   r7   r   r   r   rQ  )r  r  r7   r   )r  r  re  r   r1   )r  re  )r   r(  r   r   )r'   r   rZ  r  r   r  )r-   r  rcolorsbcolorsalphasr]  r;  re  r  rcolorr  bcolorr#   r$   r$   r%   test_rgba_markers  s&   *rt  mollweide_gridc                  C   s    t  } | jdd}|  d S )N	mollweiderU  )r'   r   r   r[  r,   r$   r$   r%   test_mollweide_grid  s   rw  c                  C   s   t  } | jdd}ttj tjd}ttj d tjd ddd }t||\}}t| | fj	}|j
|}|j
 |}tj||d d S )	Nrv  rU  r  r   rP  r7   r   r   )r'   r   r   rd   r   r  rY  r  r  rp  transProjectionr,  invertedr   r   )r-   r#   lonlatllr  ll2r$   r$   r%   &test_mollweide_forward_inverse_closure  s   $r~  c                  C   sz   t  } | jdd}tddd}t||\}}t| | fj}|j	
 |}|j	|}tj||d d S )Nrv  rU  r   r7   r  r   )r'   r   r   rd   r   rY  r  r  rp  rx  ry  r,  r   r   )r-   r#   r1   r2   r  r|  xy2r$   r$   r%   &test_mollweide_inverse_forward_closure  s   r  
test_alphac               	   C   s   t jd t jd} t \}}|j| dg dg dddd |j| d	 dg dg dddd
d |j| d dg dg ddddd |j| d dg dg dddd |j| d dg dg dddd d S )Nr   r  z-Dr  rO  rQ  rw   )r   r  r   rC  r8   r7   )r   r  r   rC  re  r(  r   r  rR  rl  r  r$   r$   r%   r  2  s(   
	eventplotc                  C   sP  t jd t jddg } t jddg }| | }t|}g dgt|  }g dg dg dg d	g d
g dg}|| }dt dt| d  }g d}| | }	dgt|  }
g d}|
| }t }| }|j	|||	|d}t|}||ksJ |||	|d}t }| }|j	dddd|d}t|}||ksJ d S )Nr   r}   rQ  r  )r   r7   ru  r  rP  rx  )r7   r  r   )r7   r   r7   )r   r7   r7   r  Q?)ir  r7   r   r  rw   )r   r8   r7   r7   r   r   )r  lineoffsetslinelengths)r   r;   lor|  r   r;   r  r|  )r  r  r  r  )
rd   re   rf   rc  r  rl   r'   r   r   r  )r?  data2r  num_datasetscolors1colors2r  lineoffsets1lineoffsets2r  linelengths1linelengths2r  r-   axobjcollsnum_collectionsr$   r$   r%   test_eventplotP  sF   

r  ztest_eventplot_defaults.pngc                  C   sZ   t jd t jddg } t jddg }| | }t }| }|| dS )zf
    test that eventplot produces the correct output given the default params
    (see bug #3728)
    r   r}   rQ  r  N)rd   re   rf   rc  r'   r   r   r  )r?  r  r  r-   r  r$   r$   r%   test_eventplot_defaults|  s   r  r  ))r  )z
tab:orangeztab:pinkztab:cyanbLacK)r  rP  N)r7   r   r7   r   c                 C   s   dgdgdgdgg}dd | D }t |dkr|d }tt|t |df}t \}}t | dkr8| d } |j|| d}t||D ]\}}t	|
 | qDd	S )
zBTest the *colors* parameter of eventplot. Inspired by issue #8193.r   r7   r8   r   c                 S   s   g | ]
}|d ur
|ndqS )Nr  r$   r   r;   r$   r$   r%   r     rX  z)test_eventplot_colors.<locals>.<listcomp>r(  r  N)r  rd   r|  r  to_rgba_arrayr'   r   r  r  r   r  )r  r  r  r-   r#   r  r9  r   r$   r$   r%   test_eventplot_colors  s   r  c                  C   s*  t  \} }|jg dg dgdd}|d  dksJ |d  dks'J |jg dg dgddgd}|d  dks@J |d  dksJJ tjtdd	 |jg dg dgg d
d W d    n1 skw   Y  tjtdd	 |jg dddgd W d    d S 1 sw   Y  d S )N)r   r8   r(  )r7   r   r   r:  ru  r  r   r7   r   zalpha and positions are unequalrf  )r   ru  re  )r'   r   r  	get_alphar   r   r|  )r-   r#   r  r$   r$   r%   test_eventplot_alpha  s    "r  z!test_eventplot_problem_kwargs.pngc              	   C   s   t jd t jdg }t jdg }||g}t }| }|j|ddgddgdd	gd	dgd
dgddgd t| dksDJ t	dd | D sOJ dS )z
    test that 'singular' versions of LineCollection props raise an
    MatplotlibDeprecationWarning rather than overriding the 'plural' versions
    (e.g., to prevent 'color' from overriding 'colors', see issue #4297)
    r   rQ  rw   r"  r  r;   r  r8   r7   r   dasheddashdotr  )r  r   r  r/  r  r   r   c                 s   s    | ]
}t |jtjV  qd S r  )
issubclasscategoryr   r<  )r   wir$   r$   r%   r        z0test_eventplot_problem_kwargs.<locals>.<genexpr>N)
rd   re   rf   rc  r'   r   r   r  r  r  )recwarnr?  r  r  r-   r  r$   r$   r%   test_eventplot_problem_kwargs  s$   r  c                  C   s.   t dd\} }|jg gdgd t   d S )Nr7   )r  r  r  r  r  r'   r   r  r  r,   r$   r$   r%   test_empty_eventplot  s   r  r  rQ   )Nr  rO   c                 C   sB   |du ri nd|i}t dd\}}|j| fi | t   dS )z#Introduced when fixing issue #6412.NrQ   r7   r  )r  rQ   optsr-   r#   r$   r$   r%   test_eventplot_orientation  s   r  c                 C   s   t  dddt  dddt  dddg}t  dddt  dddg}| }|j|dd |j|dd |  }|||g d S )	Nr9  r7   r8   r   r   r:  r   )r  )r  r   r  )r!   r"   ts_1ts_2r#   r$   r$   r%   test_eventplot_units_list  s   r  zmarker_styles.pngc               	   C   s   t  \} }ttjjjdd d}|d dg|}t|D ]%\}}|j|d d t	
dd  t	dd | d|d|d  |d	 qd S )
Nc                 S   s   t t| t |  S r  )r  r  r  r$   r$   r%   r    r  z$test_marker_styles.<locals>.<lambda>)rS  r)  r8   r   rw   r   )r   r   r   r/   )r'   r   sortedrS   markersr  r_  r  r   rd   rl   r   )r-   r#   r  r2   r   r$   r$   r%   test_marker_styles  s   


.r  zrc_markerfill.pnggl?c                  C   sP   t  \} }td}tg dD ]\}\}}|tjd< |j|| |d qd S )Nr:  ))r=   r  )rM   r   )r)  r$  zmarkers.fillstyler  )r'   r   rd   rl   r  rS   r   r   )r-   r#   r1   idxr   r   r$   r$   r%   test_markers_fillstyle_rcparams  s   

r  zvertex_markers.pngc                  C   sr   t td} d}g d}t \}}|j| d|dd |j| d d d d|dd |ddg |ddg d S )Nrw   ))r   r   rA  rK  rB  r   r  r  r   r  )r  r   r'   r   r   r  r  )r  marker_as_tuplemarker_as_listr-   r#   r$   r$   r%   test_vertex_markers  s   r  vline_hline_zordererrorbar_zorderg9v?c                  C   s  t td} t }| }|j| ddd |jddddd |jddddd |jdd	ddd |jd
dddd |d t }| }t td} t	
d}t td}|j| ||dddd tdD ]}|j|dd|d |j| dd|d qg|d d S )Nrw   r   )rC  r;  r7   r  )r   rC  r;  r   r:  r  r8   r  r   zaxvline and axhline zorder testr"  )r)  r;  rC  r   )rC  r   r;  zerrorbar zorder test)r  r   r'   r   rY  r   rn   axvliner4   rd   rq  r%  )r1   r-   r#   r2   r)  r  r$   r$   r%   test_eb_line_zorder  s&   

r  c                 C   s   |   }|jddd |ddgddgd |jddd	d
 |  }|jddd |ddgddgd |jddgddgd	d
 d S )N)r   rw   )r  r7   rF  r   rt  z.-)r7   r  )rw   r  r  r  r7   rw   r  r  )r   r   r  axliner    r$   r$   r%   test_axline_loglog<  s   r  c                 C   s   |   }|jddd |dd |jdddd |jdd	d
d |jdddd |jdddd |jdtddd |  }|jddd |ddgddg |jddd |jdd
d |jddd |jddgddgdd |jddd d S )NrB  rF  r  rK  r  r  r#  )r   r   r7   r   r  )gffffffr  r   rc  )sloper   )r7   r  r  C4)r  r7   infrN  r   r7   r   )r   r   r  r  r   rn   r  r    r$   r$   r%   test_axlineI  s    r  c                 C   s   |   }|jddd |jdd|jd |jddd|jd |jd	d
d|jd |jddd|jd |  }|jddd |ddgddg |jd
dgdd
gdd |jddgd
d
gdd |jd
d
gddgdd d S )NrB  rF  r  r7   r  r,  r  r  r  r   r,  r   r   r   r  )r   r   )r   r7   rc  )r   r,  r   r#  )r   r   r  r8  r   r    r$   r$   r%   test_axline_transaxes`  s   r  c                 C   s   |   }|jddd |jdd|jd |jddd|jd	 |jdd
d|jd	 |jddd | dd |  }|jddd |dd |d
dgd
dg |jd
dgd
dgdd |jd
dgddgdd d S )NrB  rF  r  r7   r  r  r8   r  r  r   r  )r   r   r  r   r   rw   r#  )r   r   r  r8  set_size_inchesr   r    r$   r$   r%   test_axline_transaxes_panzoomq  s   r  c                  C   sV  t  \} }tt |d W d   n1 sw   Y  tt |jdddd W d   n1 s8w   Y  |d tt |jddd W d   n1 sYw   Y  |d |d tt |jddd W d   n1 sw   Y  |d tt |dd t 	  W d   dS 1 sw   Y  dS )z3Exactly one of *xy2* and *slope* must be specified.r  NrK  r7   )r  r&  r  )
r'   r   r   r   r   r  r  r  r|  r  r,   r$   r$   r%   test_axline_args  s*   




"r  )vlines_basicvlines_with_nanvlines_maskedc                  C     g d} g d}t  \}}|j| d|ddd g d}dd	d
dtjdg}t jd
dd\}\}}}	|j|d|ddd g d}
tjdd	d
ddg}|j|
d|dd
dd g d}tjdd	d
dtjg}|	j|d|ddd ||||	fD ]}|dd qq| | ksJ | | ksJ | |	 ksJ t  \}}tjg dd}tjg dd}tjg dd}|j|||ddd |dd d S N)r8   r   r(  r   r:  )r8   r   rR  r8   r   r  r   r  r/  )r8   r   r(  r   r  r:  r8   r  r   rR  )r(  rR  )r  rW  r"  r   )r  r/  r   r  rw   )r8   r(  r  rR  rw   r  )r   r7   r   r   r8   r7   )r  r  r   r:     r	  r	  r   )	r'   r   rm   rd   r  r  r   r	   masked_equal)r  r  r  r  rI  rJ  r  r  r  r  x3r  x4y4r#   fig3ax5x5ymin5ymax5r$   r$   r%   test_vlines  2   r  c                  C   d   t  \} }tddi |ddd}tj| ds J W d    d S 1 s+w   Y  d S Nzlines.colorr  r   r   r7   )r'   r   r   r   rm   r  r>  r  r-   r#   r  r$   r$   r%   test_vlines_default  
   "r  )hlines_basichlines_with_nanhlines_maskedc                  C   r  r  )	r'   r   hlinesrd   r  r  r   r	   r  )r  r  r  r  rJ  rI  r  r  r  r  r  r  r  r  r#   r  r  y5xmin5xmax5r$   r$   r%   test_hlines  r  r  c                  C   r  r  )r'   r   r   r   r  r  r>  r  r  r$   r$   r%   test_hlines_default  r  r  r   r  r(  c                 C   s   g d}|  dddj|dd|dd |  dddj|dd|dd g d}g d}| dddj|dd|dd | dddj|dd|dd d S )	N)r  r   r   r  rI  r8   r7   r   r   r  )r7   r8   r   r   )r  r   r   rI  )r   rm   r  )r!   r"   r  test_colors	expect_xyexpect_colorr$   r$   r%   test_lines_with_colors  s   
r  vlines_hlines_blended_transform)r   r   c                  C   s   t ddd} t |  t dt j |   d }tjdddd\}\}}|| |d	 |jdd
gdd|	 dd || |d	 |j
ddgdd| dd d S )Nr  r
  r   r8   rw   r7   r  r  r   r$  rE  r   r   r"  )rb   rc   r,  r  r  r:  rJ  )yminymaxr,  r  )rd   rl   r  r  r  r'   r   r   r  get_yaxis_transformrm   get_xaxis_transform)r  r  r-   haxvaxr$   r$   r%   $test_vlines_hlines_blended_transform	  s   $
r  step_linestyler   c               	   C   sN  t d } }tdd\}}| }g d}t||D ]4\}}|j| |d|dd |j| |d d|dd |j| |d d|d	d |d
dg |d
dg q| ||d |d d}tdd\}}| }g d}t||D ]3\}}|jddd|d|d |jddd|d|d |jddd|d	|d |d
dg |d
dg qqd S )Nrw   r8   )r  r   r/  r  r   pre)rC  r   r  r7   rR  r  r   r:  )r\  Y0Y1Y2r\  r  )rC  r   r  r  r  r  )	rd   rl   r'   r   r  r  rr  r  r  )r1   r2   r-   ax_lst	ln_stylesr#   r  r  r$   r$   r%   test_step_linestyle  s*   r  mixed_collectionc                  C   s   t  \} }tjddddd}tjj|gdd}|ddgd	d	gg |d
dg tjj|gdd}|ddgddgg |d
dg |	g dg dg |j
d || || |dd |dd d S )NrR  rR  r(  r)  r   radiusr-  r.  T)match_originalr   r
  r7   r   r  ii)r   r   r   r   )r   r   r   r   r  r:  )r'   r   r@  Circler   r  PatchCollectionset_offsetsset_linewidthsset_edgecolorsr  r  r  r  r  )r-   r#   r;   r  r   r$   r$   r%   test_mixed_collection8  s   

r  c                  C   s:   t tdtdd} |   |   dksJ d S )Nr   r7   )r   r7   r   r   )r'   r`  rd   int32r{   r
  get_subplotspecget_geometryr  r$   r$   r%   test_subplot_key_hashR  s   r  )zspecgram_freqs.pngzspecgram_freqs_linear.pngzspecgram_noise.pngzspecgram_noise_linear.pngQ?)r  r   r   c                  C   s  dt jd< d} d}|d |d |d g|d |d	 |d
 gg}td| t| }td| d| }ttdtj tj	
|| jdd}td| d }tjd ttj| tj| g}g d}||f||ffD ]M\}	}
|
d }tdtt|
 }tt d|D ]\}}|j|	|
||||d qtt d|D ]\}}|j|	|
||||dt j d qqrdS )z)Test axes.specgram in default (psd) mode.r   r  rr  r
  r(  r   r  @ffffff@'@rw   r   r7   r8   r[  r  onesidedtwosidedr   )NFFTFsnoverlappad_tosidesr  )r  r  r  r  r  r%  r  N)rS   r   intrd   r+  rl   rn  r  r  multiplyouterr  re   rf   r  rx   ry   log2r  r'   r   r   specgramr  r  )r  r  fstims
NFFT_freqsr1   y_freqs
NFFT_noisey_noise	all_sidesr2   r  r  r  r#   r  r$   r$   r%   test_specgramX  s6   
,$
r   )zspecgram_magnitude_freqs.pngz#specgram_magnitude_freqs_linear.pngzspecgram_magnitude_noise.pngz#specgram_magnitude_noise_linear.pngc                  C   s  dt jd< d} d}|d |d |d g|d |d	 |d
 gg}td| t| }td| d| }tdtj tj	|| j
dd}d|dddf< t|}td| d }tjd ttj| tj| g}g d}	||f||ffD ]O\}}
|
d }tdtt|
 }tt d|	D ]\}}|j||
||||dd qtt d|	D ]\}}|j||
||||ddt j d	 qq|dS )z%Test axes.specgram in magnitude mode.r   r  rr  r
  r(  r   r  r
  r  r  r   r   r7   r8   r[  Nr   rw   r  r   	magnituder  r  r  r  r  r`   r  )r  r  r  r  r  r`   r%  r  )rS   r   r  rd   r+  rl   r  r  r  r  r  r  re   rf   rn  r  rx   ry   r  r  r'   r   r   r  r  r  )r  r  r  r  r1   r2   r  r  r  r  r  r  r  r#   r  r$   r$   r%   test_specgram_magnitude|  s6   
,&

r#  )zspecgram_angle_freqs.pngzspecgram_phase_freqs.pngzspecgram_angle_noise.pngzspecgram_phase_noise.pngc                  C   s  dt jd< d} d}|d |d |d g|d |d	 |d
 gg}td| t| }td| d| }tdtj tj	|| j
dd}d|dddf< t|}td| d }tjd ttj| tj| g}g d}	||f||ffD ]V\}}
|
d }tdtt|
 }dD ]?}tt d|	D ]2\}}|j||
|||||d tt |j||
|||||dd W d   n1 sw   Y  qqq|dS )z,Test axes.specgram in angle and phase modes.r   r  rr  r
  r(  r   r  r
  r  r  rw   r   r7   r8   r[  Nr   r  )r  phaser   r"  dB)r  r  r  r  r  r`   r%  )rS   r   r  rd   r+  rl   r  r  r  r  r  r  re   rf   rn  r  rx   ry   r  r  r'   r   r   r  r   r   r|  )r  r  r  r  r1   r2   r  r  r  r  r  r  r  r`   r#   r  r$   r$   r%   test_specgram_angle  s>   
,&
r&  c                  C   sF   t jtdddd\} }}}| \}}}}|dkr|dks!J dS )z;Test axes.specgram when Fs is None, should not throw error.r  Nr  )r  r%  r}   `   )r'   r  rd   r   
get_extent)specfreqsr  rD  rb   rc   freq0freq1r$   r$   r%   test_specgram_fs_none  s   r-  c                 C   sH   t d}t |}dtjd< | | dtjd< |  | dS )zCTest specgram ignores image.origin rcParam and uses origin 'upper'.r  upperzimage.originr   N)rd   rl   r  r'   r   r   r  )r!   r"   r  signalr$   r$   r%   test_specgram_origin_rcparam  s   



r0  c                  C   sR   t d} t | }tt tj|dd W d   dS 1 s"w   Y  dS )z4Ensure passing origin as a kwarg raises a TypeError.r  r   r  N)rd   rl   r  r   r   r   r'   r  )r  r/  r$   r$   r%   test_specgram_origin_kwarg  s
   

"r1  )zpsd_freqs.pngzcsd_freqs.pngzpsd_noise.pngzcsd_noise.pnggMb`?c               	   C   s  d} d}|d |d |d g|d |d |d gg}t d	| t| }td
| d| }tdtj tj|| jdd}t d	| d }tj	
d
 tj	| tj	| g}ddiddddddg}||f||ffD ]z\}	}
|
d }t dtt|
 }tt d|D ]*\}}|jt|	f|
|||d|}t|d|dd ksJ |jddd qtt d|D ]&\}}|j|	|
|||d|}t|d|dd ksJ |jddd qqpd S )Nr        Y@r(  r   r  r
  r  r  rr  r   r7   r8   r[  r  r  r  F)r  return_liner  Tr   )r  r  r  r  r3  r   xlabelylabel)r  rd   r+  rl   r  r  r  r  r  re   rf   r  rx   ry   r  r  r'   r   r   psdrn  r  r  r   csd)r  r  r  r  r1   ys_freqsr  ys_noise
all_kwargsr  r  r  r  r#   r  retr$   r$   r%   test_psd_csd  sD   ,&
r=  )z#magnitude_spectrum_freqs_linear.pngzmagnitude_spectrum_freqs_dB.pngzangle_spectrum_freqs.pngzphase_spectrum_freqs.pngz#magnitude_spectrum_noise_linear.pngzmagnitude_spectrum_noise_dB.pngzangle_spectrum_noise.pngzphase_spectrum_noise.pngc                  C   s  d} d}|d |d |d g}t d| t| }t dtt| }td| d	| }tdtj t|| d
td  j	d	d}tj
d ttj
| tj
| gd }g d}||d}	||fD ]}
tt d|D ]\}}|j|
fd|i|	\}}}|jddd qwtt d|D ]\}}|j|
fd|i|	ddi\}}}|jddd qtt d|D ]\}}|j|
fd|i|	\}}}|jddd qtt d|D ]\}}|j|
fd|i|	\}}}|jddd qqkd S )Nr  r2  r(  r   r  rr  r8   r   r7   rw   r   r[  r   r  )r  r  r  r   r4  r%  r%  )r  r+  rd   ry   r  rl   r  r  r  r  re   rf   r  r  rx   r  r'   r   r   magnitude_spectrumr   angle_spectrumphase_spectrum)r  r  fstims1r  r  r1   r  r  r  r  r2   r#   r  r)  r*  r  r$   r$   r%   test_spectrum  s<   ("
rB  c                  C   s   t  d} | D ]	}|jjdd q	tjdd! | d td | d 	tdtd W d    d S 1 s=w   Y  d S )	Nr8   T)ry  r  r  r   r   r7   )
r'   r   r   rX   r   rd   r  r7  rq  r8  r  r#   r$   r$   r%   test_psd_csd_edge_cases5  s   "rD  c                 C   sH   |   }| }| }|  |  |  }|j  |j  d S r  )r   r
  r	  r_  rV   tick_bottomrX   	tick_left)r!   r"   rq   ax_twinxax_twinyrr   r$   r$   r%   test_twin_remove?  s   
rI  ztwin_spines.pnggI+?c            	      C   s|  dd } t jdd}|jdd | }| }| }|jjd | | |jjd |	g d	g d	d
\}|	g d	g dd\}|	g d	g dd\}|
dd |dd |dd |dd |jj|  |jj|  |jj|  tddd}|jdd| d| |jdd| d| |jdd| d| |jdddi| d S )Nc                 S   s.   |  d | jd | jd d  d d S )NTF)set_frame_onr  r   spinesr  r$   r$   r%   make_patch_spines_invisibleR  s   
z5test_twin_spines.<locals>.make_patch_spines_invisible)r(  r   r   r  )rA   )r^  r  Tr>  zb-)r   r   r8   zr-)r  r   r   zg-r   r8   r(  r7   A   r   )r  r  r2   )r  r  r  r1   r$   )r'   r   r^  r   r
  rK  rA   set_positionr   r   r  r  rX   r/   r  r  rZ  tick_params)	rL  r-   hostpar1par2r  r   p3tkwr$   r$   r%   test_twin_spinesN  s0   rU  ztwin_spines_on_top.pngc                  C   s@  dt jd< dt jd< t } | ddd}tg dg dg}| }|j|d |d d d	d
 |j	|d |d d d	dd |j|d |d d dd
 |j	|d |d d ddd |d |d d d}t } | ddd}| }|jddd	|d |j	ddd	d|d |jddd|d |j	dddd|d d S )Ng      H@r*  rd  r7   )rr  iL  i  i  )rN  i-  r  r  r   g     @@z#BEAED4r#  r  )r   re  z#7FC97Fr   )r;  r  r;  r  )r   r  )r   re  r  )
rS   r   r'   r   r   rd   rF  r
  r   r  )r-   r  r  r  r$   r$   r%   test_twin_spines_on_top|  s(   


rV  z(grid_which, major_visible, minor_visible))r   TT)majorTF)r=  FTc                    sh   t jd| d t \}}|j  t fdd|jj	D s#J tfdd|jj
D s2J d S )NT)r5  zaxes.grid.whichc                 3       | ]
}|j   kV  qd S r  gridliner  r   tick)major_visibler$   r%   r    r  z*test_rcparam_grid_minor.<locals>.<genexpr>c                 3   rX  r  rY  r[  )minor_visibler$   r%   r    r  )r   r   r  r'   r   r>  r  r  rV   
majorTicks
minorTicks)
grid_whichr]  r^  r-   r#   r$   )r]  r^  r%   test_rcparam_grid_minor  s   
rb  c                  C   s   t  \} }|  | j  |jjd j sJ |jdd | j  |jjd j r0J |jdd | j  |jjd j sFJ |  | j  |jjd j rZJ d S )Nr   FvisibleT)	r'   r   r[  r>  r  rV   r_  rZ  r  r,   r$   r$   r%   	test_grid  s   



re  c                  C   s   t  \} }|jdddd |jjd j rJ |jdd |jjd j s*J t ddi  |	  |jdddd |jjd j sIJ W d    d S 1 sTw   Y  d S )	NTrW  rw   )resetr?  	labelsizer   r  r#  r5  )
r'   r   rO  rV   r_  rZ  r  r[  r   r  r,   r$   r$   r%   test_reset_grid  s   "rh  c                 C   sn   || fD ] }|  }|d |jdddddddd	d
dddd |  q| jD ]}|j  |j  q(d S )NTinrw   r   r  r  r  r  r  r  r  r   r   )	directionlengthr  r   padrg  
labelcolorlabelrotation
grid_color
grid_alphagrid_linewidthgrid_linestyle)r   r[  rO  draw_without_renderingr^  rV   reset_ticksrX   )r!   r"   r-   r#   r$   r$   r%   test_reset_ticks  s   




ru  c                  C   s   t dddddddddd		 t  \} }W d    n1 s w   Y  |   |jjd
 }|jjdd  D ]2}|j|jksAJ |j|jksIJ |j|jksQJ |j	|j	ksYJ |j
|j
ksaJ |j|jksiJ q7d S )Nri  r   r   r  r  Tr  r  )	xtick.directionzxtick.major.sizezxtick.major.widthxtick.colorzxtick.major.padxtick.bottom	xtick.topxtick.labelsizextick.labelcolorr   r7   )r'   r   r   rs  rV   r_  _size_width	_base_pad_labelrotation_zorder_tickdir)r-   r#   
first_tickr\  r$   r$   r%   test_context_ticks  s$   
r  c                  C   s:   t  } |  }|d |g d t| d d S )Nr   )r  r   r   r   )r  r   )r'   r   rY  r  r   r   r   r,   r$   r$   r%   test_vline_limit  s
   
r  zfv, fh, args)r7   rK  c                 C   s
  t jd}tjtdd | ||dd W d    n1 s w   Y  tjtdd | |d|d W d    n1 s?w   Y  tjtdd |||dd W d    n1 s^w   Y  tjtdd ||d|d W d    d S 1 s~w   Y  d S )	Nr7   z"ymin must be a single scalar valuerf  r  r  z"ymax must be a single scalar valuez"xmin must be a single scalar value)rb   rc   z"xmax must be a single scalar value)rS   rP  num2dater   r   r|  )fvfhargsbad_limr$   r$   r%   test_axline_minmax  s   "r  c                  C   s   t jddddd\} }|d g dg d |d  \}}|d  \}}|dks-J |dks3J |dks9J |d	ks?J d S )
Nr7   r8   T)r  r  r  rX  r   r  )r8   r(  r  r   r  )r'   r   r   r   r   )r-   r  r  r  r  r  r$   r$   r%   test_empty_shared_subplots	  s   r  c                  C   s   dD ]E} t jddd\}}|d jd| dd |d  dks!J |d  | ks+J t jddd\}}|d jd| d |d  d	ksGJ qd S )
Nboxdatalimr8   T)r  r  r   )
adjustablesharer7   r  ro  )r'   r   
set_aspect
get_aspectget_adjustable)r  r-   r  r$   r$   r%   test_shared_with_aspect_1  s   r  c                  C   s   t jdddd\} }|d jddd |d ddgddg |d ddgddg t   |d  |d  ks=J |d  |d  ksKJ d S )	Nr8   T)r  r  rX  r   )r  r7   r   r(  )r'   r   r  r   r  r   r   ry  r$   r$   r%   test_shared_with_aspect_2"  s    r  c            	      C   s  dD ]} t jddd\}}|d jd| d |d jd| d |d ddgd	d
g |d d	d
gddg t   |d  |d  ksIJ |d  |d  ksWJ |jj|jj	 }|D ](}|
 }|j|j	 }|jj|jj	 }|| | }t|d
t| d
ksJ qaqd S )Nr  r8   T)r  rX  r   r  r7   r   r   r(  )r'   r   r  r   r  r   r   bbox_inchesr  r  get_positionviewLimr  r  )	r  r-   r  
fig_aspectr#   r   
box_aspect
lim_aspectr  r$   r$   r%   test_shared_with_aspect_3-  s$   r  c                  C   s`   t jddddd\} }|d d tjtdd |   W d    d S 1 s)w   Y  d S )	Nr7   r8   Tr  rX  r   r  zset_aspect\(..., adjustable=rf  )r'   r   r  r   r   RuntimeErrorrs  )r-   r^  r$   r$   r%   test_shared_aspect_errorA  s
   
"r  zerr, args, kwargs, matchr'  z@axis\(\) takes from 0 to 1 positional arguments but 2 were given)r  z4Unrecognized string 'foo' to axis; try 'on' or 'off'zThe first argument to axis*r  z1axis\(\) got an unexpected keyword argument 'foo'c                 C   D   t j| |d tj|i | W d    d S 1 sw   Y  d S Nrf  )r   r   r'   r  errr  r  rg  r$   r$   r%   test_axis_errorsH  s   "r  c                  C   s  t  } tjtdd | d W d    n1 sw   Y  tjtdd | d W d    n1 s7w   Y  tjtdd | jddd W d    n1 sUw   Y  tjtdd | 	d W d    n1 sqw   Y  tjtdd | j
d	d
 W d    n1 sw   Y  tjtdd | jddd	d W d    n1 sw   Y  tjtdd | jddd	d W d    n1 sw   Y  tjtdd | jdd	d W d    n1 sw   Y  tjtdd | jdd	d W d    n	1 s	w   Y  tjtdd | jdd	d W d    n	1 s(w   Y  tjtdd | jdd	d W d    d S 1 sHw   Y  d S )Nzunknown value for which: 'foo'rf  r  z!Cannot supply both positional andr  r  zargument must be amongzscilimits must be a sequencer7   )	scilimitszSpecifying 'loc' is disallowedrN   )r?   r1   r=   )r?   r2   zCannot pass both 'left'r   )rN   rb   zCannot pass both 'right')rA   rc   zCannot pass both 'bottom')rM   r  zCannot pass both 'top')r=   r  )r'   rY  r   r   r|  r  r  r   set_prop_cycle
set_anchorticklabel_formatr(   r)   r  r  r  r$   r$   r%   test_axis_method_errorsZ  sD   $r  c                 C   sH   t  \}}t|d|   }|d |d t|jj|jj d S )Nr  r   r8   )r'   r   r  r  r   bboxextents)r  r-   r#   r:  r$   r$   r%   test_twin_with_aspectt  s   

r  c                  C   s   d} d}d}d}t j }| }|| | | | ksJ | |ks'J |||\}| |ks6J | |ks>J |d | |ksKJ | |ksSJ |jdd |	  | | kseJ | |ksmJ d S )N)r  r
  )r  r  )r  r  FT)visible_only)
rS   r   r   r   r   r   r   r   relimr   )r  r  rI  rJ  r-   r#   r  r$   r$   r%   test_relim_visible_only  s&   

r  c                  C   s,   t  } |  }|jdd |jdd dS )z
    tests for issue #1172
    large)rg  outrj  N)r'   r   rY  rO  r,   r$   r$   r%   test_text_labelsize  s   r  zpie_default.pngr  c               	   C   sD   d} g d}g d}d}t jdd\}}|j||| |ddd	d
 d S )NFrogsHogsDogsLogsr   r   r  rw   yellowgreengoldlightskyblue
lightcoralr   r   r   r   )rR  r  r   %1.1f%%Tr  )exploderT  r  autopctshadow
startangler'   r   pie)rT  sizesr  r  r  r  r$   r$   r%   test_pie_default  s   
r  )pie_linewidth_0r  r  )r   r   r   c                  C   s   d} g d}g d}d}t j||| |ddddd	id
 t d | |||d}t  }| }|jddddddddd	i|d	 |d t   t jddddddddd	i|d	 t d d S )Nr  r  r  r  r  Tr  r/  r   r  rT  r  r  r  r  
wedgepropsr  )lr  r;   exr  r  r  r;   )r  rT  r  r  r  r  r  r  )r'   r  r  r   rY  )rT  r  r  r  r  r-   r#   r$   r$   r%   test_pie_linewidth_0  s.   

r  zpie_center_radius.pngc                  C   sl   d} g d}g d}d}t j||| |ddddd	id
dd
 t jdd
dtdddtdddd t d d S )Nr  r  r  r  r  Tr  r/  r   r'  r   )	r  rT  r  r  r  r  r  rU   r  zCenter point)r7   r0  ->arc3)
arrowstyleconnectionstylesquare	lightgrey)boxstyler-  )r  r  r  r  r  )r'   r  r  rZ  r  rT  r  r  r  r$   r$   r%   test_pie_center_radius  s   


r  zpie_linewidth_2.pngc               
   C   sD   d} g d}g d}d}t j||| |ddddd	id
 t d d S )Nr  r  r  r  r  Tr  r/  r8   r  r  r'   r  r  r  r$   r$   r%   test_pie_linewidth_2  s   r  zpie_ccw_true.pngc               
   C   @   d} g d}g d}d}t j||| |ddddd t d	 d S )
Nr  r  r  r  r  Tr  )r  rT  r  r  r  r  counterclockr  r  r  r$   r$   r%   test_pie_ccw_true     r  zpie_frame_grid.pngc                  C   s   d} g d}g d}d}t j||| |ddddd	idd
d
 t j|d d d || |ddddd	iddd
 t j||d d d | |ddddd	iddd
 t d d S )Nr  r  r  r  r  Tr  r/  r   rL  )	r  rT  r  r  r  r  r  framerU   r   )r   r8   )r   r   r  r  r  r$   r$   r%   test_pie_frame_grid  s(   r  zpie_rotatelabels_true.pngc               
   C   r  )
N)Hogwartsr  r  r  r  r  r  r  Tr  )r  rT  r  r  r  r  rotatelabelsr  r  r  r$   r$   r%   test_pie_rotatelabels_true!  r  r  zpie_no_label.pngc                  C   sV   d} g d}g d}d}t j||| |dddd dd	 t d	 t d
d t   d S )Nr  r  r  r  r  Tr  )r  rT  r  r  r  r  labeldistancer  r  g333333r  )r'   r  r  r   rG   r  r$   r$   r%   test_pie_nolabel_but_legend0  s   
r  zpie_shadow.pngc               	   C   s   g d} g d}d}t dd\}}|d d j| ||ddddid	 |d d
 j| ||ddddid	 |d
 d j| ||ddddddddid	 |d
 d
 j| ||dddddddid	 d S )Nr  r  r  r8   r   Tr  r/  )r  r  r  r  r  r7   Fr  re  r)  )oxoyshader.  r_  r   )r  r/  r  r  )r  r  r  r  r^  r$   r$   r%   test_pie_shadow>  s(   

r  c                  C   s   g d} g d}t ddddddd	}t j| |d
|d\}}}||fD ]C}|D ]>}| |d ks4J | |d ks>J | |d ksHJ | |d ksRJ | |d ks\J |	 |d ksfJ q(q$d S )N)rO  "   r  )zLong name 1zLong name 2zLong name 3rU   r=   r  anchorr  r  )horizontalalignmentverticalalignmentr  rotation_moder  r   %.2f)rT  r  	textpropsr  r  r  r  r  r   )
rZ  r'   rY  r  get_haget_vaget_rotationget_rotation_modeget_sizer  )r  rT  r  r  textsr  txr$   r$   r%   test_pie_textpropsV  s*   r  c                  C   sR   t  \} }tt |jg dg dd W d    d S 1 s"w   Y  d S )N)r   r   r  )r   r   r   r  r'   r   r   r   r|  r  r,   r$   r$   r%   test_pie_get_negative_valuesl     "r  c                  C   sR   t  \} }tt |jg dddgd W d    d S 1 s"w   Y  d S )Nr  r   r  r  r,   r$   r$   r%   test_pie_invalid_explodes  r  r  c                  C   sR   t  \} }tt |jg dddgd W d    d S 1 s"w   Y  d S )Nr  OneTworT  r  r,   r$   r$   r%   test_pie_invalid_labelsz  r  r	  c                  C   sN   t  \} }tt |jg ddd W d    d S 1 s w   Y  d S )Nr  r  )r  r  r,   r$   r$   r%   test_pie_invalid_radius  s   "r	  c                  C   sp   t  \} }g d}|j|dd}t|d d jd dk s J |j|dd}t|d d jd dks6J d S )	Nr   r   r   T)r1   	normalizer   r   g     v@r  F)r'   r   r  r+  r  )r-   r#   r1   t1t2r$   r$   r%   test_normalize_kwarg_pie  s   "r	  c                    sF   g d}d |   j| d |  |\}} fdd|D  d S )Nr	  r  rX  c                    s   g | ]}|  qS r$   	set_hatch)r   r'  rX  r$   r%   r     r   z)test_pie_hatch_single.<locals>.<listcomp>r   r  )r!   r"   r1   wedgesr  r$   rX  r%   test_pie_hatch_single  s
   r	  c                 C   sL   g d}g d}|   j||d |  |\}}dd t||D  d S )Nr	  )r  r  r~   rX  c                 S   s   g | ]	\}}| |qS r$   r		  )r   r'  hpr$   r$   r%   r     rY  z(test_pie_hatch_multi.<locals>.<listcomp>)r   r  r  )r!   r"   r1   r  r	  r  r$   r$   r%   test_pie_hatch_multi  s
   r	  zset_get_ticklabels.pngg?c                  C   s  t d\} }ddg}|d td |d |d  |d td |d |d  |d td |d td |d 	g ddd	g   |d 
g d
dd	g   |d |d   |d |d   |d 	|d   |d 
|d   d S )Nr8   rg   zset_x/yticklabelsr   rw   r7   r   r  r;   rU  r  r   )11121314)r'   r   r   rd   rl   r4   
set_xticksr   
set_yticksset_xticklabelsset_yticklabelsr  
get_yticksrb  get_yticklabels)r-   r#   rD   r$   r$   r%   test_set_get_ticklabels  s   r	  c                  C   sX   t  \} }g d}tjtdd |jj|dd W d   dS 1 s%w   Y  dS )z]
    When labels=None and any kwarg is passed, axis.set_ticks() raises a
    ValueError.
    r  z)Incorrect use of keyword argument 'alpha'rf  r   r  N)r'   r   r   r   r|  rV   	set_ticks)r-   r#   ticksr$   r$   r%   0test_set_ticks_kwargs_raise_error_without_labels  s
   "r	  c                 C   s   |  }|g d |jg ddd |g d |jddgdd	 |jd
dgdd	 |   }|jg dg ddd |g d |jddgd
dgdd	 dS )z
    Test that these two are identical::

        set_xticks(ticks); set_xticklabels(labels, **kwargs)
        set_xticks(ticks, labels, **kwargs)

    )r7   r8   r(  r  r	  bold)
fontweightr  r8   r(  T)r=  r`  ra  N)r   r	  r	  r	  r	  r    r$   r$   r%   test_set_ticks_with_labels  s   	r!	  c                  C   sp  t   } tjtdd | ddgd W d    n1 s w   Y  tjtdd t t	
dd W d    n1 sBw   Y  tjtdd t t	
dd	 W d    n1 sdw   Y  tjtdd t jt	
ddtdd
 W d    n1 sw   Y  tjtdd t jt	
dd	tdd
 W d    d S 1 sw   Y  d S )Nzmust be a sequencerf  r8   rE  r  z
must be 1Dr(  rB  rA  r	  )r'   r   r   r   r   r   r	  r|  r  rd   rl   r   r   r  r$   r$   r%   test_xticks_bad_args  s      "r"	  c                  C   s   t  \} }|td |jtdd  |jdd |jdd dD  t 	  dd |j
 D }|g d	ks@J d S )
Nrw   r   r   r  c                 S   s   g | ]}|qS r$   r$   r  r$   r$   r%   r     s    z.test_subsampled_ticklabels.<locals>.<listcomp>
bcdefghijkc                 S   rc  r$   r  r  r$   r$   r%   r     r  )r  rU  r  r  r  )r'   r   r   rd   rl   rV   r	  r  set_ticklabelsr  get_ticklabels)r-   r#   rT  r$   r$   r%   test_subsampled_ticklabels  s   r&	  c                  C   sl   t  \} }|td |jddg tt	 |j
g d W d    d S 1 s/w   Y  d S )Nrw   r   r  r^  )r'   r   r   rd   rl   rV   r	  r   r   r|  r$	  r,   r$   r$   r%   test_mismatched_ticklabels  s   "r'	  c                  C   s<   t  \} }|ddgddg |ddg |g  d S )Nr7   r8   )r'   r   r  r	  r	  r,   r$   r$   r%   test_empty_ticks_fixed_loc  s   r(	  zretain_tick_visibility.pngc                  C   sF   t  \} }t g dg d t j| dd |jdddd d S )	Nr>  )r   r   r(  Frc  r2   r   r   r  r?  rk  )r'   r   r   setpr	  rO  r,   r$   r$   r%   test_retain_tick_visibility	  s   r+	  c                  C   s   t  \} }tjtdd |ddg W d    n1 sw   Y  t  \} }|g d tjtdd |ddg W d    d S 1 sKw   Y  d S )Nz:set_ticklabels\(\) should only be used with a fixed numberrf  r  r.  r=  z$The number of FixedLocator locations)	r'   r   r   r  r$  r	  r	  r   r|  r,   r$   r$   r%   test_warn_too_few_labels  s   "r,	  c                  C   sr   t  \} }dd }|jt | |g d |dd | j  dd |j	 D }|g dks7J d S )	Nc                 S   s   | dkrdS dS )Nr7   
unit valuer   r$   r   r$   r$   r%   formatter_func&  s   z.test_tick_label_update.<locals>.formatter_func)r   r   r7   r8   r   r  r  c                 S   rc  r$   r  r[  r$   r$   r%   r   /  r  z*test_tick_label_update.<locals>.<listcomp>)r   r   r-	  r   r   )
r'   r   rV   r   r   r	  r  r>  r  r%	  )r-   r#   r.	  
tick_textsr$   r$   r%   test_tick_label_update   s   
r0	  zo_marker_path_snap.pngH   c                  C   s~   t  \} }|d tddD ]}|jddgtd| d|d qtdddD ]}|jd	d
gtd| d|d q*d S )Nr   r7   r   r8   r   )r  rw   r  r   r(  )r'   r   rH  r   r   rd   r   r   )r-   r#   r  r$   r$   r%   test_o_marker_path_snap3  s   
""r2	  c                  C   s  ddg} d}t | d }t| }t| }tdd\}}||  |d | dks.J | ||| d  ||| d  fksDJ | ||| d  ||| d  fksZJ tdd\}}||  |dd | dksuJ | ||| d  ||| d  fksJ | ||| d  ||| d  fksJ tdd\}	}
|
|  |
jd	dd
 |
 dksJ |
 ||| d	  ||| d	  fksJ |
 ||| d  ||| d  fksJ d S )Nr7   rw   r  r   rK  r   r8   )r   r8   皙ɿr  )r3	  r   )	r  r+  r  r'   r   r   rH  r   r   )r  rb   rc   r  r  r  r  r  r  r  r  r$   r$   r%   test_margins>  sF   








r4	  c                  C   s@   t  } |  }|dd | dksJ | dksJ d S )Nr   r   )r'   r   r   rH  get_xmarginget_ymarginr,   r$   r$   r%   test_margin_gettersb  s
   r7	  c                  C   sN   t jd t \} }|ddgddg |jddd | dks%J d S )Nr  r7   r8   r&  r   )r  xmarginr'  )r   r   r   r'   r   r   r   r   r,   r$   r$   r%   test_set_margin_updates_limitsj  s
   r9	  )r   z!margin must be greater than -0\.5rA  r   z@Cannot pass both positional and keyword arguments for x and/or yrF  z&Must pass a single positional argumentc                 C   sT   t j| |d t }| }|j|i | W d    d S 1 s#w   Y  d S r  )r   r   r'   r   r   rH  )r  r  r  rg  r-   r#   r$   r$   r%   test_margins_errorsr  s
   "r:	  c                  C   s&   t  \} }|d |dg d S Nr7   )r'   r   ro  r,   r$   r$   r%   test_length_one_hist  s   
r<	  c                  C   s  t  } |  }|dd | dksJ | dksJ |jdd | dks,J | dks4J |jdd | dksBJ | dksJJ |dd | dksXJ | dks`J |jdd | dksnJ | dksvJ |jdd | dksJ | dksJ d S )	Nr   r  )r   r  r  )r.  )r   r  )r   )r  r  )	r'   r   r   r  
get_xboundr   r  
get_yboundr   r,   r$   r$   r%   test_set_xy_bound  s(   r?	  c                  C   s8   dgd } t dd\}}|| |  |t  d S )Nrw   r   r7   )r'   r   rx  r  r  r  )mylistr-   r#   r$   r$   r%   test_pathological_hexbin  s   
rA	  c                  C   s(   t  \} }|jddgddgd d d S )Nr7   r8   r#  r  r,   r$   r$   r%   test_color_None  s   rB	  c                  C   s6   t  \} }|jddgddd }d| ksJ d S )Nr   r7   r  r  )r'   r   r   r  r-   r#   r  r$   r$   r%   test_color_alias  s   rD	  c                  C   s4   t  \} }|jtdgd tdd |  d S )Nr   r   r  )r'   r   ro  r   rG   r,   r$   r$   r%   test_numerical_hist_label  s   rE	  c                  C   sN   t  \} }d}d}|dd|dg}|jtdgd |d |  d S )Ns    很漂亮, römän chäráctèrss   שלוםzutf-8zhi aardvarkr   r   r  )r'   r   decodero  r   rG   )r-   r#   r   r  rT  r$   r$   r%   test_unicode_hist_label  s   rG	  c                  C   sD  t jdd } t \}}||  |j  |jj	 }|jj
 dks(J |j  |j  |jj	 }|d |d krH|d |d ksJJ |jj
 dksTJ t \}}||  |j  |jj	 }|jj dkstJ |j  |j  |jj	 }|d |d kr|d |d ksJ |jj dksJ d S )	Nrw   g^ 9^;rN   r   r7   rA   r=   rM   )rd   re   r'   r   r   r>  r  rX   
offsetTextr  rW   
tick_rightrV   get_verticalalignmentr   )r  r-   r#   beforeafterr$   r$   r%   test_move_offsetlabel  s*   



$



$rM	  zrc_spines.pngc                  C   sF   ddddd} t |  t  W d    d S 1 sw   Y  d S )NF)zaxes.spines.leftzaxes.spines.rightzaxes.spines.topzaxes.spines.bottom)rS   r   r'   r   )r*  r$   r$   r%   test_rc_spines  s   
"rN	  zrc_grid.pngc               	   C   s   t  } ddd}ddd}ddd}|||g}t|dD ] \}}t| | dd| W d    n1 s8w   Y  qd S )NTr   )r5  zaxes.grid.axisr1   r2   r7   r   )r'   r   r  rS   r   r   )r-   rc_dict0rc_dict1rc_dict2r4  r;  r*  r$   r$   r%   test_rc_grid  s"   
rR	  c                  C   s   ddddd} t j| dR t  }|ddd}|j}|j}|jd r&J |jd s-J |jd r4J |jd s;J |jd sBJ |jd rIJ |jd sPJ |jd rWJ W d    d S 1 sbw   Y  d S )NFT)rx  ry  
ytick.leftytick.rightr  r7   tick1Ontick2Onr'   r   r   r   rV   rX   _major_tick_kw_minor_tick_kwrU  r-   r  xaxyaxr$   r$   r%   test_rc_tick  s"   "r^	  c               	   C   s   ddddddddd} t j| dR t  }|ddd}|j}|j}|jd r*J |jd s1J |jd r8J |jd s?J |jd rFJ |jd sMJ |jd rTJ |jd s[J W d    d S 1 sfw   Y  d S )NTF)ry  rT	  rx  rS	  zxtick.minor.bottomzxtick.major.bottomzytick.major.leftzytick.minor.leftrU	  r7   rV	  rW	  rX	  r[	  r$   r$   r%   test_rc_major_minor_tick  s&   "r_	  c                  C   s   t d} t g d}t \}}|| |d |d | | }}t 	|t 	|ks3J |
 dks;J t|jddjd t|jd	djd
 d S )Nr(  )r   r  r  r  mor  r7   T)originalg      ?r   re  re  F)g333333?r   g      ?re  )rd   rl   rF  r'   r   r   r  r   r   r  r  r   r  r  )r1   r2   r-   r#   r   r   r$   r$   r%   test_square_plot1  s   

rc	  c                   C   s  t t td  W d    n1 sw   Y  t t td d  W d    n1 s0w   Y  t t ttdtd W d    n1 sQw   Y  t t ttddtdddf W d    d S 1 s{w   Y  d S )NrL  r  r   rA  r   r7   )	r   r   r|  r'   r   rd   rq  rl   r   r$   r$   r$   r%   test_bad_plot_args@  s   *"rd	  zxy, clsr$   ))r   r:  )r8   r  )r7   r8   r(  rR  r:  r7  )r(  r   r  )r   r(  c                 C   s0   t  \}}t|jg | |R  |ksJ d S r  )r'   r   r  r  )r  r  clsr-   r#   r$   r$   r%   test_pcolorfastK  s   $rf	  c                  C   s`   t  \} }tjtdd |tdtdtd W d    d S 1 s)w   Y  d S )Nz)the given X was 1D and the given Y was 2Drf  r  )r(  r:  r  )r'   r   r   r   r   r  rd   emptyr,   r$   r$   r%   test_pcolorfast_bad_dims^  s   ""rh	  c               	   C   s   t  \} }tjtdd |tdtdtj	dd W d   n1 s*w   Y  tjtdd |td	tdtj	dd W d   dS 1 sTw   Y  dS )
am  
    Test that the sizes of X, Y, C are compatible for regularly spaced X, Y.

    Note that after the regualar-spacing check, pcolorfast may go into the
    fast "image" mode, where the individual X, Y positions are not used anymore.
    Therefore, the algorithm had worked with any regularly number of regularly
    spaced values, but discarded their values.
    zKLength of X \(5\) must be one larger than the number of columns in C \(20\)rf  r   r  rw   rQ  NzHLength of Y \(5\) must be one larger than the number of rows in C \(10\)rb  )
r'   r   r   r   r|  r  rd   rl   re   rx   r,   r$   r$   r%   ,test_pcolorfast_regular_xy_incompatible_sizee  s   	&&"ri	  c                  C   s   t jddddd\} }|d d |d d |jD ]}| dks&J | dks.J q|d d |d d |jD ]}| dksJJ | dksRJ q@d S )Nr8   Tr  r  r&  rK  r  )r'   r   r  r  r  
get_yscale
get_xscaler-   r  r#   r$   r$   r%   test_shared_scalez  s   

rm	  c                   C   sr   t t tjdd W d    n1 sw   Y  t t tjdd W d    d S 1 s2w   Y  d S )NTr  rW  )r   r   r   r'   r`  r$   r$   r$   r%   test_shared_bool  s   "rn	  c                   C   s   t tddg dS )z4Violin plot should handle point mass pdf gracefully.r   N)r'   r]  rd   rF  r$   r$   r$   r%   test_violin_point_mass  s   ro	  c               	   C   s   t dtdgt dtdg } t ddg dg dg dgtdtdd g}t d|}t d	|}t dg gt dg g t dg d g t d	g d g }| | }| | }| | | }g ||||S )
Nr1   r   r2   r  r7   )r7   r7   r7   r7   r7   )r8   r   r(  r)  )r
   rd   rl   r   )base_xy
err_cyclerxerr_cyyerr_cyrg	  	xerr_only	yerr_onlyboth_errr$   r$   r%   generate_errorbar_inputs  s*   $

rw	  c                 C   s$   t  }|jdi | }|  d S )Nr$   )r'   rY  r%  r_  )r  r#   ebr$   r$   r%   test_errorbar_inputs_shotgun  s   ry	  dash_offsetc                  C   sT   t  \} }tdd}t|}tdddD ]}|j||| |dfddd qd S )	Nr   rw   r   r8   rw   rw   r   r  )r  rC  r   )r'   r   rd   r   	ones_liker   r   )r-   r#   r1   r2   r  r$   r$   r%   test_dash_offset  s   
r}	  c                  C   s   t  \} }|jddd |j }|d d| j ksJ |jddd |j }|d dks1J |jdd d |j }|d tjd d | j ksMJ d S )	Naardvarkr  rl  rA  g?r  zaxes.titlepadr  )r'   r   r4   titleOffsetTrans
get_matrixr  rS   r   )r-   r#   r  r$   r$   r%   test_title_pad  s   


$r	  c                  C   s   t  \} }dt jd< |d |jddd |jddd d|jddks(J d|jddks2J d|jddks<J tt |jdd W d    n1 sRw   Y  tt |jddd W d    d S 1 sow   Y  d S )	NrU   zaxes.titlelocationr~	  rN   r>   rA   r  fail)r'   r   r   r4   	get_titler   r   r|  r,   r$   r$   r%   test_title_location_roundtrip  s   

"r	  r  c                 C   s   t jdd| d\}}|d jddd |d jddd |  |d j \}}|d j \}}||  kr=d	ks@J  J d S )
Nr8   r7   r  r   r`  ir	  ra  r   )r'   r   r4   rs  r0   r  )r  r-   r  r1   r  rJ  r$   r$   r%   test_title_location_shared  s    r	  z
loglog.pngc                  C   sN   t  \} }tdd}|j||d dd |jddd |jd	dd
d d S )Nr7   r  r   r   rE  r  r8   )rk  r  r   r=  )rk  r  r?  )r'   r   rd   rl   r  rO  )r-   r#   r1   r$   r$   r%   test_loglog  s
   r	  ztest_loglog_nonpos.pnggV-?)r  r   r   c                  C   s   t dd\} }tdd}|d }d|d< d|d< ttg dd	d
|jD ]?\\}}}||krI|r>|j||d d	|d q&|j||d d	d q&|j||d d	d |r\|jd|d |re|j	d|d q&d S )Nr   r7   r  g      r:  r  r(  )rn  r  r   r8   )r_  )rC  nonpositiverE  r&  )r	  )
r'   r   rd   rl   r  r   r  r  r  r  )r-   r  r1   r2   mcymcxr#   r$   r$   r%   test_loglog_nonpos  s&   r	  c                  C   s~  t  \} }|g d | d dksJ t  \} }|g dg d | d dks1J t  \} }|g dg d | d dksKJ t  \} }|t	d | dksaJ | dksiJ t  \} }|
t	d | dksJ | dksJ t  \} }|td | d dksJ t  \} }|t	d | dksJ | dksJ d S )Nr7  r   )r7   r7   r7   r7   r{	  r  rw   )r  g      #@)r'   r   r   r>	  r  r  r=	  r"  rd   rq  r  ro  rl   r   r,   r$   r$   r%   test_axes_margins  s0   r	  )rq  c                 C   s    dd }dd }||d| j  S )Nc                 S   <   |   }|  | dd | j  }|d dksJ d S Nr   r   r   r  )r
  r_  r  rV   get_major_locatorr#   r  r"  r$   r$   r%   	_helper_x/  
   z&shared_axis_remover.<locals>._helper_xc                 S   r	  r	  )r	  r_  r  rX   r	  r	  r$   r$   r%   	_helper_y6  r	  z&shared_axis_remover.<locals>._helper_yr  )param)requestr	  r	  r$   r$   r%   shared_axis_remover-  s   r	  )rY  r   subplots_sharedrG  c                 C   s   | j dkrt }| }||fS | j dkr t \}}||fS | j dkr:tjddddd\}}|d d }||fS | j dkrJt }|g d	}||fS )
NrY  r   r	  r8   r  r  r   rG  r   r   r  r  )r	  r'   r   rY  r   rG  )r	  r-   r#   r  r$   r$   r%   shared_axes_generator@  s   
	


r	  c                 C   s   | \}}|| d S r  r$   )r	  r	  r-   r#   r$   r$   r%   test_remove_shared_axesQ  s   r	  c                  C   s`   t jddddd\} }|d d }|d d  }|  |dd t|d d  | d S )Nr8   r  r  r   r7   r   )r'   r   r   r_  r  r   )r-   r  r#   	orig_xlimr$   r$   r%   test_remove_shared_axes_relimW  s   r	  c                  C   s   t ddd} t j| j| jf}tjddddd\}\}}|dd	 |dd	 |	| | | |	| | | |
 s@|
 rBJ | sJ| rLJ | |   kr[d
ks^J  J | |   krmd
kspJ  J d S )Nir  r  r8   r7   Tr  rr  )r	  rr  )rd   rl   re   random_sampler  r'   r   r  r  r  rM  rO  r   r   )r  r  r-   r  r  r$   r$   r%   test_shared_axes_autoscale`  s   $(r	  c                  C   s   t  \} }|j jdd |dd |d | j  t	|j  dks+J |
dd | j  t	|j  dksBJ d S )Nro  r  r   rr  r  r8   )r'   r   rX   r	  
set_paramsr  r  r>  r  r  r  r,   r$   r$   r%   test_adjust_numtick_aspectq  s   


r	  c                  C   sP   t  dd} | jD ]}g | g |   kr"g dks%J  J qd S )Nr(  r=  )r'   r   r   r  r  r	  rC  r$   r$   r%   test_auto_numticks}  s   
2r	  c                  C   sv   t  \} }dtjd< |ddgddg t| tddd	k	 s'J t|
 td
ddk	 s9J d S )Nround_numbersaxes.autolimit_modeg#B;g     j@缉ؗҜ<rw   ir	  r(  r  r   )r'   r   r   r   r  rd   log10r  rl   r  r	  r,   r$   r$   r%   test_auto_numticks_log  s
   
$(r	  c                  C   s   t  \} }|g d d S )N)r   r   )r'   r   r  r,   r$   r$   r%   test_broken_barh_empty  r  r	  c                  C   s   t  \} }tdddddd}||tjddfgddg}| d jd t|ks/J | d jd	 t|d
 ksAJ dS )z9Check that timedelta works as x, dx pair for this method.r  r  rE  r   r7   r  r8   r  r8   r   gUUUUUU?N)	r'   r   r  r  rS  r  verticesr  r  )r-   r#   d0ppr$   r$   r%   test_broken_barh_timedelta  s
    (r	  c                 C   sD   | j ddd}td}tjdd}t \}}|||| d S )Nz
2000-01-01rw   )r&  rQ  r  rE  )r*  rd   rl   re   rx   r'   r   r  )r.  timedepthr  r-   r#   r$   r$   r%   test_pandas_pcolormesh  s
   
r	  c                 C   sj   t jdddd}t tt|}| ||d}t }|t |j	d dk 
 }|jdd	|d
 d S )Nz2005-02z2005-03r  r  )rP  r  r8   r7   rP  r  r7  )rd   rl   r  r   r  rJ  r'   rY  rF  rM  r  r   )r.  rP  r  rN  r#   without_zero_indexr$   r$   r%   test_pandas_indexing_dates  s   r	  c                 C   sF   | j tjjddg dg dd}t \}}|jdddd	|d
 d S )N)r   r(  r  )r1   r2   xeyer  )columnsrM  r1   r2   r	  r	  r  )rJ  rd   re   uniformr'   r   r%  rV  r$   r$   r%   test_pandas_errorbar_indexing  s   r	  c                 C   s:   |  g dg dd}t \}}||j|d  d S )Nr  )r:  r7   r8   )XXYYr	  )rJ  r'   r   r   rM  rV  r$   r$   r%   test_pandas_index_shape  s   r	  c                 C   s8   | j g dd}|jdd  }t \}}|| d S )N)
r7   r8   r8   r   r   r(  r(  r(  r(  r   r7  r7   )rk  ilocr'   r   ro  )r.  ser_1ser_2r-   r#   r$   r$   r%   test_pandas_indexing_hist  s   r	  c                 C   sh   |  tdtdd}td\}}|j|j|d dkdf |j|d dkdf dd |j  d S )Nr8   r6  r7   r   r  rU   r  )rJ  r   r'   r   r  r?   r>  r  rV  r$   r$   r%   test_pandas_bar_align_center  s   r	  c            	      C   s   t  j} i | jdd}i | jdd}| j| jdd}| j| jdd}||ks+J ||ks1J | jdddd	d
 i | jdd}i | jdd}| |}| |}||ksZJ | j|ksaJ ||ksgJ | j|ksnJ d S )NrW  r>  r=  T)reverser   r  r  r   )rg  rm  rj  r?  )r'   r`  rX   get_tick_params_translate_tick_paramsrY	  rZ	  set_tick_params)	r  initial_major_style_translatedinitial_minor_style_translatedtranslated_major_kwtranslated_minor_kwnew_major_style_translatednew_minor_style_translatednew_major_stylenew_minor_styler$   r$   r%   test_axis_get_tick_params  s,   


r	  c                  C   s   t  } | jjdddd | jjd jdksJ | jjd j dks%J | jjd j	 dks2J | jjd j dks?J d S )	Nr   r  r  )rg  rm  rj  r   r  r  r  )
r'   r`  rX   r	  r_  r|  	tick1liner  r  r  )axis_1r$   r$   r%   .test_axis_set_tick_params_labelsize_labelcolor  s   
r	  c                  C   s   t  } | jddddd | j| jfD ]2}|jd j dks!J |jd j dks-J |jd j	 dks9J |jd j
 dksEJ qd S )Nr  r   r   r  )ro  rq  rp  rr  r   r/  )r'   r`  rO  rV   rX   r_  rZ  r  get_linewidthr  r%  )r#   r  r$   r$   r%   test_axes_tick_params_gridlines  s   
r	  c                  C   s   t  } | jdddd | jdddd | jjd j du s!J | jjd j du s.J | jjd j du s;J | jjd j du sHJ d S )NFTrW  )	labelleft
labelrightr?  r=  r   )	r'   r`  rO  rX   r_  r  r  label2r`  r  r$   r$   r%    test_axes_tick_params_ylabelside  s   r	  c                  C   s   t  } | jdddd | jdddd | jjd j du s!J | jjd j du s.J | jjd j du s;J | jjd j du sHJ d S )NTFrW  )labeltoplabelbottomr?  r=  r   )	r'   r`  rO  rV   r_  r  r  r	  r`  r  r$   r$   r%    test_axes_tick_params_xlabelside  s   r	  c                  C   s4   t   } | jtdd d\}| dksJ d S )Nr}   r!  r  )r'   r   r   r   r   r%  )r#   lnr$   r$   r%   test_none_kwargs$     r	  c                  C   sT   g d} t jtj| tjdg ddd}t|j| D ]\}}|jd |ks'J qd S )Nr7  r  )r8   r   r(  r   r  r  r   )r'   r  rd   rF  uint8r  r4  r  )r  r  r  r1   r$   r$   r%   test_bar_uint8*  s
    r	  zdate_timezone_x.pngr   c                  C   s   dd t dD } tjdd tddd ttj tj| dgd dd	 W d    n1 s2w   Y  tddd ttj tj| dgd d
d	 W d    d S 1 s[w   Y  d S )Nc              
   S   (   g | ]}t j d dd|tjddqS   r8   rc  Canada/Easternr   tzinfor  r*  tzgettzr   r$   r$   r%   r   4  
    

z(test_date_timezone_x.<locals>.<listcomp>r   rQ  r  r   r8   r7   r	  )r	  UTC	r   r'   r   r`  r   r  r   r<  r=  
time_indexr$   r$   r%   test_date_timezone_x1  s   "r	  zdate_timezone_y.pngc                  C   s   dd t dD } tjdd tddd ttj tjdgd | dd	d
d W d    n1 s4w   Y  tddd ttj tjdgd | dd	d
d W d    d S 1 s_w   Y  d S )Nc              
   S   r	  r	  r	  r   r$   r$   r%   r   G  r	  z(test_date_timezone_y.<locals>.<listcomp>r   r	  r   r8   r7   r	  FT)r	  xdateydater	  r	  r	  r$   r$   r%   test_date_timezone_yD  s   "r	  zdate_timezone_x_and_y.pngc                     s   t jj  fddtdD } tjdd tddd tt	j
 tj| | dd	d
 W d    n1 s6w   Y  tddd tt	j
 tj| | dd	d
 W d    d S 1 s]w   Y  d S )Nc              	      s    g | ]}t j d dd| dqS )r	  r8   rc  r	  r  r   r	  r$   r%   r   [  s    z.test_date_timezone_x_and_y.<locals>.<listcomp>r   r	  r   r8   r7   r	  T)r	  r	  z
US/Eastern)r  timezoneutcr   r'   r   r`  r   r  r   r<  r=  r	  r$   r	  r%   test_date_timezone_x_and_yW  s   
"r	  zaxisbelow.pngc                  C   s   t  jdddd} d}t| |D ]A\}}|jddddd tjd	d
d}|| |jdddd |j	ddddd |j
d d  d || | |ksRJ qd S )Nr   T)r  r  rX  )Fr  Tr  rw   r  r  r  r"  r#  r;   r  )r   r   r/  F)r=   rM   rN   rA   )r'   r   r   r  r   r@  r  r?  r[  rO  rK  r   set_axisbelowget_axisbelow)r  settingsr#   settingcircr$   r$   r%   test_axisbelowj  s   

r	  c            	      C   s   t jd t jdd\} }| }|d}|d}| j  | j	 }|
|j}|
|j}|| }t|ds>J d S )Nr   r1	  )r  Xlabel2Titler   )r'   r   r   r   r	  r(   r4   r>  r  get_rendererget_window_extentr  r  rd   isclose)	r-   r#   r  xlabel2r0   rendererbbox_y0_titlebbox_y1_xlabel2y_diffr$   r$   r%   test_titletwiny~  s   



r
  c                  C   sx   t  \} }| jdd | }|d |d |d d}|j| | j	 }|
| |j |ks:J d S )Nr  )r=   Xlabelr	  r	  )r   g(\?)r'   r   r^  r	  r(   r4   r0   rN  r>  r	  _update_title_positionr  )r-   r#   r  r   r
  r$   r$   r%   test_titlesetpos  s   




r
  c                  C   sP   d t jd< t \} }|jd |d | j  |j	
 d dks&J d S )Naxes.titleyr=   
xlabel topr7   p=
ף?)r   r   r'   r   rV   r   r4   r>  r  r0   r  r,   r$   r$   r%   test_title_xticks_top  s   


r
  c                  C   sX   d t jd< t \} }|jdddddd |d | j  |j	 d dks*J d S )Nr	
  r1   Tr  rM   r=   r	  r	  r

  r7   r
  )
r   r   r'   r   rO  r4   r>  r  r0   r  r,   r$   r$   r%   test_title_xticks_top_both  s   


r
  zleft, center))rN   r   )r   rU   )rN   rU   )zleft title movedzcenter title keptzboth titles alignedc                 C   s   d t jd< t \}}|d |j| dd || |  | r1|s1|j d dks/J d S | sB|rB|j	 d dks@J d S |j d }|j	 d }|dksVJ ||ks\J d S )Nr	
  g   vH7BrN   r>   r7   r   )
r   r   r'   r   r  r4   rs  _left_titler  r0   )rN   rU   r-   r#   yleftr  r$   r$   r%   test_title_above_offset  s   


r
  c                  C   s`   d t jd< t } | g d}|jdddddd |d}| j  |	 d dks.J d S )	Nr	
  )r   r  r  r   r1   Tr
  Boor7   r   )
r   r   r'   r   rG  rO  r4   r>  r  r  )r-   r#   ttr$   r$   r%   test_title_no_move_off_page  s   


r
  c                  C   sn   d t jd< t \} }|d |   |j d dksJ |g d |   |j d dks5J d S )Nr	
  r	  r7   )r   r7   r7   r   r$  )	r   r   r'   r   r4   rs  r0   r  
inset_axesr,   r$   r$   r%   test_title_inset_ax  s   

r
  c                  C   sB   t  \} }|g d |jjdd |j  dksJ d S )Ng   @Ag   eAg   FAr  )rm  )r'   r   r   rX   r	  get_offset_textr  r,   r$   r$   r%   test_offset_label_color  s   r
  c                  C   s`   t  \} }|g d |jjddd |j  sJ |jjdd |j  r.J d S )Nr
  FT)label1Onlabel2On)r
  )r'   r   r   rX   r	  r
  r  r,   r$   r$   r%   test_offset_text_visible  s   r
  c                  C   s6   t  \} }|dtddg d  | j  d S )Nr7   r   g-q=gqىE)r'   r   r   rd   rF  r>  r  r,   r$   r$   r%   test_large_offset  s   r
  c                  C   ^   t  \} }dd tdddD }tddt|}tddt| }}||||| d S )Nc              	   S      g | ]}t  d ddd|qS r  r:  r   r	  r  r  r$   r$   r%   r     r   z#test_barb_units.<locals>.<listcomp>r   r  rw   r   r  )r'   r   r   rd   r   r  barbsr-   r#   rP  r2   ur  r$   r$   r%   test_barb_units  
   r$
  c                  C   r
  )Nc              	   S   r
  r 
  r  r  r$   r$   r%   r   	  r   z%test_quiver_units.<locals>.<listcomp>r   r  rw   r   r  )r'   r   r   rd   r   r  quiverr"
  r$   r$   r%   test_quiver_units  r%
  r'
  c                  C   sn   t j} t \}}tdD ]'}|td\}|tdtd}|D ]}| | | | ks3J q#qd S )Nr   r   )	r  to_rgbr'   r   r   r   r  r  r  )r(
  r-   r#   r  r	  brsbrr$   r$   r%   test_bar_color_cycle  s   r+
  c                  C   s  t dd\} \}}|ddgddg |ddgddg |jjddd |jjddd |jddD ]
}| dks>J q4|jddD ]
}| dksOJ qE|j	d	d
d |j	ddd |jddD ]
}| d
ksnJ qd|jddD ]
}| dksJ qud S )Nr7   r8   r   r   r   )r?  r  r  r>  r1   5   )r  rn  r2   r   )r  r  rW  )
r'   r   r   rV   r	  rX   rb  r  r	  rO  )r  r#   r  textr$   r$   r%   test_tick_param_label_rotation  s    r.
  c            	      C   s  t  \} }tdD ]"}|tdtd}td| }t|  t|ks,J q
tddD ]"}|	tdtd}td| }t|  t|ksTJ q2td}dD ] }|jtdtdfi |di}t|  t|ks|J q\td}t
ddgddD ]8\}}|jtdtdfi |di}td| }t|  t|ksJ t|  t|ksJ qd S )	Nr   r  r  r  )r-  rh  r   r.  r&  )r"  )r'   r   r   r  r  r  r}  r5  squeezer  r  r6  )	r-   r#   r  cctargetaledge_targetelface_targetr$   r$   r%   test_fillbetween_cycle,  s(   
 
 r6
  c                  C   s   dt jd< t  \} }d}|| |ddgddg | \}}|j }|||g\}}|ddg\}}	|	| | }
t	||
 ||
 g||	g d S )Nr  r	  r_  rw   r   )
r'   r   r   set_xmarginr  r   rV   r  r,  r   )r-   r#   marginxlim0xlim1r,  xlim0txlim1tx0tx1tr  r$   r$   r%   test_log_marginsH  s   


r?
  c                  C   s   d} t | t | }}t | d }t \}}tt |j|||d W d    n1 s2w   Y  tjdd |j||dd W d    n1 sPw   Y  |j||dg|  d d S )Nr   r7   r  rS  rf  )r   r   r   )	rd   rl   r'   r   r   r   r|  r9   r  )r  r1   r2   r  r-   r#   r$   r$   r%   test_color_length_mismatchV  s   r@
  c                   C   s   t jdgdd t   d S )Nr   Labelr  )r'   r  rG   r$   r$   r$   r%   test_eventplot_legendb  s   rB
  r  zlineoffsets cannot be emptyr  zlinelengths cannot be emptyr  zlinewidths cannot be emptyr  zlinestyles cannot be emptyre  zalpha cannot be emptyz!positions must be one-dimensionalz5lineoffsets and positions are unequal sized sequencesz5linelengths and positions are unequal sized sequencesz4linewidths and positions are unequal sized sequencesz4linestyles and positions are unequal sized sequencesz/alpha and positions are unequal sized sequencesz0colors and positions are unequal sized sequencesc                 C   r  r  )r   r   r'   r  r  r$   r$   r%   test_eventplot_errorsg  s   "rC
  c                  C   sv   t  \} }|tdd |jddtddd |jddgddgdd\}}| |   kr6dks9J  J d S )Nr(  r7   r   )rN   r  )r   r   r   r  r=  )r'   r   r  r   r  r  )r-   r#   rect1rect2r$   r$   r%   test_bar_broadcast_args  s
   (rF
  c                   C   s   t ddgddg tt t tj W d    n1 s w   Y  tt t tj W d    n1 s;w   Y  tt t 	tj W d    n1 sVw   Y  tt t 	tj W d    d S 1 srw   Y  d S r  )
r'   r   r   r   r|  r   rd   r  r  r   r$   r$   r$   r%   test_invalid_axis_limits  s   "rG
  r  r  r&  r  c                 C   s6   t  }|g d ||  || |  d S )Nr  )r'   r`  r   r  r  r  )r  r  r#   r$   r$   r%   test_minorticks_on  s
   

rH
  c                  C   sv   t  \} }|dd | }|ddgddg t  \}}|dd |ddgddg t|jj|jj d S )Nr7   r8   r   r   )r'   r   rg  r
  r   r   r  	intervalx)r-   r#   xtwinr  r  r$   r$   r%   test_twinx_knows_limits  s   rK
  c                   C   s   t jddgddgddd d S )Nr   r7   r   )r  rC  )r'   r   r$   r$   r$   r%   test_zero_linewidth  s   rL
  c                  C   s<   t  \} }|jg g g dd |jg g g dd |  d S )Nzempty y)r(  r/   zempty x)r)  r/   )r'   r   r%  rG   r,   r$   r$   r%   test_empty_errorbar_legend  s   rM
  c                 C   s\   t ddd}dd |D }dd |D }dd |D }|  || | || d S )Nr  rw   r   c                 S   s0   g | ]}d |d  d|d   d|  d qS )g@r   r  r8   g\(\@r|  r$   r   r$   r$   r%   r     s   0 z%test_plot_decimal.<locals>.<listcomp>c                 S   r  r$   r   r  r$   r$   r%   r     r  c                 S   r  r$   r   r  r$   r$   r%   r     r  )rd   rl   r   r   )r!   r"   r  r  r1   r2   r$   r$   r%   test_plot_decimal  s   rN
  c                 C   s0   |   jddddd |  jddddd d S )Nr   r   r)  r   )r  re  r'  )r   r   )r!   r"   r$   r$   r%   test_markerfacecolor_none_alpha  s   rO
  c                  C   sn   dt jd< dt jd< t  \} }|| j }|d || j }|j|jk s-J |j|jk s5J dS )z5Test that tick padding gets turned off if axis is offr  rv  zytick.directionoffN)	r'   r   r   get_tightbboxr>  r	  r  r  r  r-   r#   bbbb2r$   r$   r%   test_tick_padding_tightbbox  s   


rU
  c                  C   s.  d\} }t jtdd| |tdd|  | f \}}t |d t d||  t |  }t \}}||||ddddf  |d |	  dd	g}d
dg}|d |d |d |d  |d |d  g}	|j
|	d}
|
jdu swJ |j  t ddgd	dgg}t |
j  |ksJ dS )z:
    Ensure that inset_ax argument is indeed optional
    r_  r_  r7   r   rw   Nr   r   r   333333@r8   r  r   )boundsr   )rd   mgridr  r  r\  r'   r   r  r  apply_aspectindicate_inset
connectorsr>  r  rF  r  	rectangleget_bbox
get_points)dxdyr2   r1   rh  r-   r#   r   r   r  insetxxr$   r$   r%   
test_inset  s&   
*
,

 rd
  c                  C   s  d\} }t jtdd| |tdd|  | f \}}t |d t d||  t |  }t \}}||||d dd df  |d |	  |
g d}||||d dd df  |dd	g |d
dg ||  ttj ||\}}	W d    n1 sw   Y  |j  t|	dksJ t ddgd	dgg}
t |  |
ksJ t ddgddgg}
t jj|  |
dd d S )NrV
  r7   r   rw   r   r   )ru  ru  r  r  r   rW
  r8   r  r(  r   g=
ףp=?gKXc'?g(\?g?g-C6?rtol)rd   rY
  r  r  r\  r'   r   r  r  rZ
  r
  r  r  r  r   r  r   r<  indicate_inset_zoomr>  r  r  rF  r  r^
  r_
  r   r   r  )r`
  ra
  r2   r1   rh  r-   r#   axin1recr\
  rc
  r$   r$   r%   test_zoom_inset  s:   
*




rj
  zinset_polar.pngc                  C   sd   t  \} }|jg ddd}t|tsJ tddd}dtj | }||| ||| d S )N)r   r   r  r  Tr  r   r8   r  )	r'   r   r
  r   r   rd   rl   r  r   )r  r#   axinsr"  r  r$   r$   r%   test_inset_polar  s   rl
  c                  C   s0   t  \} }|jg ddd}t|tsJ d S )Nr   r   r   r   hammerrU  )r'   r   r
  r   r   r  r#   rk
  r$   r$   r%   test_inset_projection(  s   rp
  c                  C   s4   t  \} }|jg dtjd}t|tjsJ d S )Nrm
  )r  )r'   r   r
  AAr   r   ro
  r$   r$   r%   test_inset_subclass.  r	  rr
  
x_inverted
y_invertedc                 C   s   t dd\}\}}td}|||d | r|  |r"|  |g d|}|j\}}}	}
| r5dnd}|r;dnd}||	j	d |j	d   dksMJ ||
j	d |j	d   dks]J ||j	d |j	d   dksmJ ||
j	d |	j	d   dks}J dS )	zR
    Test that the inset lines are correctly located with inverted data axes.
    r7   r8   rw   r   )r8   r8   r   r(  r   r   N)
r'   r   rd   rl   r   r  r  r[
  r\
  r  )rs
  rt
  r-   r  r  r1   rb
  
lower_left
upper_leftlower_rightupper_rightsign_xsign_yr$   r$   r%   test_indicate_inset_inverted4  s   
   $r{
  c                  C   s   t  \} }|d |jg ddd t| jdsJ |d |jg ddd t| jds8J |d |jg dd	d t| jdsQJ d S )
Nr  )r   r   r  r  r   r>  r   r   ra	  rJ  active)r'   r   r  rN  rd   r  r  r  r,   r$   r$   r%   test_set_positionN  s   


r}
  c                  C   sd   t  \} }|jj| j }|ddd dd |jj| j }tj	j
| | dd d S )N)rM  rM  r  r  ri  Fgư>re
  )r'   r   rK  rM   r	  r>  r	  _set_view_from_bboxrd   r   r   r_
  rR
  r$   r$   r%   !test_spines_properbbox_after_zoom[  s   r
  c                  C   sV  t  \} }d}d}|| |j|d |d d |d tjj|| dd tjj||	 dd |d	 d
}d}| }|	 }tj|d |d  d tj|d |d  d tjj||dd tj|| |d | }|	 }tj|d |d  d tj|d |d  d tjj||dd tjj||dd d S )Nr  r   )r   r8   r   r7   r  )r  r  r   r	  atol)r  r  r   )gB!Bؿg{?)g      ڿg     ?r   r   )r  r  r   r   )
r'   r   r  r  r~
  rd   r   r   r   r   )r-   r#   r   r   r  new_ylimres_xlimres_ylimr$   r$   r%   test_limits_after_scroll_zoome  s0   



r
  c                  C   sH   t  \} }t|  |tjdd | j }tj	|
|jd d S )Nr7   r
  i  )r'   r   r   r-
  rd   r  r>  r	  r   r   rQ
  r  )r-   r#   r
  r$   r$   r%   test_gettightbbox_ignore_nan  s
   
r
  c                 C   s^   t dd}| jtjjdd|d}| jtjjdd|d}| jg d|d}t||| d S )Nrw   r	  rR  r  rU  )r7   r7   r7   r7   r7   r   r   r   )r   rk  rd   re   r	  r'   r9   )r.  r  r1   r2   r;   r$   r$   r%   "test_scatter_series_non_zero_index  s
   
r
  c                   C   s"   t g g  t jg g g g d d S )N)r  r;   )r'   r9   r$   r$   r$   r%   test_scatter_empty_data  s   r
  zannotate_across_transforms.png)r   r  r   c               	   C   s   t ddd} t |  t |  }tjdd\}}|| | |g d}|d |j	
d |j
d |jd	| d
 |d
 f|jd|jtddd d S )Nr   rw   r  )gQ@r   r   )r  r   r   r   r   Fr   r  r  r  )r  )r  r  r  r  r  )rd   r   r  r  r'   r   r   r
  r  rV   r   rX   r  	transDatar8  rZ  )r1   r2   r-   r#   rk
  r$   r$   r%   test_annotate_across_transforms  s   

r
  c                   @   s,   e Zd ZdZdZdd Zdd Zdd ZdS )	_Translationr7   c                 C   s
   || _ d S r  r`
  )rl  r`
  r$   r$   r%   r    r  z_Translation.__init__c                 C   s
   || j  S r  r
  )rl  r  r$   r$   r%   r,    r  z_Translation.transformc                 C   s   t | j S r  )r
  r`
  rk  r$   r$   r%   ry    ro  z_Translation.invertedN)rp  rq  rr  
input_dimsoutput_dimsr  r,  ry  r$   r$   r$   r%   r
    s    r
  zsecondary_xy.pnggS㥛?c                  C   s   t jddddd\} }dd }t|D ]V\}}|tddtdd |d	kr-|j}n|j}|d
||fd |ddd dd fd |ddd dd fd |d ||d	kr[dndtdd |d|j	d qd S )Nr7   r8   r  T)rW  r  c                 S   8   t jdd d|  W  d    S 1 sw   Y  d S Nr  r  r7   rd   r  r  r$   r$   r%   invert     $z!test_secondary_xy.<locals>.invertr  r   r   	functionsr  c                 S   s   d|  S Nr8   r$   r  r$   r$   r%   r    r  z#test_secondary_xy.<locals>.<lambda>c                 S   s   | d S r
  r$   r  r$   r$   r%   r    r  rt  c                 S      | d S r
  r$   r  r$   r$   r%   r    r  c                 S   r
  )Nr   r$   r  r$   r$   r%   r    r  r  r=   rA   g      @r  )
r'   r   r  r   rd   rl   secondary_xaxissecondary_yaxisr
  r
  )r-   r  r
  nnr#   secaxr$   r$   r%   test_secondary_xy  s   r
  c                  C   s
  t  \} }|tddtdd tt |jddd d W d    n1 s-w   Y  tt |d W d    n1 sGw   Y  tt |	d W d    n1 saw   Y  tt
 |jdd	d
 W d    d S 1 s~w   Y  d S )Nr8   r  r   c                 S   s   d|  S r;	  r$   r  r$   r$   r%   r    r  z%test_secondary_fail.<locals>.<lambda>r
  rA   rM   errorr  )r'   r   r   rd   rl   r   r   r|  r
  r
  r   r,   r$   r$   r%   test_secondary_fail  s   "r
  c                  C   sr   t jdd\} }|tddtdd dd }|jd||fd | j  | d	 t	|
 jg d
 d S )Nr  r   r8   r  c                 S   r
  r
  r
  r  r$   r$   r%   r
    r
  z%test_secondary_resize.<locals>.invertr=   r
  )r:  r(  rb	  )r'   r   r   rd   rl   r
  r>  r  r  r   r  r  )r-   r#   r
  r$   r$   r%   test_secondary_resize  s   

r
  c                  C   s   t jdd\} }|tddtdd dd }|jd||fd}t|j t	j
s.J |  t|j t	js=J |d	 t   t|j t	jsQJ |d
 t   t|j t	j
seJ d S )Nr  r   r8   r  c                 S   r
  r
  r
  r  r$   r$   r%   r
    r
  z'test_secondary_minorloc.<locals>.invertr=   r
  r&  r  )r'   r   r   rd   rl   r
  r   _axisget_minor_locatorr   NullLocatorr  AutoMinorLocatorr  r  
LogLocator)r-   r#   r
  r
  r$   r$   r%   test_secondary_minorloc  s*   





r
  c                  C   sT   t  \} }|d |d}|jt  | j	  t
|j tjs(J d S )Nr&  r=   )r'   r   r  r
  rV   r   r   ScalarFormatterr>  r  r   r   r-   r#   r
  r$   r$   r%   test_secondary_formatter
  s   


r
  c                  C   s*   t  \} }|d}t|dksJ d S )Nr=   z<SecondaryAxis: >)r'   r   r
  r5   r
  r$   r$   r%   test_secondary_repr  s   
r
  zaxis_options.pngc               	   C   s   t dd\} }tdD ]J\}}|d|f dd |d|f | |d|f tjddd	d
d |d|f dd |d|f | |d|f tjddd	d
d qd S )Nr8   r   )scaledr  imager   r'  )r7   g	@)r   r   r   r)  r  r  r7   )r7   g      @)r7   r   )r   r   rd  )r'   r   r  r   r  r3  r@  r  )r-   r^  r;  optionr$   r$   r%   test_axis_options  s   
r
  c                 C   sd  | j   | j  }g }t|j|jgD ](\}}||}|r7tj|j	|j
f|j|jdddddd}| | ||g7 }qg }tg dD ])\}}	|j|	 |}tj|j	|j
f|j|jdddddd}
| |
 ||g7 }qE| }tj|j	|j
f|j|jd	d
dddd}| | |}||}tj|j	|j
f|j|jdddddd}| | |}||||fS )zS
    Helper for the tests below that test the extents of various Axes elements
    ru  r2   r)  Nr   )r  r  r/  r.  r-  r,  r;  )rM   r=   rN   rA   r   r   r  r8   r  r7   )r>  r  r	  r  rV   rX   rQ
  r@  rA  r  r  r  r  r3  rK  r	  )r-   r#   r
  bbaxisr
  axxrS
  axisrbbspinesr   spinerrE
  bbaxrT
  bbtbr$   r$   r%   color_boxes,  sP   







r
  c                  C   sn  t ddi$ tjddd\} }| j  t|  t| |\}}}}W d    n1 s-w   Y  g dg dg}t|D ]\}}tj	j
||  }	t|j|	jdd	 q>g d
g dg dg dg}t|D ]\}}tj	j
||  }	t|j|	jdd	 qfg d}tj	j
| }	t|j|	jdd	 g d}tj	j
| }	t|j|	jdd	 | | jj}
t|
| jdd	 d S )N_internal.classic_modeFr  r  r  rW  )g     ^@g8R@g     @g     @@)g8`U@gX@g     J@g     @r8   )decimal     b@-]@     @F8&@r
  g     @r
  r  r
  Nё\]@r
        @gR!@r
  r
  r
  r
  g]@r
  r
  )g+e`U@gbhR@g{G@g{Gȏ@)r   r'   r   r>  r  rb  r
  r  r  Bboxfrom_boundsr   rX
  r  transformedtransFigurer	  )r-   r#   r
  r
  r
  r
  r1
  r
  r  targetbbaxbbr$   r$   r%   test_normal_axesZ  s8   

r
  c            
      C   s&  t ddi& tjddd\} }| j  |jg g d t| |\}}}}W d    n1 s/w   Y  t|D ]
\}}|d u sBJ q8g dg dg d	g d
g}t|D ]\}}tj	j
||  }	t|j|	jdd qUg d}tj	j
| }	t|j|	jdd g d}tj	j
| }	t|j|	jdd d S )Nr
  Fr  r  r
  xticklabelsyticklabelsr
  r
  r
  r
  r  r
  r
  )r
        ^@r
  r
  )r   r'   r   r>  r  r   r
  r  r  r
  r
  r   rX
  )
r-   r#   r
  r
  r
  r
  r
  r  r1
  r
  r$   r$   r%   test_nodecorator  s,   
r
  c            
      C   s  t ddi- tjddd\} }|jg g d |jjd | j  t	| |\}}}}W d    n1 s6w   Y  g dg d	g d
g dg}t
||D ]\}}tjj| }	t|j|	jdd qNg d}tjj| }	t|j|	jdd g d}tjj| }	t|j|	jdd d S )Nr
  Fr  r  r
  r
  )r^  r  )r
        8@r
  gs~8&@r
  r
  r
  r  r
  r
  )r
  r
  r
  g     @)r   r'   r   r   rK  rM   rN  r>  r  r
  r  r  r
  r
  r   rX
  )
r-   r#   r
  r
  r
  r
  targetsr1
  bbspiner
  r$   r$   r%   test_displaced_spine  s*   
r
  c               	   C   s   g dg dgg dg dgg dg dgg} t g dD ]W\}}tdd	iE tjd
dd\}}|j|d |j  t||\}}}}t ddgD ]\}	}
tj	j
| | |	  }t||
 j|jdd qKW d   n1 sow   Y  qdS )zG
    Switch the tickdirs and make sure the bboxes switch with them
    )r
  r
  r
  +e8&@)r
  r
  r
  r
  )r
  gZӼ8[@r
  88&@)g r\a@r:  r
  r
  )r
  gqq\@r
  r
  )g88b@r
  r
  r
  )ri  r  inoutr
  Fr  r  r
  r  r   r8   r  r
  N)r  r   r'   r   rO  r>  r  r
  r  r
  r
  r   rX
  )r
  dnumdirsr-   r#   r
  r
  r
  r
  r
  r   r
  r$   r$   r%   test_tickdirs  s2   
r
  c            	      C   sn  t ddi tjddd\} }| j  |jddd t| |\}}}}t| |\}}}}g d	g d
g}tdD ]}tj	j
||  }t||d  j|jdd q:tjddd\} }| j  |jddd |  |jdddd | j  t| |\}}}}g dg dg}tdD ]}tj	j
||  }t||d  j|jdd qW d    d S 1 sw   Y  d S )Nr
  Fr  r  r
  r   r  )r?  rj  )r
  g88[@r
  r
  )gCiq\a@r
  r
  r
  r8   r  r
  r=  r   r)	  )r
  gPUUUUUB@r
  gVUUUUT@)gz6P@r
  goTT@r
  )r   r'   r   r>  r  rO  r
  r   r  r
  r
  r   rX
  r  )	r-   r#   r
  r
  r
  r
  r
  r  r
  r$   r$   r%   test_minor_accountedfor  s>   


"r
  c                 C   sR   |  dd | dd |  d}|d |d | dd d S )NrU  FrP
  rV  Ton)r   r  r    r$   r$   r%   test_axis_bool_arguments  s   


r
  c                  C   sr   t  \} }d}d}d}d}|||||g}t|||||fks#J ||f| ks-J ||f| ks7J d S )Nr   rw   r   rQ  r'   r   r  r}  r   r   r-   r#   rb   rc   r  r  rw  r$   r$   r%   test_axis_extent_arg  s   r
  c                  C   sr   t  \} }d}d}d}d}|j||||d}t|||||fks#J ||f| ks-J ||f| ks7J d S )Nr   rw   r   rQ  )rb   rc   r  r  r
  r
  r$   r$   r%   test_axis_extent_arg2   s   r
  c                  C   sD   t jg dg dgdd\} }} |d dksJ |d dks J d S )	Nr  )r   r(  r   r  ro  r  r   r7   r   r  )r'   ro  r  r  r$   r$   r%   test_hist_auto_bins   s    r
  c            
      C   s   t d\} \}}g d}|tjg }||\}}}tjdd ||\}}	}W d    n1 s4w   Y  tj|| tj||	 d S )Nr8   r  r  rk  )r'   r   rd   r  ro  r  r   r   )
r-   r  r  r  nan_datar  r  r  nanbinsnanedgesr$   r$   r%   test_hist_nan_data"   s   r
  c                  C   sD   t jtjddddd\} }} |d dksJ |d dks J d S )	Nrw   ro  r  T)r   r  r   r   r7   )r'   ro  rd   re   rx   r
  r$   r$   r%   test_hist_range_and_density0   s
   r
  c            
      C   sz   t  \} }g d}|j|||ddd}|jj\}}}|jD ]}|D ]
}|j|jks,J q"|D ]
}	|	j|jks9J q/qd S )Nr  r   r   )r1   r  r)  r  r;  )r'   r   r  r%  r  r4  r;  )
r-   r#   r1   barcont	data_linecaplinesbarlinecolsr  capline
barlinecolr$   r$   r%   test_bar_errbar_zorder7   s   
r
  c                  C   sR   t  \} }|  |ddg | dksJ |dg | dks'J d S )Nr   ru  r  r   rA  )r'   r   r  r	  r   r,   r$   r$   r%   test_set_ticks_invertedF   s   r
  c                  C   sv   t jdd} |  }|ddgddg |jddddd	d
d |d | j| }|j	|j
 tdks9J d S )Nr{	  r   r  rt  r&  r  logit)F]tE?g'^P?r7   r  r  r   r  r   r   r  r   r8   )r'   r   r   r   r   rH  r
  transform_bboxr  r  r  r   approx)r-   r#   r   r$   r$   r%   $test_aspect_nonlinear_adjustable_boxO   s   
r
  c                  C   s   t jdd} | g d}|ddgddg |jdddd	d
dd |d |  | t	ddgks8J |
 d	ks@J d S )Nr{	  r   r	  r  rt  r&  )r7   r   r
  )gbeF?r
  r7   r  r
  r   gS[:XL	@g'In?@)r'   r   rG  r   r   rH  rZ
  r   r   r
  r   r,   r$   r$   r%   (test_aspect_nonlinear_adjustable_datalim\   s   
r
  c                  C   s   t  \} }| }|ddg |d | dksJ t  \}}|d |ddgddg |jd	d
d | j	  |j	  |
 }|
 }|
 }t|j|j t|j|j d S )Nr  iX  r7   r   r   r8   r  rR  r  r  r  )r'   r   r
  r   set_box_aspectget_box_aspectrH  r  r>  r  r  r   r  )r  r  axtwinr  r  bb1bbtrT
  r$   r$   r%   test_box_aspectk   s    



r
  c                  C   s   t  \} }|g d | j  |d t  \}}|d |j  |g d | j  |j  | }| }t|j|j d S )N)r   r   re  r   r   )	r'   r   rN  r>  r  r
  r  r   r  )r  r  r  r  r
  rT
  r$   r$   r%   test_box_aspect_custom_position   s   





r
  c                  C   sn   t jddtdddd\} }| j  | j }g }|jD ]}||}||j	|j
g qt||d  d S )Nr8   r   r7   )r  T)rX  r  r   )r'   r   rZ  r>  r  r	  r  r	  r  r  r  r   )r-   r  r
  r  r#   rS
  r$   r$   r%   test_bbox_aspect_axes_init   s   




r
  c                  C   s   t  \} }tjtdd |d W d    n1 sw   Y  tjtdd |d W d    n1 s9w   Y  tjtdd |tj W d    n1 sVw   Y  tjtdd |tj  W d    d S 1 suw   Y  d S )Nzmust be finite and positiverf  r   r   )r'   r   r   r   r|  r  rd   r  r,   r$   r$   r%   test_set_aspect_negative   s   "r   c                  C   s4   t dd\} }|g d | j  |  d S )Nr7   r  )r'   r   r   r>  r  redraw_in_framer,   r$   r$   r%   test_redraw_in_frame   s   
r  c                  C   sB   t  \} }| jdd usJ |d | jdd u sJ d S )N)r  r  F)r'   r   r>  inaxesr   r,   r$   r$   r%   test_invisible_axes_events   s   
r  c                  C   <   dt jd< t  } | j }|D ]}|j dksJ qd S Nr  zlines.markeredgecolor)r'   r   r^  rV   get_major_ticksr	  get_markeredgecolorr#   r	  r\  r$   r$   r%   "test_xtickcolor_is_not_markercolor      

r
  c                  C   r  r  )r'   r   r^  rX   r  r	  r  r	  r$   r$   r%   "test_ytickcolor_is_not_markercolor   r  r  r  ro  )TFNc           
      C   s   t  \}}td}tddd}||| t|d|  d}t|d|  d}t|d|  d}|d u r:| n|}	|d	|d
 |	| ksIJ |j  t	| d	 d S )Nr   r  r   get_autoscale_onr$  limget_r
  )ro  )
r'   r   rd   rl   r   r9   r  r>  r  r   )
r  ro  r-   r#   r1   r2   r  set_limget_lim	post_autor$   r$   r%   test_unautoscale   s   

r  c                 C   s`   | j dddtjd gddg\}d| _|j ddtdtjd dtddd d S )Nr  rU  r   r8   r7   r   e   )r   r   rd   r  get_path_interpolation_stepsr   )r!   r"   r  r$   r$   r%   )test_polar_interpolation_steps_variable_r   s
   $
r  c                  C   s6   t  \} }|dd | j  | dksJ d S )Nr   g&.>)r   g	>)r'   r   r  r>  r  r   r,   r$   r$   r%   test_autoscale_tiny_sticky   s   
r  c                  C   X   dt jd< dt jd< t  } | j }|D ]}|j dks J |j dks)J qd S )Nr  rw  r   r{  )r'   r   r^  rV   r  r	  r  r  r	  r$   r$   r%   &test_xtickcolor_is_not_xticklabelcolor      


r  c                  C   r  )Nr  ytick.colorr   ytick.labelcolor)r'   r   r^  rX   r  r	  r  r  r	  r$   r$   r%   &test_ytickcolor_is_not_yticklabelcolor!  r  r  c                  C   Z   dt jd< t  } | jj dksJ dt jd< dt jd< t  } | jj dks+J d S )Nr   r{  r  rw  inherit)r'   r   r^  rV   rH	  r  r  r$   r$   r%   test_xaxis_offsetText_color!     


r"  c                  C   r   )Nr   r  r  r  r!  )r'   r   r^  rX   rH	  r  r  r$   r$   r%   test_yaxis_offsetText_color!  r#  r$  r  c                 C   s   g | ]}|d ur|qS r  r$   )r   r  r$   r$   r%   r   %!  s    r   )rR  rw   r  c                 C   st   | t jd< | t jd< t \}}|j  tddg|j|jgD ]\}}|	 D ]}|j
 ||ks6J q(q d S )Nrz  zytick.labelsizer1   r2   )r   r   r'   r   r>  r  r  rV   rX   r  r  r  _get_tick_label_size)r  r-   r#   namer  r\  r$   r$   r%   test_relative_ticklabel_sizes%!  s   


r'  c                  C   s\   t  } | jdddd\}}|g dg d |dd | }t|d	d
gs,J d S )Nr8   r7   r  r  )PF  iJG  iDH  i>I  )r8   r   r8   r   r  r   r(  ipI  )r'   r   r   r   ro  r   rd   r  )r-   r  r  r   r$   r$   r%   test_multiplot_autoscale3!  s   r)  c                  C   sX   t  } | d}| jg d|d}| }| jdd |  | k s*J d S )NrY  )rt  rt  r   r   r  r   rN   )r'   r   r   rG  r  r^  r_
  r  )r-   rd  r  init_posr$   r$   r%   $test_sharing_does_not_link_positions<!  s   
 r,  c                 C   s  t g d}| d}|d jddgddg|dd tjdd	 |d jddgddg|dd W d    n1 s?w   Y  |d jddgddg|dd |d
 j	t 
d|dd |d jt 
dt 
d|dd |d}|d jddgddg|dd |d jddgddg|dd |d jddgddg|dd |d
 j	t 
d|dd |d jt 
dt 
d|dd d S )Nr   r   r   r7   r8   r   r  rS  rf  r   rw   r#  r(  rA  )rd   rF  r   r   r   r   r  r9   rr  ro  rl   r  )r!   r"   r   r  r$   r$   r%   test_2dcolor_plotE!  s   
"$"&
"""*r-  c                 C   s~   t ddt j d}t |}|jddddd}|jD ]}||| q| jddddd}|jD ]}|  ||| q0d S )Nr  r8   r  Tr  )rd   rl   r  r  r   r  r   r  )r!   r"   r1   r2   r  r#   r$   r$   r%   test_shared_axes_clearY!  s   


r.  c                  C   s   t jddddd\} }|jD ]
}|ddgd q|d g d |jD ]}| |d  ks3J q%|d g d |jD ]}| |d  ksNJ q@d S )	Nr8   r  r  r   zo-r  )r  r   r7   r   )r  r   r8   r  )r'   r   r  r   r	  r   r	  r   rl	  r$   r$   r%   test_shared_axes_retickh!  s   


r/  rD   )rN   rU   rA   c                 C   sB   t  }| }|jdd| d |jd |jj | ksJ d S )Ntestr7   rC   rA   )r   r   r)   rX   set_label_positionr/   r  )rD   r-   r#   r$   r$   r%   test_ylabel_ha_with_positionw!  s
   r2  c                  C   s   t  } ddgddg}}| ||}| |}|d j|d |d fks'J |d  dks1J |d  dks;J |d j|d |d fksJJ |d  dksTJ |d  dks^J d S )	Nr7   r8   r   r   rU   rM   r=   r'   rY  r  	bar_labelr  rW   rJ	  r#   r  heightsr  rT  r$   r$   r%    test_bar_label_location_vertical!     
r8  c                  C   s   t  } |   ddgddg}}| ||}| |}|d j|d |d fks+J |d  dks5J |d  dks?J |d j|d |d fksNJ |d  dksXJ |d  dksbJ d S )	Nr7   r8   r   r3  r   rU   r=   rM   )r'   rY  r  r  r5  r  rW   rJ	  r6  r$   r$   r%   *test_bar_label_location_vertical_yinverted!     
r:  c                  C   s   t  } ddgddg}}| ||}| |}|d j|d |d fks'J |d  dks1J |d  dks;J |d j|d |d fksJJ |d  dksTJ |d  dks^J d S 	Nr7   r8   r   r3  r   rN   rU   rA   r'   rY  r  r5  r  rW   rJ	  r#   r  r1  r  rT  r$   r$   r%   "test_bar_label_location_horizontal!  r9  r?  c                  C      t  } |   ddgddg}}| ||}| |}|d j|d |d fks+J |d  dks5J |d  dks?J |d j|d |d fksNJ |d  dksXJ |d  dksbJ d S r<  )r'   rY  r  r  r5  r  rW   rJ	  r>  r$   r$   r%   ,test_bar_label_location_horizontal_yinverted!  r;  rA  c                  C   r@  	Nr7   r8   r   r3  r   rA   rU   rN   )r'   rY  r  r  r5  r  rW   rJ	  r>  r$   r$   r%   ,test_bar_label_location_horizontal_xinverted!  r;  rC  c                  C   s   t  } |   |   ddgddg}}| ||}| |}|d j|d |d fks/J |d  dks9J |d  dksCJ |d j|d |d fksRJ |d  dks\J |d  dksfJ d S rB  )	r'   rY  r  r  r  r5  r  rW   rJ	  r>  r$   r$   r%   -test_bar_label_location_horizontal_xyinverted!  s   
rD  c                  C   s   t  } ddgddg}}| ||}| j|dd}|d jdks#J |d  dks-J |d  dks7J |d jdks@J |d  dksJJ |d  dksTJ d S )	Nr7   r8   r   r3  rU   
label_typer   r  r=  r>  r$   r$   r%   test_bar_label_location_center!  s   rG  z%test_centered_bar_label_nonlinear.svgc                  C   sP   t  \} }|g dg d}|d |dd  |j|dd |  d S )N)r;   r  r   )rr  i  iX  r&  r7   rU   rE  )r'   r   r  r  r  r5  r4  )r  r#   re  r$   r$   r%   !test_centered_bar_label_nonlinear!  s   
rH  c                  C   sl   t  \} }d}tg dg dD ]\}}|jd|||d}|j|dd ||7 }q|d d |   d S )	Nr   r^  )rw   rQ  r  rI  )r/   rN   rU   rE  rQ  )r'   r   r  r  r5  r  rs  )r-   r#   lastr/   r  re  r$   r$   r%   +test_centered_bar_label_label_beyond_limits!  s   
rJ  c                  C   s   t  } ddgddg}}| j||dd}| |}|d j|d |d d fks+J |d  dks5J |d  dks?J |d j|d |d d fksPJ |d  dksZJ |d  d	ksdJ d S )
Nr7   r8   r   r3  r)  r   rU   rM   r=   r4  r6  r$   r$   r%   !test_bar_label_location_errorbars!  s   
""rL  r   r  z{:.2f}c                 C   sV   t  }|ddgddg}|j|| d}|d  dksJ |d  dks)J d S )	Nr7   r8   r   r3  r  r   z3.00z-4.00r'   rY  r  r5  r  )r   r#   r  rT  r$   r$   r%   test_bar_label_fmt"  s
   rN  c                  C   s^   t  } | ddgddg}tjtdd | j|dd}W d    d S 1 s(w   Y  d S )	Nr7   r8   r   r3  zstr or callablerf  rw   r  )r'   rY  r  r   r   r   r5  )r#   r  r  r$   r$   r%   test_bar_label_fmt_error"  s
   "rO  c                  C   sZ   t  } | ddgddg}| j|ddgd}|d  dks!J |d  dks+J d S )	Nr7   r8   r   r3  r`  ra  r	  r   rM  )r#   r  rT  r$   r$   r%   test_bar_label_labels"  s
   rP  c                  C   sl   t  } | ddgtjdg}| |}dd |D ddgks!J |d jd	ks*J |d  d
ks4J d S )Nr8   r   r7   c                 S   rc  r$   r  r   r  r$   r$   r%   r   ""  r  z,test_bar_label_nan_ydata.<locals>.<listcomp>r   r  r   r	  rM   )r'   rY  r  rd   r  r5  r  rJ	  r#   rb  rT  r$   r$   r%   test_bar_label_nan_ydata"  s   
rS  c                  C   st   t  } |   | ddgtjdg}| |}dd |D ddgks%J |d jd	ks.J |d  d
ks8J d S )Nr8   r   r7   c                 S   rc  r$   r  rQ  r$   r$   r%   r   ,"  r  z5test_bar_label_nan_ydata_inverted.<locals>.<listcomp>r   r  r   r	  rM   )	r'   rY  r[  r  rd   r  r5  r  rJ	  rR  r$   r$   r%   !test_bar_label_nan_ydata_inverted'"  s   
rT  c                  C   s6  t  \} }|jg dtjddgg dd}||}dd |D g dks(J t| d	s2J t  \} }|jg dg d
dtjdgd}||}dd |D g dksZJ t| dsdJ t  \} }|jg dtjddgtjtjdgd}||}dd |D g dksJ t| d	sJ d S )Nr  r7   r8   rT  rK  c                 S   rc  r$   r  rQ  r$   r$   r%   r   5"  r  z&test_nan_barlabels.<locals>.<listcomp>)r   r  2)r  r  r>  r   rt  c                 S   rc  r$   r  rQ  r$   r$   r%   r   ;"  r  )r  r  rU  )r  r  c                 S   rc  r$   r  rQ  r$   r$   r%   r   A"  r  )r'   r   r  rd   r  r5  r  r   )r-   r#   rb  rT  r$   r$   r%   test_nan_barlabels1"  s    
 
&
rV  c               	   C   sr   t  \} }|tjdddddd dtdtj d	 d
  }tj	t
d|d  d|d f|jjd d S )N)r   r   r  r  r:  r   r  gffffff?r   rP  r8   gr_  r:  )r'   r   r?  r@  Wedgerd   r  r  r   assert_array_almost_equal_nulprF  dataLimrX
  )r-   r#   botr$   r$   r%   test_patch_boundsE"  s   "r[  c                	   C   sN   t jtdd tjdgdgddddd W d    d S 1 s w   Y  d S )	Nz!You passed a edgecolor/edgecolorsrf  r   r  r  r"  r  )r   r  r-  r.  )r   r  r$  r'   r9   r$   r$   r$   r%    test_warn_ignored_scatter_kwargsM"  s
   "r\  c                     sR  t  \}   fddtdD } tdtd} td} t	
ddd} ddd}t j|gks@J t j|gksJJ t j|ksSJ t j|gks]J  jrbJ t j|gkslJ  jd |d u swJ  jd	 |d	 u sJ tjtd
d  jt|d   W d    n1 sw   Y   jg d g |dddksJ g d j dddg|ksJ  jd g |dddR ksJ d j dddg|R ksJ |   jrJ |   jrJ |   jrJ  jrJ |   jrJ  jD ]}|  qt jdks'J d S )Nc              	      s&   g | ]}  t||d  d qS )r   r   )r   rd   rl   r  r  r$   r%   r   V"  s   & z(test_artist_sublists.<locals>.<listcomp>r  r   r   r   r  r   r  r   zout of rangerf  r7   r  r8   r   )r'   r   r   r9   rd   rl   r   rq  r?  r@  rA  r-
  r  r  imagesr  r4  tablesr  r   r   
IndexErrorr  r_  )r-   r  rI  rD  r  r-
  r	  r$   r  r%   test_artist_sublistsT"  sB   
& $


ra  c                  C   sh   t d} t d}t \}}|| |}t|dksJ t \}}|g g }t|dks2J d S )Nrw   )rw   r   r   r7   )rd   r   r'   r   r   r  )r1   r2   r  r#   r  r$   r$   r%   test_empty_line_plots"  s   

rb  z
fmt, match))r  z?'f' is not a valid format string \(unrecognized character 'f'\))zo+z9'o\+' is not a valid format string \(two marker symbols\))z:-z;':-' is not a valid format string \(two linestyle symbols\))rkz7'rk' is not a valid format string \(two color symbols\))z:o-rz=':o-r' is not a valid format string \(two linestyle symbols\))r  zE'C' is not a valid format string \('C' must be followed by a number\))z.CzF'.C' is not a valid format string \('C' must be followed by a number\)stringc                 C   sl   t  \}}|d ur|dd}tjtd| d d |jd| |d W d    d S 1 s/w   Y  d S )Nnotzneither a data key norz\Az\Zrf  rd  r7  )r'   r   replacer   r   r|  r   )r   rg  r  r-   r#   r$   r$   r%   test_plot_format_errors"  s   "rg  c                  C   s  t  \} }|g dd}|d  dksJ |d  dks"J t  \} }|g dd}|d  dks:J t  \} }|ddgddgdd}| j  |d  dks[J | d  dksgJ t  \} }|ddgddgdd}| j  |d  dksJ | d  dksJ t  \} }|g dd	}|d  d
ksJ |d  dksJ t  \} }|g dd}|d  dksJ |d  t	
dksJ |d  dksJ d S )Nr  z1.0r   )r   r   r   r   r  r  r7   r8   k3r   r  z.C12:r~   C12r  )r'   r   r   r  rK  r>  r  r	  r  r  r  r%  rC	  r$   r$   r%   test_plot_format"  s4   

rj  c                  C   s   t  \} }|jddddid | }| j  | d  dks%J | d  dks1J t  \} }|jdddddid | }| j  | d  dksWJ |	 d  dkscJ | d  dksoJ d S )Nr   r  rU  r8   r7  r   r;   )
r'   r   r   rG   r>  r  	get_textsr  r	  rb  )r-   r#   legr$   r$   r%   test_automatic_legend"  s   

rm  c                   C   s`  t jtdd tjg ddd W d    n1 sw   Y  t jtdd tjg dg dg dg ddd	gd
 W d    n1 sFw   Y  t jtdd tg ddg W d    n1 sfw   Y  t jtdd ttd W d    n1 sw   Y  t jtdd tjddddddid W d    d S 1 sw   Y  d S )Nz"plot\(\) got an unexpected keywordrf  r  r7   r  zplot\(\) with multiple groupsr  r  rU  r  zx and y must have same firstzx and y can be no greater than)r8   r8   r8   zUsing arbitrary long args withr   r  r;   rU  r8   r7  )r   r   r   r'   r   r|  rd   r   r$   r$   r$   r%   test_plot_errors"  s   *"rn  c               	   C   s   t   } t| jtdtdtddt| jddgddggt| jddgddggt| jddgddggt| j	ddgddggfD ]}d}||dj
}|j|jf|ksVJ qBd S )Nr   r  r   r7   r8   )r:  rR  )clim)r'   r   r   r   r9   r   r   r"  r  r  r  r  r  )r#   plot_methodro  r  r$   r$   r%   	test_clim"  s   rq  c                  C   s   ddgddgddgddgg} t jjt jjt jjt jjg}t | |}t \}}|t	| |
  | d dks>J d S )Nr   r   r7   r  )mpathr2  r6  CURVE3	CLOSEPOLYr'   r   r?  r@  r5  r]  r   )r  codesr   r-   r#   r$   r$   r%   test_bezier_autoscale"  s   rv  c                  C   s  t g ddgddgddgddgddgddgddgddgddgddgddgddgddgddgd	dgd	dgd
dgd
dgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgdd gd!d gd!d"gd#d"gd#d$gd%d$gd%d&gd'd&gd'd(gd)d(gd)d*gd+d*gd+d,gd-d,gd-d.gd/d.gd/d0gd1d0gd1d2gd3d2gd3d4gd5d4gd5d2gd6d2gd6d4gd.d4gd.d7gd8d7gd8d0gd9d0gd9d:gd;d:gd;d*gd<d*gd<d(gd=d(gd=d&gd>d&gd>d$gd?d$gd?d@gdAd@gdAd gdBd gdBdgdCdgdCdgdDdgdDdEgdFdEgdFdGgdHdGgdHdgdIdgdIdgdJdgdJdgdKdgdKdgdLdgdLdgdMdgdMdgdNdgdNdgdOdgdOdgdPdgdPdgdQdgdQdgdRdgdRdgdSdgdSdgdTdgdTdgdUdgdUdgdVdgdVdgdWdgdWdgdXdgdXdg} t | d d dYf }t | d d dZf }t | d d dYf }t | d d dZf }t| }t \}}|t	
| |  | dY |kscJ | dZ |ksnJ | dY |ksyJ | dZ |ksJ d S )[Ngr  g)\(gQgzGg=
ףp=gg(\gRQgGzg
ףp=
ggQg=
ףp=g(\
gGz	r  gffffffgQgףp=
rn  g(\g{Gz?gHzGg      rD  gQ g{Gz?gzGr   gRQrQ  g(\gQ?g333333g(\?gr#  g=
ףp=g)\(?g\(\rd  g=
ףp=gHzG?gQg(\?r  r   g(\տg{Gz?g
ףp=
ǿr  g{Gzgp=
ף?gףp=
?r  gQ?gQ?p=
ף?g(\?gq=
ףp?r   g(\?rJ  gQ?gGz?g=
ףp=?gffffff @r	  gGz@r_  g(\@g=
ףp=@gQ@g@gGz@g\(\	@gp=
ף
@gQ@g333333@g{Gz@g(\@gQ@g)\(@g@gq=
ףp@gGz@gQ@r   r7   )rd   rF  r+  r  rr  r2  r'   r   r?  r@  r5  r]  r   r   )r  minxminymaxxmaxyr   r-   r#   r$   r$   r%   test_small_autoscale #  s*  				



!
r|  c                  C   sb   t  \} }|td tdD ]}| |  | ks!J | |  | ks.J qd S )Nrw   )	r'   r   r   rd   rl   r   rb  r  r	  )r-   r#   r  r$   r$   r%   test_get_xticklabel4#  s   r}  c                  C   s   t jdtd} t g d}t dgd }t j| d< t \}}|j| ||d}|j| ||d}||fD ]2}|^}}	t 	|j
  sFJ t 	| sOJ |	D ]}
t 	|
j
 s]J t 	|
 sfJ qQq5d S )Nr   r  )r   r   r   rw  r   r  r*  )rd   rl   r  rF  r  r'   r   r  r  isfiniter  any	get_widthr  )barx
barheights	barstartsr-   r#   rb  hbarsbar_setnanfulrestr  r$   r$   r%   test_bar_leading_nan<#  s    
r  c                 C   sh   t jd |  }| }|tjgtjg |dgdg |dgdg  |dgdg d S )Nr   r7   )r   r   r   r   r  rd   r  r_  r  r$   r$   r%   test_bar_all_nanT#  s   r  zextent_units.pngc                  C   s  t dd\} }tdd}tdd}dd tdD }|d d	 |d j|d
dd||gtjd d}|d d |d j|d
||ddgtjd d}|d j	
td |d j|d
||||gtjd d}|d j	
td |d jdd |d j|d
tjd d}|||||g |d j	
td |d jdd tjtdd |jdd||gdd W d    d S 1 sw   Y  d S )Nr8   z
2020-01-01r1  z
2020-01-11c                    s"   g | ]  fd dt dD qS )c                    s   g | ]}|  qS r$   r$   r  r  r$   r%   r   f#  r  z0test_extent_units.<locals>.<listcomp>.<listcomp>rw   )r   )r   r$   r  r%   r   f#  r  z%test_extent_units.<locals>.<listcomp>rw   r  zDate extents on y axisr   r7   r  plasma)r   rw  r   r  z(Date extents on x axis (Day of Jan 2020)z%dr  zDay of Jan 2020)r5  rK  )r   r   z set_extent\(\) got an unexpectedrf  r  Fr  )r'   r   rd   r;  r   r4   r   r   r<  rV   r   r  DateFormatterr   
set_extentr   r   r   )r  r  
date_first	date_lastarrrD  r$   r$   r%   test_extent_unitsa#  s@   

"r  c                  C   s   t  \} }|g g g g }|dgg}||g D ]}|j|u s#J | | u s+J q|  ||g D ]}|jd u s>J | d u sFJ q5d S r;	  )r'   r   r   r   r^  
get_figurer  )r-   r#   r  rc  artr$   r$   r%   %test_cla_clears_children_axes_and_fig#  s   r  c                  C   s@   t  \} }|jg d|d}| }|  |jddd d S )N)r7   r   r   r7   rW  rB  r  rF  )r'   r   r
  r
  r_  r   )r-   r#   marginalmarginal_twinr$   r$   r%   test_child_axes_removal#  s
   r  c                  C   s\   dd } d}t jt|d d}tjjj|d i d| d W d    d S 1 s'w   Y  d S )Nc                   S   r   rQ  r$   r$   r$   r$   r%   rR  #  r   z5test_scatter_color_repr_error.<locals>.get_next_colorzZ'c' argument must be a color, a sequence of colors, or a sequence of numbers, not 'red\\n'rf  zred
r8   ry  )r   r   r|  r   r^  r   rY  )rR  msgr;   r$   r$   r%   test_scatter_color_repr_error#  s   
"r  c                  C   sh   t  \} }|d |jtdddd\}t }| j|dd W d    d S 1 s-w   Y  d S )Nr   Tr7   )r  r;  r8  )r  )r'   r   set_rasterization_zorderr   r   r  r  r  )r-   r#   r	  r  r$   r$   r%   &test_zorder_and_explicit_rasterization#  s   

"r  zpreset_clip_paths.pngc               	   C   s>  t  \} }tjjddgddgddgddggddddd}|| tjjd	d
d|d}|t	
 g || tjjd	ddd|d}|| tjjddgddgddggddddd|d}|| |jdddddid|d tjjddgddgddgddggddddd|d}| j|dd |dd |dd d S )Nr7   r   r   z#ddffddz#00ff00r8   r   )r-  r.  r/  re  rB  r  T)clip_onr  )r  r  r"  )r   r  r  r  z#beefc0r   z#faded0)r-  re  r.  r/  r  r  r  )r  r  )r   r  r   r  )r  r  r  r  r  r2   )r-  r.  r/  re  r  r  r  )r'   r   r   r4  Polygonr?  r  r   set_path_effectsr   withTickedStroker3  r  r  r  )r-   r#   polyr  poly2poly3r$   r$   r%   test_preset_clip_paths#  s8   





r  c                  C   sf   dt jd< dt jd< dt jd< t } | jj dksJ | jj dks'J | jj dks1J d S )Nr  zaxes.labelcolorrQ  zaxes.labelsizer	  zaxes.labelweight)	r   r   r'   r^  rV   r/   r  get_fontsizeget_fontweightr  r$   r$   r%   test_rc_axes_label_formatting#  s   


r  c           
   
   C   s  t dt j t j t jdddg}tt|}| dd}t|ddgD ]F\}}|j||d}|jddt j||d	|d
d
dd}t|	 t 
|  d ksTJ t|	 th |t 
|  d ksjJ q$|dd}	|	d jt j t j t j ddddt jgt dd dd |	d jt j dddt jt jgt g dd ddd |	d jt dd t j t j t j ddddt jgdd |	d jt g dd t j dddt jt jgddd d S )Nr   r7   r8   r  rO   rP   rU  r'  )rU  r'  Tr  )r  rQ   complementarycompressr  rR  r:  r  )ds)rb  rQ  r	  r  r   r   rb  z	steps-pre)r  r  )rd   rF  r  r   r  r   r  ecdfr	   	get_xdataisnanr  r   rl   )
r!   r"   r  r  r   r#   rQ   l0l1r   r$   r$   r%   	test_ecdf#  s6   "$.( 
r  c                   C   s   t t tdtjg W d    n1 sw   Y  t t ttjjddgddgd W d    d S 1 s=w   Y  d S )Nr7   r8   TFrm  )	r   r   r|  r'   r  rd   r  r	   rF  r$   r$   r$   r%   test_ecdf_invalid#  s    "r  c               	   C   s   t  \} }tddtj d}dt| }d}|j||dd | | f}|j	|dd	d
d |j
|dd||kdd| d | | f|ksNJ d S )Nr   r(  r  r   g333333?r  r#  r   r8   ru  )r   rC  re  r7   r   )r  r   re  r,  )r'   r   rd   rl   r  r  r   r   r   rn   r  r  )r-   r#   r1   r2   	thresholdoriginal_limsr$   r$   r%   test_fill_between_axes_limits$  s   
r  c                  C   sz   t  \} }|g dg d |jddd |jddd |jddd	 t d
 | D ]}| d dks:J q.d S )Nr  zX label in Impact fontImpact)fontnamezY label in xkcd scriptzxkcd scriptr"  	monospace)r   labelfontfamilyzTitle in sans-serifr   )	r'   r   r   r(   r)   rO  r0   rb  get_fontfamily)r-   r#   r-
  r$   r$   r%   test_tick_param_labelfont$  s   
r  c                  C   s   t  \} }|jddd}t|jd  dsJ t|jd  ds'J t|j d ds4J t|j d dsAJ t|jj	
 dsMJ d S )Nr=   r  r#  rM   r   rm  )r'   r   r
  r  r>  rK  r  rV   r	  r/   r  )r-   r#   saxr$   r$   r%   test_set_secondary_axis_color$  s   r  c                  C   sv   t jdddd\} }g }|d jd|j |d jd|j |d jddgdd	gd
 ||d |d gks9J d S )Nr8   Tr  r7   xlim_changedylim_changedr   r   r(  rF  )r'   r   	callbacksconnectr=  r   )r-   r  eventsr$   r$   r%   test_xylim_changed_shared'$  s   r  zaxhvlinespan_interpolation.pngc                  C   s   t  jdd} |   | jddd | jdddd	 | jd
ddddd	 | jdddd | jddddd | jddddddd d S )Nr  rU  r   r  r  r   r   r  )fcr  r   r  r7   )r;   re  r  re  )r  re  rt  ru  )r'   r   r   r4  r  rg  rn   ro  r  r$   r$   r%   test_axhvlinespan_interpolation0$  s   r  r?  c                 C   s   |  }|   }i dddddddddd	d
dddddddddddddddddddddddddddd }|jd#d!|i| |jd#d!|i| |  |d |d d"S )$z<Test that the given tick params are not reset by ax.clear().rj  ri  rk  rw   r  r   zxkcd:wine redrl  r   r  serifr;  r:  rn  r  rm  zxkcd:shocking pinkro  zxkcd:fluorescent greenrp  r   rq  r   rr  r  rM   Fr=   TrN   rA   )r	  r	  r	  r	  r  Nr$   )r   rO  r  r[  )r"   r!   r?  rq   rr   r1
  r$   r$   r%   test_axes_clear_behavior<$  s^   	
r  )r   r  r   finalz/https://github.com/python/cpython/issues/124538)reasonc               	   C   s   dd } t  }| }tjd}||| ||| | }|  dd |D }t	|dks4J |D ]}| | q6t	|dksEJ |D ]}z|
  W qG tyX   Y qGw d S )Nc                 S   sz   t | g}t }t|dkr;| }t|D ]}t||v r q|| us&J |t| || qt|dksd S d S r  )	r   r   r  popleftgcget_referentsidaddr=  )r"  to_visitexploredparentchildr$   r$   r%   assert_not_in_reference_cyclel$  s   
zFtest_axes_clear_reference_cycle.<locals>.assert_not_in_reference_cyclerr  c                 S   s   g | ]}t |ttfr|qS r$   )r   r   r   )r   r   r$   r$   r%   r   $  s    z3test_axes_clear_reference_cycle.<locals>.<listcomp>r   )r   r   rd   re   rx   r   r9   r0  r  r  r_  NotImplementedError)r  r-   r#   rj  ax_childrenbig_artists
big_artistr  r$   r$   r%   test_axes_clear_reference_cycleg$  s,   
r  c                  C   s   t jd t jd} tjdddd\}}tjtjdd |d	 j	| g d
d W d    n1 s4w   Y  dd |d	 
 D g d
ksJJ |d j	| g d
d dd |d 
 D g d
ksfJ d S )NrZ   )rw   r   r7   r8   T)r  r  rX  zhas been renamed 'tick_labels'rf  r   r_  r	  c                 S   rc  r$   r  rQ  r$   r$   r%   r   $  r  z,test_boxplot_tick_labels.<locals>.<listcomp>)tick_labelsc                 S   rc  r$   r  rQ  r$   r$   r%   r   $  r  )rd   re   rf   r'   r   r   r  r   r<  r-  rb  r4  r$   r$   r%   test_boxplot_tick_labels$  s   "&r  c                 C   sD   g d}|   }|j|dddid |  }|j|dddid d S )N)rQ  rw   r  z%1.0f%%usetexT)r  r  z%1.0f\%%r	  )r!   r"   r  r#   r  r$   r$   r%   test_latex_pie_percent$  s
   r  c                 C   s@  t jddd\}}tjd dd tddD }|d	 | t|d	  t	g d
ks1J t|d	 
 t	g dksBJ |d j|dd t|d  t	g dks\J t|d 
 t	g d
ksmJ t   | }tjtdd |j|dd W d    n1 sw   Y  |  }|j|dd d S )Nr7   r   r  rZ   c                 S      g | ]
}t jd |dqS r   r   rW  r   stdr$   r$   r%   r   $  rX  z/test_violinplot_orientation.<locals>.<listcomp>r  rw   r   )	r   r   r   r   r  r  r7  r  r|  g      >g      4r  r  r
  r  r  rO   rP   z
vert: boolrf  F)vert)r'   r   rd   re   rf   r   r]  r  r  rF  r	  rb  r   r  r  r!   r"   r-   r  all_datarr   rq   r$   r$   r%   test_violinplot_orientation$  s0   r  c              	   C   sn  t jddd\}}tjd dd tddD }|d	 | t|d	  t	g d
ks1J t|d	 
 t	g dksBJ |d j|dd t|d  t	g dks\J t|d 
 t	g d
ksmJ t   tjtjdd4 tddi | }|| W d    n1 sw   Y  |  }|j|dd W d    d S 1 sw   Y  d S )Nr7   r8   r  rZ   c                 S   r  r  rW  r  r$   r$   r%   r   $  rX  z,test_boxplot_orientation.<locals>.<listcomp>r  rw   r   r  r  rO   rP   z!was deprecated in Matplotlib 3.10rf  zboxplot.verticalF)r'   r   rd   re   rf   r   r-  r  r  rF  r	  rb  r   r  r   r<  r   r  r$   r$   r%   test_boxplot_orientation$  s8   "r  zuse_colorizer_keyword.pngr_  c                  C   s  t jd t jd} t jd}t jdddd}tdd\}}tj	dd	}tj
j|d
d}|d j||||d |d j| ||dd |d j||d |d j||d |d j||d |d j||d |d jdd||d |d j|j||d dd df |d |d j||d |d j||d |d j|j | | |d |d j|j | | |d |jt jt j|dddddd|d t| d S ) Nr   r   r  float32r  r]  r   r(  rQ  r  r  r   r  r;   	colorizerr  rL  r  r  r  )r  r   r   r  rK  r'  r  r   r7   r8   r   r   r  r7   r   r	  r8   r7   r  r   r[  r7   )rd   re   rf   rl   r   r'   r   r   r  r  r  	Colorizerr9   rx  r   r"  r  r  rp  r  r  
tricontourrG  tricontourffigimager_  r   )rand_xrand_yr;   r-   r^  r  clr$   r$   r%   test_use_colorizer_keyword$  s.   ($$&r  c            
   	   C   sr  t jd t jd} t jd}t jdddd}tdd\}}tj	dd	}tj
j|d
d}d}ddiddiddiddig}|D ](}	tjt|d |d j||f||d|	 W d    n1 shw   Y  qE|D ](}	tjt|d |d j||f||d|	 W d    n1 sw   Y  qp|D ](}	tjt|d |d j| |f|dd|	 W d    n1 sw   Y  q|D ]&}	tjt|d |d j|fd|i|	 W d    n1 sw   Y  q|D ]'}	tjt|d |d j|fd|i|	 W d    n	1 sw   Y  q|D ](}	tjt|d |d j|fd|i|	 W d    n	1 s;w   Y  q|D ](}	tjt|d |d j|fd|i|	 W d    n	1 sfw   Y  qD|D ]*}	tjt|d |d jdd |fd|i|	 W d    n	1 sw   Y  qo|D ]3}	tjt|d  |d! j|j||d dd df fd|i|	 W d    n	1 sw   Y  q|D ](}	tjt|d |d" j|fd|i|	 W d    n	1 sw   Y  q|D ](}	tjt|d |d# j|fd|i|	 W d    n	1 sw   Y  q|D ]1}	tjt|d |d j|j | | fd|i|	 W d    n	1 sSw   Y  q(|D ]1}	tjt|d |d$ j|j | | fd|i|	 W d    n	1 sw   Y  q\|D ]&}	tjt|d |j|fd|i|	 W d    n	1 sw   Y  qd S )%Nr   r   r  r  r  r]  r   r(  rQ  r  r  z5The `colorizer` keyword cannot be used simultaneouslyr  r  r  r&  r   r9  rf  r  r  r  rL  r  r  r  r  r  rK  r'  r  r  r  r	  r  r  )rd   re   rf   rl   r   r'   r   r   r  r  r  r  r   r   r|  r9   rx  r   r"  r  r  rp  r  r  r  rG  r  r  )
r  r  r;   r-   r^  r  r  	match_strkwrdskwrdr$   r$   r%   test_wrong_use_colorizer%  s      2$$r  c                  C   s   t  \} }|g dg d}|D ]}t| dsJ q|jg dg ddd}|D ]}t| ds9J q-|jg dg ddd	}|D ]}t| dsTJ qH|jg d
g dddd}|D ]}t| dspJ qdd S )Nr  r  r   )r  r  r  r  r#  )rb  rc  rO  r  r  )   r}   !   r   )r   r-  )r'   r   r  r  r>  r  )r-   r#   rb  r  r$   r$   r%   test_bar_color_precedenceW%  s   r  c                 C   s\   t ddgddgg}| |}|g d |j|j|j|jfdks%J |g d}d S )Nr  r   )rd  rd  r   r   )r  r  r   r   )r  r
  rG  rN  r  r  r  r  )r!   r"   r  rq   rr   r$   r$   r%   .test_axes_set_position_external_bbox_unchangedp%  s
   
r  r
  (  rW  r  r   r   r  r
  r   	functoolsr   r  r  r  	itertoolsr   platformsysr  r   dateutil.tzr*  numpyrd   r	   r
   r   rS   r   r   r   matplotlib.colorsr  r  matplotlib.datesrP  r  matplotlib.figurer   matplotlib.axesr   r  r   matplotlib.collectionsr   matplotlib.font_managerfont_managermfont_managermatplotlib.markersr  r  matplotlib.patchesr4  r@  matplotlib.pathr1  rr  matplotlib.projections.geor   matplotlib.projections.polarr   matplotlib.pyplotpyplotr'   matplotlib.textr-
  r  matplotlib.tickertickerr   matplotlib.transforms
transformsr  mpl_toolkits.axisartist
axisartistrq
  numpy.testingr   r   r   matplotlib.testing.decoratorsr   r   r   matplotlib.testing._markersr   r&   r.   r6   rL   rR   rT   rY   ru   r|   r   r   r   machiner   r   r   r  r  rj  rk  r"  r<  rK  rT  re  r  r  r  r  r   rl  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r&  r-  r2  r8  r@  rJ  rO  r^  rk  rp  rz  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rq  rl   r  r  r  r  r  r  r  r#  r1  r6  rD  rG  rH  ri  rj  ru  rz  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r	  r  r  r  r   r$  r/  r5  rE  rS  rW  rZ  rg  rh  rl  rq  rv  rz  r}  r  r  r  r  r  r  r  r  r  Patchr;  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r	  r  r  rn  rp  rF  rt  rZ  rx  rz  r{  COLOR_TEST_CASESr|  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r  r	  r  r  r  r  r  r  r  r  r"  r#  r%  r'  r)  r+  r6  r;  r@  rB  rC  rF  rP  rT  r^  ra  rc  rd  rg  rk  rm  ro  rp  rq  rr  rs  rv  rx  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r#  r)  r-  r/  r1  r3  r5  r6  r>  r@  rJ  rK  rU  r^  rb  rd  re  rh  rj  rm  rt  rw  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r#  r&  r-  r0  r1  r=  rB  rD  rI  rU  rV  rb  re  rh  ru  r  r  r  rn   rg  ro  r  r  r  r  r  r  r   r|  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r	  r	  r	  r	  r	  r	  r	  r!	  r"	  r&	  r'	  r(	  r+	  r,	  r0	  r2	  r4	  r7	  r9	  r:	  r<	  r?	  rA	  rB	  rD	  rE	  rG	  rM	  rN	  rR	  r^	  r_	  rc	  rd	  r
  	AxesImager   PcolorImagere   QuadMeshr   rx   rf	  rh	  ri	  rm	  rn	  ro	  rw	  ry	  r}	  r	  r	  r	  r	  r	  r	  fixturer	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r$
  r'
  r+
  r.
  r6
  r?
  r@
  rB
  rC
  rF
  rG
  rH
  rK
  rL
  rM
  rN
  rO
  rU
  rd
  rj
  rl
  rp
  rr
  r{
  r}
  r
  r
  r
  r
  r
  r
  	Transformr
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r   r  r  r
  r  r  r  r  r  r  r"  r$  font_scalingsr'  r)  r,  r-  r.  r/  r2  r8  r:  r?  rA  rC  rD  rG  rH  rJ  rL  r  rN  rO  rP  rS  rT  rV  r[  r\  ra  rb  rg  rj  rm  rn  rq  rv  r|  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  skipifversion_inforeleaselevelr  r  r  r  r  r  r  r  r  r$   r$   r$   r%   <module>   sd
   














'

*
43













		

	

!












	





&


















!

	
8



!$






	
	












+




  /    .	
 &,
	



		
=


	

	
	
	


	
	
	
	
	
	


				

	
	6!
.
+

	
	

A
	




	+*
	
&
&
 !"
	
 


,








	


$

"
".

!	

			

	




"
$	

.'
	




	
	

	
	

.	4
"
$

	
)('

#&<
