o
    :a                    @   s(+  d dl mZ d dlZd dlmZ d dlZd dlm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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-Z. d dl/m0Z1 d d
l2m3Z3m4Z4m5Z5 d dlm6Z6 d dl7m8Z8 dd Z9e dd Z:e dd Z;e 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  ZAeg d!d"d# ZBd$d% ZCd&d' ZDed(gd)d* ZEd+d, ZFejGHd-d.edgd/d0d1d2 ZIed3gd4d5 ZJd6d7 ZKd8d9 ZLedgdd:d; ZMed<gd=d>d?d@ ZNejOPdAdBdC ZQejOPdAdDdE ZRejOPdAdFdG ZSedgddHdI ZTedJgd=d>dKdL ZUedMgd=d>dNdO ZVdPdQ ZWdRdS ZXdTdU ZYedVgdWdXidYdZd[ ZZd\d] Z[ed^d^gd_d` Z\edagdbdcddde Z]edgddfdg Z^dhdi Z_edjgdgd=dbdkdldm Z`edngdodp Zaedqgdrdsdtdu Zbedvdvgd=d>dwdx Zcedygd=d>dzd{ Zdd|d} Zeed~gdbdcdd Zfdd Zgdd Zhedgdd ZiejOPdAejGHdg dedgddd Zjeddgd=dbddd Zkedgdbdcdd Zldd Zmedgd=d>dd ZnejGjHdeodddfepdddreoddfepdddrdeodfgg dddd ZqejGjHdeodddfepdddreoddfepdddrdeodfgg dddd Zredgd=d>dd Zsedgdbd=ddd Ztedgd=d>dd Zuedgdd Zvedgd=d>dd Zwdd Zxedgd=d>dd Zyedgddgd=ddd Zzedgddbddd Z{edgddbdddń Z|ddǄ Z}edgdddɄ Z~edgddd˄ Zdd̈́ ZejGHddd=gedgdddЄ Zedgddӄ Zedgd=dAdddք Zedgd=d>ddل Zddۄ Zedgd=d>ddބ Zedgd=dddd Zedgd=dddd Zedgd=d>dd Zedgd=d>dd ZedgdAd=ddd Zedgd=d>dd Zeddgdd Zdd Zdd Zedgdd Zedgdd  Zedgddd Zedgddd Zedgddd Zedgddd Zd	d
 Zdd Zedgdd Zdd Zdd Zdd Zdd Zdd ZejOPdAe dd Zdd Zedgd=d>dd  Zedgdd!d" Zd#d$ Zed%gd=d>d&d' Zed(gd=d>d)d* Zed+gd=d>d,d- Zed.gd/d0 Zd1d2 Zd3d4 ZejGjHd5ejjd6d7 d8d7 gg d9dd:d; ZejGHd<g d=fg gd=fg g gdfgd>d? ZejGHd@dAej'jjfdBejjjfdCej'jjfgdDdE ZedgddFdG ZedgddHdI ZedgddJdK ZedgddLdM ZdNdO ZdPdQ ZdRdS ZdTdU ZdVdW ZedXgd=d>dYdZ Zed[gd\d] Zd^d_ Zed`gd=dbddadb Zedcgdbdcddde Zedfdfgd=dbddgdh Zedigd=dbddjdk Zdldm ZG dndo doZdddpdqdrZedsdtZejGHdueÃ eĐdveg dwgdxfeÐdydzeĐdyeg d{gdxfeÐdydvdxeĐdyeg d{gdxfeÐdvd|eĐdveg dwgdxfeÐdvd}gd|eĐdvd}geg dwg d~gdxfgdd Z[[ejGHdeǃ dfeǐdvdzdfeǐdydpdyfeǐdyd}gdpdyd}gfeǐdyddyfeǐddpdfeǐddpdfeǐdyd}ddyfeǐdvdyd}ddyfeǐdyd|dyfeǐdyd}dd}fgdd ZȐdd Zɐdd Zʐdd Zedgdd Z̐dd Z͐dd Zeddgdd Zedgd=d>dd Zi dd7 i fddZedgdWdidAddd ZedgdWdidAddd ZedgdWdidAddd ZedgdWdidAddd Zedgd=dWdidAdddd ZedgdWdidAdddd Zedgd=dWdidAddd Zedgd=dWdidAddd Zedgd=dWdidAddd Zedgd=dWdidAddÐdĄ Zedgd=dWdidAddƐdǄ Zedgd=dWdidAddɐdʄ Zedgd=dWdidAdd̐d̈́ Zedgd=dWdidAddϐdЄ Zedgd=dWdidAddҐdӄ Zedgd=dWdidAddՐdք Zedgd=dWdidAddؐdل Zedgd=dWdidAddېd܄ Zedgd=dWdidAddސd߄ Zedgd=dWdidAddd Zedgd=dWdidAddd Zedgd=dWdidAddd Zedgd=dWdidAddd Zdd Zdd ZeddgddAddd Zedgd=dAddd Zedgd=dWdidAddd ZeddgdAdcdd Zdd  ZedgdWdid=d=dAddd Zedgd=dWdidAddd Zedgd=dWdidAdd	d
 Zdd Zdd Zdd Zdd Zedgd=dWdidAddd Zeddgdd Zedgdd Zedgdd Zedgd d! Zed"gd#d$ Zed%gd&d' Zed(gd)d* Zed+gd,d- Zed.gd/d0 Z ed1gd2d3 Zed4gd5d6 Zed7gd8d9 Zed:gd;d< Zed=gd>d? Zd@dA ZdBdC ZdDdE ZdFdG Z	edgddHdI Z
edgddJdK ZdLdM ZdNdO ZdPdQ Zeg dRdSdT ZdUdV Ze dWdX ZdYdZ Zed[gd\d] Zd^d_ Zd`da Zedgddbdc Zddde Ze dfdg ZejGHdhg dieg did=gdjdk Zedldlgdmdn Zedogdpdq Zedrgd=d>dsdt Zedugdvdw Zedxgdydz ZejGjHd{d=dgd|d}gded~gdbd=ddd Zdd Z dd Z!ejGjHd{d=dgd|d}gdedgdbd=ddd Z"edgdd Z#edgdd Z$edgdd Z%edgd=d>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.edgdd Z/dd Z0dd Z1dd Z2edgd=d>dd Z3edgd=d>dd Z4edgd=d>dd Z5dd Z6dd Z7edgd=d>dd Z8eddgd=d>dd Z9edgd=d>dd Z:ejGHdg dâdĐdń Z;edgd=d>dǐdȄ Z<dɐdʄ Z=ejGHdg gg d d=ggd d=gg ggejGHdg d͢dΐdτ Z>edgddАdф Z?edgd=d>dӐdԄ Z@edgd֐dׄ ZAedgd=d>dِdڄ ZBedېdgeC dkrWd ndrdsdސd߄ ZDe dd ZEe dd ZFe dd ZGe dd ZHdd ZIeg ddgddd ZJdd ZKeg ddgddd ZLdd ZMejGHdːd=ddejNdgejOg ddgedgddd ZPeddgd=d>dd ZQedgd=d>dd ZRdd  ZSeg dd=ddAddd ZTeg dd=ddAddd ZUeg d	d=ddAdd
d ZVdd ZWedgddd ZXdd ZYeg dd=dddd ZZeg dd=d>dd Z[dd Z\edgddd Z]edgd=d>dd Z^ed d gd=d>d!d" Z_ejGHd#g d$d%d& Z`d'd( Zad)d* Zbedgdd+d, Zcd-d. ZdejGHd/e+jee+jfd0ge+jge+jhd1ggd2d3 Zid4d5 Zjd6d7 Zkd8d9 Zld:d; ZmejGHd-d.d<d= Znd>d? Zod@dA ZpedBgdCdD Zqeg dEdgddFdG ZredHgdIdJ ZsedKgdLdM ZtedNgdOdP ZuedQgdRdS ZvedTgdUdV ZwedWgdXdY ZxdZd[ Zyd\d] Zzd^d_ Z{ed`gdadb Z|edgddcdd Z}dedf Z~dgdh Zdidj Zedkgdldm Zdndo ZedpgdWdqidYdrds Zdtdu Zdvdw Zdxdy Zdzd{ Zd|d} Zd~d Zdd Zdd Zdd ZedgdWdidYdd ZedgdWdidYdd Zdd Zdd Zdd Zdd ZejGHddejjfdejjfededfejjfg dg dfejjfejdejdfej jfgejGHdepddejdddgdd Zdd Zdd Zdd Zdd ZejGHde dd Zedgd=d>dd Zdd Zdd Zedgd=drddd Zedgd=dbddd ZejOPdAdd Zejddgddd Zejg dddd Zdd ZddÄ ZdĐdń ZdƐdǄ ZedgdAdcdɐdʄ ZedgdAdc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 Zedgddsdd Zedgdd Zedgddsdd Zedgd=d>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 ZejOPdAdd Zϐdd ZАdd Zѐdd ZҐdd ZӐdd ZejGHdddgejGHd ddgd!d" ZՐd#d$ Z֐d%d& Zאd'd( Zedgdd)d* Zedgdd+d, Zڐd-d. Zېd/d0 Zܐd1d2 ZejGHd3dd=gejGHd4dd=gd5d6 Zސd7d8 Zߐd9d: Zd;d< Zd=d> Zd?d@ ZedAgdbd=ddBdC ZedDgdbdcdEdF ZdGdH ZdIdJ ZdKdL ZdMdN ZdOdP ZdQdR ZdSdT ZdUdV ZdWdX ZdYdZ Zedgdd[d\ Zd]d^ Zd_d` Zdadb Zdcdd Zdedf Zdgdh Zdidj Zdkdl Zdmdn Zdodp Zdqdr Zdsdt Zdudv Zdwdx Zdydz Zd{d| Z ejGHd}d.ejGHd~ddd Zedgddd ZejOPdAdd Zdd Zdd ZejGHddd e"jD g d ejOPdAdd Zdd Zdd Z	edgddd Z
edgddd Zdd ZejGH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 Zdd ZejOPdAdd Zdd Zdd Zdd ZdS (      )
namedtupleNDecimal)product)SimpleNamespace)ma)cycler)image_comparisoncheck_figures_equalremove_ticks_and_titles)Figure)assert_allcloseassert_array_equalassert_array_almost_equal)
rc_context)MatplotlibDeprecationWarningc                  C   sD   t  \} }|d |d | dksJ | dks J d S )Nzx labelzy label)pltsubplots
set_xlabel
set_ylabelZ
get_xlabelZ
get_ylabelfigax r   </usr/lib/python3/dist-packages/matplotlib/tests/test_axes.pytest_get_labels(   s
   

r   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clabelY LabeltoplocX LabelrightZ Labelyhaxr+   r   r   legendr   r   colorbar	set_labelfig_testfig_refr   Zsccbarr   r   r   test_label_loc_vertical0   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}|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 )Nr   r   r   r   r"   bottomr$   r&   left
horizontalorientationr(   r   r)   r,   r.   r2   r   r   r   test_label_loc_horizontalC   s   r<   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 )Nr'   r#   )zxaxis.labellocationzyaxis.labellocationr   r   r   r   r"   r&   r9   r:   r(   r)   r,   )	
matplotlibr   r   r   r/   r   r   r0   r1   r2   r   r   r   test_label_loc_rcV   s$   



r>   c                  C   s  t  \} }|jddd |jddd |j  dksJ |jddd |j  dks1J |jddd |j  dksCJ |jddd |jddd |j  dks\J |jddd |j  dksnJ |jddd |j  dksJ d S )Nz
Test labelr8   r$   centerr'   r#   r7   )r   r   r   xaxis	get_labelget_horizontalalignmentr   yaxisr   r   r   r   test_label_shiftk   s   rD   png
extensionsc           	      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   r   )maxlagsfull)mode)r*   xminxmax)nprandomseednormalZcumsumr   ZacorrZ	correlatedotarangevlinesaxhline)	r3   r4   ZNxr-   rJ   ax_testax_refZnorm_auto_corrZlagsr   r   r   
test_acorr   s   rY   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 )NrH   i   r   rI   )    rZ   r   r   .lower)markeroriginZgray_rnearest)cmapinterpolation
markersize
   )aspectxlimylimboth)rO   rP   rQ   onesZshufflereshaper   spyimshowr@   tick_topplotZnonzerosetget_xlimget_ylimset_ticks_position)r3   r4   aaxs_testaxs_refr   r   r   r   test_spy   s$   



. rv   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 )Nra   r_   osolid)r]   	linestyle   )r   r   pytestraises	TypeErrorrk   rO   Zeye)r   r   Z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mpl20rZ   rh   )mplstyleuserO   rP   randadd_subplotZmatshowrl   r@   rm   rr   )r3   r4   rs   rX   r   r   r   test_matshow   s   

r   )Zformatter_ticker_001Zformatter_ticker_002Zformatter_ticker_003Zformatter_ticker_004Zformatter_ticker_005c                     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.0r-   unitsr   r   
<listcomp>   s    z)test_formatter_ticker.<locals>.<listcomp>rd   c                       g | ]}d | d  j  qS )      ?      ?Zkmr   r*   r   r   r   r          c                    r   )g      ?      ?r   r   r   r   r   r      r   zx-label 001bluer   )colorZxunitszx-label 003greenhourzx-label 004zx-label 005)matplotlib.testing.jpl_unitstesting	jpl_unitsregisterr=   rcParamsranger   figurer   r   rn   autoscale_view)ZxdataZydata1Z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   r-   posr   r   r   	_formfunc      z4test_funcformatter_auto_formatter.<locals>._formfunc)r   r   r   r@   isDefault_majfmtisDefault_minfmtrC   set_major_formattermtickerFuncFormatter
isinstanceget_major_formatterfunc)r   r   Z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   r@   r   r   rC   set_minor_formatterr   ZStrMethodFormatterr   Zget_minor_formatterfmt)Zformstrr   Ztarg_strformatterr   r   r   &test_strmethodformatter_auto_formatter  s    

r   Z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   r      T)numZendpoint       @皙?皙?333333?r   d   z%08.2lf)ZtricksZmindZjedi)   #   7   K   z%05.2lf)r    3prw   )rO   linspacesinpir   ZFixedLocatorr   r   rn   rC   set_major_locatorZset_minor_locatorr   ZFormatStrFormatterr   ZFixedFormatterr@   ZLinearLocatortwinytwinx)ZvalsZlocsZmajlZminlr   ax1r   r   r   "test_twin_axis_locators_formatters  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   drawr@   get_visiblepatchclarC   )r   r   ax2ax3r   r   r   test_twinx_cla6  s"   r   twinr-   r*   gRQ?)rG   tolc                 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_scaler   r   r   logr   r   r      nonezaxes.edgecoloraxes.linewidthZ
projecting)	transform	facecolor	edgecolor	linewidthZcapstyle)rO   rT   r   getattrrn   r=   pathPath
add_artistpatches	PathPatchZMOVETOZLINETO	transAxesr   r   r   )r3   r4   r   Z	twin_funcZ	set_scaler-   rW   ax_twinirX   r   r   r   r   test_twin_logscaleM  sD   
r   ztwin_autoscale.pngc                  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   r   r   r   r   r   r   r   r   r   r   F)Zautoscalex_onZautoscaley_onr   rd   r   lwr--r   r   r   )rO   arrayr   r   add_axesrn   r   margins)r-   r*   x2y2r   r   r   r   r   r   test_twinx_axis_scalesy  s   
r   c                  C   st   t  \} }| }|d | }| sJ | rJ | }|d | }| s2J | r8J d S )NF)r   r   r   Zset_autoscalex_onget_autoscalex_onr   Zset_autoscaley_onget_autoscaley_on)r   r   Zax_x_onZax_x_offZax_y_onZax_y_offr   r   r   #test_twin_inherit_autoscale_setting  s   

r   c                  C   sl  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 t d}t jd|d}|jd | sJ ||t| |  | rJ |  || ||t| |  | sJ t |  d S )	Nr   )r   r   r   r         shareyT)r   r   gcaZxaxis_invertedZyaxis_invertedrO   rP   rl   r   r   r   rn   cos	autoscalesubplotrC   Zset_invertedclose)r   r   Zimgr-   ax0r   r   r   r   test_inverted_cla  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   r   )r=   r   r   rn   minorticks_on)r3   r4   rW   rX   r   r   r    test_minorticks_on_rcParams_both  s   
r  Zautoscale_tiny_rangeTremove_textc                  C   sL   t dd\} }t|jD ]\}}dd|  }|ddgdd| g qd S )Nr   rd   ir   r   )r   r   	enumerateflatrn   )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	 d S )
Nr   r   r   rz   r   Tr-   F)Zenableaxistightr*   )g333333ÿg333333	@)r         @)r   r   rn   r   r   rp   rq   r   r   r   r   test_autoscale_tight  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   Zdtyper   r   Tsharex)r  rb   r   )rO   rT   floatr   r   loglogsemilogxr   r   r   rp   rq   )r-   r   r   r   Z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   r   r   rz   r\   r^   )      r   Fr  r   T)
r   r   rl   r   rp   rq   Zuse_sticky_edgesr   Z_xmarginZ_ymargin)r   r   rf   rg   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   rz   r   r   r   r  )rO   rT   rj   r   
pcolormesh)r3   r4   Zr  r   r   r   r   test_sticky_shared_axes  s   


r  Z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)rb   r   r   )Zautoscale_onrf   rg   rz   purpler   r   z	local maxrz   r   datarz   rz   zoffset points)xyxycoordsxytext
textcoords)	rO   rT   r   r   r   r   r   rn   annotate)tsr   r   liner   r   r   test_basic_annotate   s   

r1  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)rK   r8   r'   rz   r   r      r   r   rd   )widthlength_includes_headshapehead_starts_at_zeroZ
head_widthhead_length)r   r   r   r  zipr  set_xlimset_ylimrO   r   arrowr   r   )
r5  r6  r7  kwargsr   r	  r   r   kwargthetar   r   r   test_arrow_simple0  s&   
r@  c                  C   s$   t  \} }|jdddddd d S )Nr   )r8  )r   r   r<  _r   r   r   r   test_arrow_emptyK  s   rC  c                  C   s@   t  \} }|dddd | dksJ | dksJ d S )Nr   )皙?皙@)r   r   r<  rp   rq   rA  r   r   r   test_arrow_in_viewQ  s   rF  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foor   r   r   rz   )r+  )r+  
arrowprops)r   r   r-  Zarrow_patch)r   r   annr   r   r   test_annotate_default_arrowX  s
   rL  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  r   degred)Zyunitsr   r   )r  r  g     V@r  bZ   r   )r   )r   r   r   r   EpochdatetimerW  DurationrO   rT   mdatesdate2numr   r   rn   fillZautofmt_xdate)r   r.  valuedayrT  Zdtnr   r   r   r   ax4Zindr   r   r   test_fill_unitsa  s4   


""rd  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 definedmatchr   rw   r-   r]   z linestyle .* redundantly defined---ry   zcolor .* redundantly definedrr   r   r   )r   r   )r{   warnsUserWarningr   rn   errorbarr   r   r   r    test_plot_format_kwarg_redundant  s   rp  Z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 )Nrw   lines.markerT	axes.gridr   r   r   rs   rY  rs   r'  rY  )r=   r   r   r   rn   )r   r   r   r'  r   r   r   test_single_point  s   

ru  zsingle_date.pngr   )r   c                  C   sj   dt jd< ttd} dg}dg}t dd\}}|d j||  |d	d
d |d j||d	d
d d S )N)   zaxes.formatter.limits
0000-12-31g    X&Ag(\bPr   r   r   rw   rk  rl  )	r   r   r^  r_  rO   
datetime64r   	plot_datern   )rT  Z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 )Nrd   r   rb   r   r   rb   r   r   r   gQQ?)r   	   rf   rg   )rO   rT   rj   r   rn   ro   Zravel)r3   r4   rowcolr	  r   r   r   test_shaped_data  s   

r  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r   r   r   r   ri   Ztwosr  r   r   rt  r   rk  )rO   r   r  r   r   r   rn   )Zptsr	  r   r   r   test_structured_data  s   r  Zaitoff_proj)rG   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
       r   )   @ZaitoffZ
projection)figsize
subplot_kwrw   r   )rc   N)
rO   r   r   meshgridr   r   dictgridrn   r  )r-   r*   XYr   r   r   r   r   test_aitoff_proj  s   
r  Z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   rP  rQ  r   r  rS     r   r         ?)r   alphar   )r   r   r   r   r[  r\  r]  rb  convertr   r   axvspanr:  r   Zt0ZtfrT  r   r   r   r   test_axvspan_epoch      r  Zaxhspan_epoch{Gz?)r   c                  C   r  r  )r   r   r   r   r[  r\  r]  rb  r  r   r   axhspanr;  r  r   r   r   test_axhspan_epoch  r  r  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  )r     r   r   333333?ffffff?)extentr   r-   r*   )r  r'  )r   r   rO   rT   rj   hexbin)r   r   r'  r-   r*   r   r   r   test_hexbin_extent  s   
r  zhexbin_empty.pngc                  C   s   t  } | g g  d S r   )r   r   r  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   r   r   r  rb   )r  Zpicker  ,  r   r   )r   r   rO   rT   rj   r  r   contains)r   r   r'  r-   r*   ZhbZ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 
| d S )NFpcolormesh.snaprH   順 r         @r  r   r   r   T)yscalebinsZ	marginalsZreduce_C_function)r   r   rO   rP   rQ   standard_normalZpowerr   r  sumr0   )nr-   r*   r   r   hr   r   r   test_hexbin_log  s   
r  c                  C   sH   t dd\} }t \}}|j| |dddd}| dks"J d S )Nr  r  r   r   r   )r  vminvmax)rO   rT   rj   r   r   r  Zget_clim)r-   r*   r   r   r  r   r   r   test_hexbin_log_clim+  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#  r   r   )r   r   r#  r   )r   r  r"  )r  r   )r   r#  r   rd   r   )rd   r   )
r   r   invert_xaxisrn   rp   rq   r   invert_yaxis
set_yscaler;  r   r   r   r   test_inverted_limits2  s   
r  Z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Zdivider   )
rO   rT   eerrstater   nanlenr   r   rn   r-   r*   r   r   r   r   r   test_nonfinite_limitsL  s   r  plot_fun)r   rn   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-01rV  r  C0rl  )rO   rT   r   r   r   r  )r  r3   r4   r-   rW   rX   r   r   r   r   test_limits_empty_dataW  s   r  rl   )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   r   rk  rt  )r=   r   rO   indicessqrtr   r   rl   )Nr-   r*   rk  r   r   r'  r   r   r   test_imshowh  s   

r  Zimshow_clipc            	      C   s   dt jd< d} t| | f\}}|| d 8 }|| d 8 }t|d |d  ||  }t \}}||| d g}|jd }|	 d }|
 }t||}|j||d d S )Nr_   r  r   r   r   r   )	clip_path)r=   r   rO   r  r  r   r   contourcollections	get_pathsget_transformmtransformsZTransformedPathrl   )	r  r-   r*   rk  r   r   r    r  Zclip_transformr   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.r   r   rz   r   FPassing parameters norm and vmin/vmax simultaneously is not supported.re  rd   r   r   )normr  r  N)r   axesr{   r|   
ValueErrorrl   mcolors	Normalize)rs   r   r   r   r   test_imshow_norm_vminvmax  s   "r  Z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 )Nr   r   (   )
linewidthsr   rz   )
r   r   rO   r   r   r  ZPolyCollectionadd_collectionZ
set_xboundZ
set_ybound)r   r   vertsr    r   r   r   test_polycollection_joinstyle  s   "
r  z	x, y1, y2r  rz   r  r   )Z
2d_x_inputZ2d_y1_inputZ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  )r-   r
  r   r   r   r   r   r   test_fill_between_input     "r  z	y, x1, x2)Z
2d_y_inputZ2d_x1_inputZ2d_x2_inputc                 C   r  r   )r   r   r{   r|   r  fill_betweenx)r*   x1r   r   r   r   r   r   test_fill_betweenx_input  r  r  Zfill_between_interpolatec               	   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  r   r  333333?r   r   Tr  blackrl  white/)wherer   hatchinterpolaterX  )r  r   r  r   r   r   )rO   rT   r   r   r   r   rn   r  r   masked_greaterZmasked)r-   r
  r   r   r   r   r   r   r   test_fill_between_interpolate  s(   
r  Z#fill_between_interpolate_decreasingr   r  c               	   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"@rw  rE  )g@gffffff@gffffff@)r  r  r  ztab:redkr   T皙?r  r   r  r  rX  r   r      X  )rO   r   r   r   rn   r  r:  r;  )r   r.  Zprofr   r   r   r   r   (test_fill_between_interpolate_decreasing  s   r  Z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
 ||k}||k}W d    n1 s?w   Y  t \}}|j| |dd |j| |dd |j| |||dddd |j| |||dddd d S )Nrd   r              rZ   r   r  Zinvalidr  r    rY  r   Tr   r  rX  )	rO   rT   asarrayr  r  r   r   rn   r  )r-   r
  r   Zgreater2Zgreater1r   r   r   r   r   !test_fill_between_interpolate_nan  s    
 "

r	  z
symlog.pdfc                  C   sX   t g d} t g d}t \}}|| | |d |d |dd d S )N)r   r   r   r      r  r3  r  )i@B i  r  r   r   r   r   r   symloglinearrb   i )rO   r   r   r   rn   r  
set_xscaler;  r  r   r   r   test_symlog  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   r   )      4@r   r   r   r!  r  )	linthreshTrb   皙r   )	rO   rT   r   r   r9  rn   r  r  r;  )r-   r   r	  r   r  r   r   r   test_symlog2"  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  g      r  g      пr   r  r         ?r         ?r   rb   )rO   r  hypotr   pcolorlist)r-   r*   r  r  r  r   r   r   test_pcolorargs_5205/  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|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  r3  r  r   r   皙?r   r   r   rz   rb   r  )r   
edgecolorsrY  wgouraudshading)r   r   rO   r   r  r   r   r  minptpr   Zmasked_whereabsmaxr   r  )r  r-   r*   r  r  QxZQzr  ZZmr   r   r   r   r   r   r   test_pcolormesh=  s   
 $(r)  Zpcolormesh_alphaZpdf)rG   r  c               
   C   s  dt jd< d} ttdd| tdd| d \}}|}|t| }t||d }||  |  }t 	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 qn|
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  r3  r  r   r   r   viridis   r   rz   )r   r  )r  r   r   r   r   )r   zorderrb   r  r  r   )r`   r  r#  r,  r!  )r`   r#  r,  )r   r   rO   r  r   r   r  r$  r%  get_cmaprT   r  ZListedColormapr   	add_patchmpatches	Rectangler  )r  r  r  r(  ZQyr  Zvircolorsr`   r   r   r   r   rc  r   r   r   r   test_pcolormesh_alphaU  s2   
$

&r2  zpcolormesh_datetime_axis.pngFc                     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?Zhspacer#   r7     r   c                       g | ]
} t j|d  qS )Zdaysr\  	timedeltar   dbaser   r   r         z1test_pcolormesh_datetime_axis.<locals>.<listcomp>r  r     rb      r   r        r'   r   )r   r   r   subplots_adjustr\  rO   r   r   rT   r  r   r  repeatnewaxisget_axesget_xticklabelsset_haset_rotationr   r-   r*   Zz1Zz2zr   r!   r   r=  r   test_pcolormesh_datetime_axisx  s0   


.

>

rN  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  r3  r4  r5  r6  r   c                    r7  r8  r9  r;  r=  r   r   r     r?  z-test_pcolor_datetime_axis.<locals>.<listcomp>r  r  r@  rb   rA  r   rB  rC  rD  r'   r   )r   r   rE  r\  rO   r   r   rT   r  r   r  rF  rG  rH  rI  rJ  rK  rL  r   r=  r   test_pcolor_datetime_axis  s.   

.

>

rO  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    d S 1 sNw   Y  d S )Nr3  r  r   r   r   rb   r!  r"  r   r  r  )mask)ig  r   r   r  rd   z.are not monotonically increasing or decreasingre  auto)rO   r   r  r  r   r   r{   r|   r}   r  TNaNr  r  r   r   zerosr6  rm  rn  )r  r-   r*   r  r  r  rB  r   r   r   r   test_pcolorargs  sJ   $$
  $rU  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   rd   rz   rH   r   r  r  r"  rb   r_   r   rO   rT   rP   rQ   randnr  diffr3   r4   r   r-   r*   r  r   r   r   r   r   test_pcolornearest  s   rZ  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\  Zfromtimestampr   r   r   r   r     r?  z+test_pcolornearestunits.<locals>.<listcomp>rd   r   rz   rH   r   r  r  r"  c                 S   s    g | ]}t j |d  d qS )r   r[  r\  r   r   r   r   r     s     rb   r_   )	r   r   rO   rT   rP   rQ   rW  r  rX  rY  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   r  rz   rH   zDimensions of Cre  r  r"  )r   r   rO   rT   rP   rQ   rW  r{   r|   r}   r  )r   r   r-   r*   r  r   r   r   test_pcolorflaterror  s   "r^  snapc           	      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   rd   r   rH   rz   r  )r_  rb   r   rV  )	r3   r4   r_  r   r-   r*   r  r   r   r   r   r   test_pcolorauto  s   r`  Z	canonicalc                  C   s   t  \} }|g d d S )Nr   r   rz   r   r   rn   r   r   r   r   test_canonical  s   rc  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 )Nr   r   )r   r   rz   h  r  -   r   )r  )theta1theta2r   rd   )r   r   r  r  r.  r/  ZEllipseArcrn   rO   r   deg2radr   r:  r;  )
r   r  Zcentrer   r   r	  r   r   rg  rf  r   r   r   test_arc_angles  s4   
rj  Zarc_ellipsec               	   C   sN  d\} }d\}}d}t t d}|d t | }|d t | }t |}t t |t | gt |t |gg}	t |	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   rd  r   r   rQ  )re   r   yellowr   )r  r   r   r   r,  r   F)angler   r`  r,  r   equalr   )r  r   r   r,  )rO   ri  rT   r   r   r   rS   r   r   r   r`  r/  rh  r.  )ZxcenterZycenterr4  heightrm  r?  r-   r*   ZrthetaRr   r   e1e2r   r   r   test_arc_ellipse5  s:   

rs  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 )Nrw   ra  rz   r   r   rg  )r   rz   r   )r   r   rz   )r   r   mmarkersMarkerStylern   r   ro  )fixr   mr   r   r   test_marker_as_markerstyleZ  s
   
ry  	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   rd   r   r   rw   r  r!   r<  mark allrz  r!   r/  mark every 10+r   r  mark every 5 starting at 10rO   r   r   r  r   r   rn   r/   r  r   r   r   test_markeveryb  s   r  Zmarkevery_lineg{Gzt?)r  r   c                  C   r{  )Nr   rd   r   r   z-or  r|  z-dr}  r~  z-sr  z-+r  r  r  r  r   r   r   test_markevery_linep  s   r  Z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 )Nr  r   r  r+  r  r   r   rb   r   r  rz   r   r   r   r  r   g?r   r   )\(?rd   r   r   markevery=%srw   rh  r   lsmsrz  )slicer=   gridspecGridSpecr  rO   r   r   r  r   r   titlestrrn   
casescolsgsdeltar-   r*   r   caser  r  r   r   r   test_markevery_linear_scales  s(   
r  Z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 )Nr  r  r  r   rb   r   r  rz   r   r   r   r  r  r   r  rd   r   r   r  rw   rh  r   r  )r
  g@)r  g333333?)r  r=   r  r  r  rO   r   r   r  r   r   r  r  rn   rf   rg   r  r   r   r   #test_markevery_linear_scales_zoomed  s,   

r  Z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 )Nr  r  r  r   rb   r   r  rz   r   r   r   r  r  r   r  rd   r   r   r  r   rw   rh  r   r  )r  r=   r  r  r  rO   r   r   r  r   r   r  r  xscaler  rn   r  r   r   r   test_markevery_log_scales  s,   


r  Z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 )Nr  r  r  r   rb   r   r  rz   r   r   r   r  r  r   r  r   T)polarr  rw   rh  r   r  )r  r=   r  r  r  rO   r   r   r  r   r   r  r  rn   )	r  r  r  rk  r?  r   r  r  r  r   r   r   test_markevery_polar  s&   
r  Z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   r   rd   zy.      >@rk  )r  mewmecr   r   r   rY  )rO   r   r   r   rn   r   r-   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   r   edge0)align
tick_labelrs  rs   rY  )r  r  r'  )r   r   barr   )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   r   r   )r   r   r  rO   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 )
Nr         @r   r   r   rs   rY  r?   r4  r  r  )r   r   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 )Nr?   zytick.alignmentr   r  r   r   r   rs   rY  r  )r=   r   r   r   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@r  r  rE  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  r?   r  r   r  r3   r4   r   x0y0r-   r*   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   r   r   r?   ro  r  r   barhr  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  r  g333333?r   r   c                 S   r  r   r   r   r   r   r   r   r   ;  r  z*test_bar_decimal_width.<locals>.<listcomp>r?   r4  r  r  )r3   r4   r-   r*   Zw0r   r   r   r   r   test_bar_decimal_width6     r  c                 C   r  )Nr  r  r  c                 S   r  r   r   r  r   r   r   r   I  r  z,test_barh_decimal_height.<locals>.<listcomp>r?   r  r  )r3   r4   r-   r*   Zh0r  r   r   r   r   test_barh_decimal_heightD  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 )
Nr   r   r   r   r   rk  r  r   r   r   r   r   r   r   r   r   r   r   r   r  get_facecolorget_edgecolorr   rectsrectr   r   r   test_bar_color_none_alphaR     r  c                  C   r  )
Nr   r   r   r   rk  r   r  r  r  r  r  r   r   r   test_bar_edgecolor_none_alphaZ  r  r  zbarh_tick_label.pngc                  C   r  )
Nr   r  r   r   r   rs   rY  r?   )ro  r  r  )r   r   r  r  r   r   r   test_barh_tick_labelb  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.  r   r   rz   Zhoursr4  r   )xerrr4  )ro  )ro  yerrr   c                 S   s   g | ]}t j|d qS )r  r9  r   r.  r   r   r   r   ~  s    z&test_bar_timedelta.<locals>.<listcomp>r   )rd   r  N)	r   r   r  r\  r:  r  rO   r   broken_barhr   r   r   r   test_bar_timedeltaj  s<   







r  c                 C   sB   t jdd}| jdd| jdddj}t  tj||d d S )Nr   r   z1/1/2000r   )years)periodsZfreq	positions)	rO   rP   r   
date_rangeZ
DateOffsetyearr   r   boxplot)pdr'  r  r   r   r   test_boxplot_dates_pandas  s   r  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     r   rw  rb   x   )Zsecondsr   )Zpandas.plottingr  r  r   r   r   r\  r  appendr:  rO   rT   r  r6  r  canvasr   )r  r  r  r   r   timesZy_valsZ	time_axisZy_axisr6  Zz_dataimr   r   r   test_pcolor_regression  s"   

r  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  )r   r   r   ra  )r  monthrb  ra  )r  r  rb  datera  rd   r?   r  oranger   r   )		DataFrameZto_datetimegroupbyr  indexr   r   r  rn   )r  dfZmonthlydatesZforecastbaseliner   r   r   r   r   test_bar_pandas  s   r  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  r  )r-   r4  ra  r  r   r  )r  r   r   r  r-   r4  r  r  r   r   r   r   r   test_bar_pandas_indexed  s
   r  c                 C   sh   |   }|  }ddg}ddg}ddg}tdD ]}|j|| || d|| d q|j|||d d S )	Nr   r   rz   r-   rw   r  )r   r  )r  )r   r   r  )r3   r4   rW   rX   r-   r*   hatchesr   r   r   r   test_bar_hatches  s    r   c                 C   sF   | j ddgdd}t|| t|j| t| t|j d S )Nr   r   float64r  )Seriesr   rn   r  )r  r-   r   r   r   test_pandas_minimal_plot  s
   
r  Zhist_logc                  C   sH   t dddd } t d|  d|  g}t \}}|j|ddd d S )Nr   r   r  rz   FT)r`  r   )rO   r   concatenater   r   hist)Zdata0r'  r   r   r   r   r   test_hist_log  s   r  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 )
Nr   rz   )r  step
stepfilledr   r   r   histtypeT)r   r
  )r   r  r  r  )r3   r4   rt   ru   r   r
  r   r   r   r   test_hist_log_2  s   r  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 )Nr   r   r   
barstackedr	  r   r   r   r   )r   r   r  r  r  r   rq   r   r	  r   r   r   test_hist_log_barstacked  s   
 r  zhist_bar_empty.pngc                  C      t  } | jg dd d S )Nr  r	  r   r   r  r  r   r   r   test_hist_bar_empty      r  zhist_step_empty.pngc                  C   r  )Nr  r	  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||f  |j
dd q7|d j}	tdd |	D siJ d S )Nr   r  rz   rd   r`  TFr   r  r   r  )nrowsncolsr
  stackedz%s/%sr  r7   r   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>)rO   rP   rQ   rW  r   r   r9  r  r  	set_titler;  r   all)
r-   Zn_binsr=  typesr   r	  ZkgZ_typer   r   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 )NrH   r  Tdensity)rO   rP   rQ   r  r   r   r  r'  r   r   r   r   r   test_hist_density!  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#  rb   r  r   r   r   T)r  r!  )rO   rP   ZRandomStaterW  r   r  Z	histogramr   )rngr.  r  Zmpl_heightsrB  Z
np_heightsr   r   r   test_hist_unequal_bins_density)  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  r   r   T)r  F)r\  r   r   r  r"  r   r   r   test_hist_datetime_datasets4  s   r'  bins_preprocessc                 C      | S r   r   r  r   r   r   <lambda>>      r+  c                 C   s   t | dS )Nry  )rO   r  astyper*  r   r   r   r+  ?      )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  r   r   r  r   rz   r  T)r  r  F)
r\  r   r   r  rO   r   r   r   r  r_  )r(  r'  Z
date_edgesr   r   rB  r  r   r   r    test_hist_datetime_datasets_bins<  s   r/  zdata, expected_number_of_histsr   c                 C   sJ   t | \}}}t|}|jdkrd|ksJ d S |jd |ks#J d S )Nr   r   )r   r  rO   r  ndimr6  )r'  Zexpected_number_of_histsZhistsrB  r   r   r   test_hist_with_empty_inputS  s
   

r1  zhisttype, zorderr  r  r  c                 C   sH   t   }|jddg| d |jsJ |jD ]
}| |ks!J qd S )Nr   r   r	  )r   r   r   r  r   
get_zorder)r
  r,  r   r   r   r   r   test_hist_zordera  s   

r3  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
     rZ   %   0   rZ   r  r   )	r   r   r  r  r   g      @      @g       @      "@rz   r   r  r   r9   )r  r;   r:   r   r   miterbuttsolid_joinstylesolid_capstylerb   	drawstyle
steps-post)Zmatplotlib.lineslinesrO   r   r   flattenstairsZsemilogyr  rn   r  Zadd_lineLine2Dr;  r:  )r3   r4   Zmlinesr*   r-   	test_axesr   ref_axesr   r   r   test_stairsm  s@   
(((22(22(22(22rG  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)r   r   rz   r   r   )r   r   r   rz   r   r   r2  r   r   T)r`  r   r9   )r;   r`  )r  r`  rz   )r  r;   r`  rb   post)r  r   )r   rB  rC  r  rO   r  r;  r  r:  ri   r  )r3   r4   r  r  bsrE  rF  r   r   r   test_stairs_fill  s,   ""rJ  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   ra  rt  r   r   edges)r   r   r   r   r   r2  r9  )	r   rC  r;  set_datarO   rT   allcloseget_datar  )r3   r4   rg   test_axr  ref_axr   r   r   test_stairs_update  s    
 rR  c                 C   s\   |   }|jg dd d |  }ddd}|jtdg dfdd	i| |d
d  d S )N)r   r
  rw  r9  r:  r;  r<  r   )r   r
  rw  rw  r?  r@  r   )r   rC  rn   r   r;  )r3   r4   rP  rQ  r   r   r   r   test_stairs_baseline_0  s   
 rS  c                  C   s>   t   } | g dg |  dksJ |  dksJ d S )N*   )'   re  r  )r   r   r   rC  rp   rq   r  r   r   r   test_stairs_empty  s   rV  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"re  r   r   r   )r{   r|   r  r   rC  rO   r  r   r   r   r   test_stairs_invalid_nan  s   "rX  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 mismatchre  r   r   r   )r{   r|   r  r   rC  r   r   r   r   test_stairs_invalid_mismatch  s   "rZ  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 )Nr   r   r   rW  re  rK  )r   rC  r{   r|   r  rM  rO   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 )Nr   r   r   rY  re  r   rK  )r   rC  r{   r|   r  rM  rO   rT   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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r   r   rz   r   r   r  r   rz   r   TA)r   r`  r!   rX  r9   B)r   r`  r;   r!   r  ri  r   C)r   r  r   r!   r   D)r  r   r  r;   r!   rb      E)r   r  r   r  r!   r4  r  r$  F)r  r   r  r   r!   r   rR     r   G)r  r   r  r   r!   r`  r  r  r3  //H)r   r  r   r  r  r!   r   r$   )rO   r   r-  r  copyr  r   r   rC  r   r  r/   )r-   r*   Zynr   r   r   r   r   test_stairs_options  s2   $
&.
4rk  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   rC  rO   rT   ry  Zxticks)fr   r   r   r   test_stairs_datetime  s   rp  c                  C   sH   t ddd} t ddd}t | t |d d t jf  }| ||fS )Nr#  r      r  )rO   r   r   r   rG  )r-   r*   rM  r   r   r   contour_dat  s   "
rr  Zcontour_hatchingc               
   C   s@   t  \} }}t \}}|j| ||dg dtdddd d S )Nrw  )r  \rh  rh  Zgrayrh   r   )r  r`   extendr  )rr  r   r   contourfr-  )r-   r*   rM  r   r   r   r   r   test_contour_hatching  s   
rv  Zcontour_colorbarc            	   
   C   s   t  \} }}t \}}|j| ||tdddt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   ZRdBug333333r  rh   )levelsr`   r  r  rt  gg^I+r*   rx   r   )rw  r1  
linestylesr  rE  r    )rw  r1  r  r  F)Zerase)
rr  r   r   ru  rO   rT   r-  r  r0   Z	add_lines)	r-   r*   rM  r   r   csZcs1Zcs2r5   r   r   r   test_contour_colorbar&  s(   
rz  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   r   r   rd   Tr  
rasterizedr   r-   r*   )r  r'  r}  )r   r   rO   rP   rQ   rW  r   r{  )r-   r*   r   r   r'  r   r   r   test_hist2d<  s   

r~  Z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   r   rd   Tr|  )	r   r   rO   rP   rQ   r   rW  r   r{  r  r   r   r   test_hist2d_transposeN  s   
r  c                  C   s>   t jd\} }t  }|tfD ]
}|j| |dd qd S )Nr  Tr   )rO   rP   r   r   r   r{  )r-   r*   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	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&d' Zd(d)gd*fd+d,d-g d.d/fd0d1gd*fd1gd2 d*fd3gd4 d/fd3gd2 d/fd3gd5 d6fg d7d/fg d8d/fg d9d*fg d:gd/fg d:gd4 d6fg d:gd2 d/fg d:gd5 d6fg d;gd/fg d;gd4 d6fg d;gd2 d/fg d;gd5 d6fg d;gd4 g d:g d/fg d;d<d=gd6fg d;d<d=d>gd/fg d;d<d=d>g d?gd6fg d;d<d1gd*fg d;d<d=d1gd*fg d;d<d=d>d1gd*fgZejd@edAdB ZejdedgddCdD ZdEdF Zd/S )GTestScatterr   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rz   r   r   r
  r   r   r   rz   rk  r*   rY  limer  r         )0.50.60.7z0.8)r-   r*   r    r/  c2r-   r   r*   r    r/  r    r/  r  )r    r/  r'  )rO   r   r   r   r   )selfr'  r   r   r   r   r   test_scatter_plotd  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 )Nrz   r  r  r  )r   r   r   r*   rY  r  )rN  r  r  r   )r  rk  grY  r/  )r    r/  r  r]   rw   r#   	fillstyler&  r   r   r  )r   r   r   ru  rv  rO   r   r   Zcolumn_stackr   r   )
r  r   r  r   r   ZrxZryZarear?  r  r   r   r   test_scatter_markero  s2   
zTestScatter.test_scatter_markerZ
scatter_2DrE   )r  rG   c                 C   sP   t d}t d}t ||\}}|| }t \}}|j|||ddd d S )Nrz   r   r  face)r    r/  r  )rO   rT   r  r   r   r   )r  r-   r*   rM  r   r   r   r   r   test_scatter_2D  s   

zTestScatter.test_scatter_2DrF   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  )rO   r   r   r   )
r  r3   r4   r  r  r-   r*   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 )Nr   r   r   r   rl  ra  )r{   r|   r  r   r   r  r   r   r   test_scatter_color  s   "zTestScatter.test_scatter_colorc                 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   r   rz   r   z0.1z0.3r  rw   r   r  r  r  ?r    r]   r  )r   r   r   r   r   r   r   r   r   r   r   r   r   r   )	r   r   ru  rv  get_facecolorsr6  r   get_edgecolorsget_linewidthsr  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  r-   r  r  r  r  r  )r   r   r   r  r  r  r  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 yre  r   zfloat array-likerG  )rO   rT   r{   r|   r  r   r   )r  r-   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 )Nra  r   r  r   r  r   r   r   r   )r   r   rO   r  r  
same_colorr  r  r   r   r   test_scatter_edgecolor_RGB  s   z&TestScatter.test_scatter_edgecolor_RGBc                 C   s   |  }tdd}|dd |jtdtddtjdtjgg d|dd	 |  }t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 )Nr*  r+  r  r   r   r   r  Tr    r/  r`   Zplotnonfiniter   rz   r    r/  r`   )r/  r   r   r   r-  Zset_badr   r   rO   r  r  r3   r4   r   r`   r   r   r   test_scatter_invalid_color  s   $"z&TestScatter.test_scatter_invalid_colorc                 C   s~   |  }t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 )Nr*  r+  r  r   r   r   r  Fr  r   rz   r  r  r  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  ra  r  re  r  rd   r   r   )r    r  r  r  N)r   r  r{   r|   r  r   r  r  )r  r-   r   r   r   r   test_scatter_norm_vminvmax  s   "z&TestScatter.test_scatter_norm_vminvmaxc                 C   s:   |  }|jdddd |  }|jdgdgdgd d S )Nr   r  )r   r   )r  r3   r4   r   r   r   r   test_scatter_single_point  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 )Nrd   r   r   r  )rO   rT   r   r   rj   )r  r3   r4   r-   r   r   r   r   test_scatter_different_shapes  s
   
 $z)TestScatter.test_scatter_different_shapes)r  NZrgby
conversion)rX  N)r   N)NN)rk  r  rY  r   N)jauner  r  r   r   rz   r   r6  )r  z0.4r  r  )r  rX  r  C5)r  r   r  r  r  r   r   r   r   rX  z0.0r  r   r   r   zc_case, re_keyc                 C   s   dd }d}ddd}|d u rt jjj|di ||d d S tjt|| d	 t jjj|di ||d W d    d S 1 s>w   Y  d S )
Nc                   S   r   Nr   r   r   r   r   r   get_next_color3	  r   z2TestScatter.test_scatter_c.<locals>.get_next_colorr   z!^'c' argument has [0-9]+ elementsz^'c' argument must be a color)r6  r  r  )r    r  r=  xsizeget_next_color_funcre  )r   r  Axes_parse_scatter_color_argsr{   r|   r  )r  Zc_caseZre_keyr  r  ZREGEXPr   r   r   test_scatter_c1	  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)r   r   皙?)r   r   r  r   rz   rl  r   r   r  )r   r   rO   ri   r   )r  r3   r4   ZrgbZrgbarX   rW   r   r   r   test_scatter_single_color_cG	  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   rz   ra  r-   r   r   )r    r]   r/  r  C3lines.linewidth)rO   rT   r   r   r   r   rK   r  r   r   r   )r  r-   r   r   r   Zpcr   r   r   test_scatter_linewidthsW	  s   
z#TestScatter.test_scatter_linewidths)__name__
__module____qualname__r	   r  r  r  r
   r  r  r  r   r   contextr  r  r  r  r  r  r  r  Zparams_test_scatter_cr{   markparametrizer  r  r  r   r   r   r   r  c  sz    









	
	











+


r  )r  c                K   s   | ||d ur
||fS i |fS r   r   )r    r  r  r=  r   r   r   _paramsf	  s   r  _resultz	c, colorszparams, expected_resultrY  r   )r    r1  rk  r  r  rl  r  )r   r   r   r   c                 C   s@   dd }t jjj| d|i\}}}||jksJ t||j d S )Nc                   S   r   r  r   r   r   r   r   r  z	  r   z5test_parse_scatter_color_args.<locals>.get_next_colorr  )r   r  r  r  r    r   r1  )paramsZexpected_resultr  r    r1  Z_edgecolorsr   r   r   test_parse_scatter_color_argsk	  s   r  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   r  r   r   r   r   r   r  	  r   z@test_parse_scatter_color_args_edgecolors.<locals>.get_next_colorr    r  r   )r  r  )popr   r  r  r  )r=  Zexpected_edgecolorsr  r    r  rB  Zresult_edgecolorsr   r   r   (test_parse_scatter_color_args_edgecolors	  s   
r  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   r  r   r   r   r   r   r  	  r   z;test_parse_scatter_color_args_error.<locals>.get_next_colorz&RGBA values should be within 0-1 rangere  )r   r   r  )r   r  gffffff?r   )r=  r  r  )	r{   r|   r  rO   r   r   r  r  r  )r  r    r   r   r   #test_parse_scatter_color_args_error	  s   
"r  c                     s  ddl m  G  fddd} |  }|  }tj|_|  }tjg d|d}t| ks-J tj	t
dd tj|d}W d    n1 sEw   Y  ||u sPJ t  tj	t
dd tj|d}W d    n1 slw   Y  t|tjj ks~J tj	t
dd tj|d}W d    n1 sw   Y  ||u sJ tj	t
dd tj|d}W d    n1 sw   Y  ||u sJ | dksJ tj	t
dd tj|d}W d    n1 sw   Y  ||u sJ t  tjd	|d}t|tjj ksJ t  d S )
Nr   	PolarAxesc                       s    e Zd Zdd Z fddZdS )z#test_as_mpl_axes_api.<locals>.Polarc                 S   s
   d| _ d S )Nr   theta_offsetr  r   r   r   __init__	  s   
z,test_as_mpl_axes_api.<locals>.Polar.__init__c                    s    d| j ifS )Nr  r  r  r  r   r   _as_mpl_axes	  s   z0test_as_mpl_axes_api.<locals>.Polar._as_mpl_axesN)r  r  r  r  r  r   r  r   r   Polar	  s    r  r   r  z5Calling gca\(\) with keyword arguments was deprecatedre  y   )Zmatplotlib.projections.polarr  rO   r   r  r   r  typer{   rm  r   r   r   r   Z	_subplotsZsubplot_class_factoryZget_theta_offsetr   )r  ZprjZprj2Zprj3r   Z
ax_via_gcar   r  r   test_as_mpl_axes_api	  sb   r  c                  C   sZ   t  \} }t  \}}t | |t  u sJ | t  u s!J t |  t | d S r   )r   r   Zscar   Zgcfr   )fig1r   fig2r   r   r   r   test_pyplot_axes	  s   

r  Z
log_scalesc                  C   sL   t  \} }|ttdd |jddd |  |jddd d S )Nr   r   r         @r=  r8  )	r   r   rn   rO   r   r   r  r  r  r   r   r   r   test_log_scales	  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  r   r   rd   )	r   r   ro   r@   r   r   ZMultipleLocatorrp   rq   rA  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Attempted to set non-positivere  rb   rd   )	r   r   r  r{   rm  rn  r:  r  r;  r   r   r   r   test_log_scales_invalid
  s   

"r  Z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|d |d |d	 d S )Nr   rd   r   r   r   r  r   r   rd   )r   F   )r-   r
  r   y3r-   r
  r   r  rt  )	r   r   rO   r   r   	stackplotr:  r;  r   )r   r-   r
  r   r  r   r'  r   r   r   test_stackplot
  s   


r  Z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   r   r   r   r   rd   )rO   rT  r   rP   exprT   )r  rx  rs   r   jr-   r*   rM  r   r   r   layers#
  s   <z'test_stackplot_baseline.<locals>.layersrz   r   r  r   r  Zzeror9  rH  symr   r   Zwiggler  Zweighted_wiggle)rO   rP   rQ   r   r   r  r   rR  )r  r<  r   r	  r   r   r   test_stackplot_baseline
  s   

r  c                 C   r)  r   r   )r/  r   r   r   r+  9
  r,  c                 C   s   t jd tjjt jjddddfi | }t \}}|	ddr)|
d n|d |	d	d
s<tjd tjd< |j||fi | d S )N  r  r   )r5  r   )ZmeanZsigmasizevertTr   patch_artistFr  boxplot.boxprops.linewidth)rO   rP   rQ   r   ZcbookZboxplot_statsZ	lognormalr   r   getr  r  r   Zbxp)stats_kwargstransform_stats
bxp_kwargsZlogstatsr   r   r   r   r   _bxp_test_helper8
  s   
r  zbxp_baseline.pngr  )rO  r   c                   C   s
   t   d S r   r  r   r   r   r   test_bxp_baselineJ
  s   
r  zbxp_rangewhis.pngc                   C      t tddgdd d S )Nr   r   Zwhisr  r  r  r   r   r   r   test_bxp_rangewhisQ
     r  zbxp_percentilewhis.pngc                   C   r   )Nr   _   r  r  r  r   r   r   r   test_bxp_percentilewhisX
  r  r  zbxp_with_xlabels.pngc                  C   s   dd } t | d d S )Nc                 S   $   t | tdD ]\}}||d< q| S NZABCDr!   r9  r  statsr/  r!   r   r   r   r   c
     
z(test_bxp_with_xlabels.<locals>.transform)r  r  r   r   r   r   test_bxp_with_xlabels_
  s   r  zbxp_horizontal.pngr   )r  rO  r   r   c                   C      t tddd d S )NFr  r  r  r   r   r   r   test_bxp_horizontalk
  s   r  zbxp_with_ylabels.png)rO  r   r   c                  C      dd } t | tddd d S )Nc                 S   r  r	  r
  r  r   r   r   r   y
  r  z(test_bxp_with_ylabels.<locals>.transformFr  r  r  r  r  r   r   r   test_bxp_with_ylabelst
  s   r  zbxp_patchartist.png)r  rO  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 )NTrl  r   :)r   r   r  )r  boxpropsr  r  r   r   r   r   test_bxp_custompatchartist
     r  zbxp_customoutlier.pngc                   C      t ttdddddd d S )Nr   r<  r  ry   r]   mfc)Z
flierpropsr  r  r   r   r   r   test_bxp_customoutlier
     r!  zbxp_withmean_custompoint.pngc                   C   r  )NTr   r<  r   r  )	showmeansZ	meanpropsr  r  r   r   r   r   test_bxp_showcustommean
  r  r$  zbxp_custombox.pngc                   C   r  )Nri  rY  rz   ry   r   r   )r  r  r  r   r   r   r   test_bxp_custombox
  r"  r&  zbxp_custommedian.pngc                   C   r  )Nri  rY  rz   r%  )Zmedianpropsr  r  r   r   r   r   test_bxp_custommedian
  r"  r'  zbxp_customcap.pngc                   C   r  )Nri  r  rz   r%  )Zcappropsr  r  r   r   r   r   test_bxp_customcap
  r"  r(  zbxp_customwhisker.pngc                   C   r  )Nrh  rx  rz   r%  )Zwhiskerpropsr  r  r   r   r   r   test_bxp_customwhisker
  r"  r)  zbxp_withnotch.pngc                   C   r  )NT)Zshownotchesr  r  r   r   r   r   test_bxp_shownotches
  r  r*  zbxp_nocaps.pngc                   C   r  )NF)Zshowcapsr  r  r   r   r   r   test_bxp_nocaps
  r  r+  zbxp_nobox.pngc                   C   r  )NF)Z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)r  )r  r/  r   r   r   r   
  s   z*test_bxp_no_flier_stats.<locals>.transformF)Z
showfliersr  r  r  r   r   r   test_bxp_no_flier_stats
  s   
r.  zbxp_withmean_point.pngc                   C   s   t tdddd d S )NTFr#  Zmeanliner  r  r   r   r   r   test_bxp_showmean
     r0  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
  r1  r2  zbxp_scalarwidth.pngc                   C   r  )Nr  widthsr  r  r   r   r   r   test_bxp_scalarwidth  r  r5  zbxp_customwidths.pngc                   C      t tg ddd d S )N)r   r  g?g333333?r3  r  r  r   r   r   r   test_bxp_customwidths	     r7  zbxp_custompositions.pngc                   C   r6  )N)r   r   r
  rw  r  r  r  r   r   r   r   test_bxp_custompositions  r8  r9  c                   C   sB   t t ttdgdd W d    d S 1 sw   Y  d S )Nr   r3  r  r{   r|   r  r  r  r   r   r   r   test_bxp_bad_widths  s   "r;  c                   C   sD   t t ttddgdd W d    d S 1 sw   Y  d S )Nr   rz   r  r  r:  r   r   r   r   test_bxp_bad_positions  s   "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  rv  rw     rf  '  r   	bootstrapnotchrk  r   r-   )rA  rB  r'  	rO   rP   rQ   r   hstackr   r   r  r;  )r-   r   r   r'  r   r   r   test_boxplot#  s   
rF  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  rv  rw  r=  r>  rf  r   r   r?  ^)rA  r  rC  r  rD  )r-   r   r   r   r   r   r   test_boxplot_sym26  s   
rH  zboxplot_sym.pngc                  C   sJ   t ddd} t d| dg} t \}}|j| | gdd |d d S )	Nrv  rw  r=  r>  rf  r  r  rC  )rO   r   rE  r   r   r  r;  r  r   r   r   test_boxplot_symF  s
   rJ  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   r   r?  r   r@  )r  r   T)rA  rB  Z	autorange)	rO   rP   rQ   ri   rE  r   r   r  r;  )r-   r  r   r  r   r   r   r   test_boxplot_autorange_whiskersS  s   

rK  c                 C   s\   t ddd}t d|dg}t| | ||g W d    | S 1 s'w   Y  | S )Nrv  rw  r=  r>  rf  )rO   r   rE  r=   r   r  )r   rc_dictr-   r   r   r   _rc_test_bxp_helperf  s   
rM  Zboxplot_rc_parameters)rO  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"d#gdd$}dd!d!d!dddddddd%d&}|||g}t||D ]	\}}t|| qjtjj	d'd( |d) 
 D v sJ d S )*Nr  rz   zboxplot.notchTboxplot.whiskersr   r  zboxplot.bootstrapr?  zboxplot.flierprops.colorrY  boxplot.flierprops.markerrw   z"boxplot.flierprops.markerfacecolorr  z"boxplot.flierprops.markeredgecolorzboxplot.flierprops.markersizezboxplot.flierprops.linestyleri  zboxplot.flierprops.linewidthr   zboxplot.boxprops.colorrk  r  zboxplot.boxprops.linestylezboxplot.capprops.colorr    zboxplot.capprops.linewidthzboxplot.capprops.linestylezboxplot.medianprops.colorr  )zboxplot.medianprops.linewidthzboxplot.medianprops.linestyleFr   r   )zboxplot.verticalrN  zboxplot.patchartist-.)rN  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>r   )rO   rP   rQ   r   r   r9  rM  r=   r   r   Zget_children)r   r   Zrc_axis0Zrc_axis1Zrc_axis2	dict_listr  Zrc_axisr   r   r   test_boxplot_rc_parametersn  s~   	

rR  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  rv  rw  r=  r>  rf  r  r  r        @r?  r   r   )rA  usermediansconf_intervalsrB  rC  )
rO   rP   rQ   r   rE  r   r   r   r  r;  )r-   r   r   ZCIsr   r   r   test_boxplot_with_CIarray  s   rV  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)rz   i(#  rq  X   i^  i@ ix  i  r  r   Fminorwhich)
rO   r   r  r   r  r  r  rC   r  r@   )r-   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 )	Nrv  rw  r=  r>  rf  r   r   )rT  	rO   r   rE  r   r   r{   r|   r  r  r  r   r   r   test_boxplot_bad_medians  s    "r]  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 )	Nrv  rw  r=  r>  rf  r   r   )rU  r\  r  r   r   r   test_boxplot_bad_ci  s   "r^  c                  C   sZ   t d} t \}}|| d d  dksJ |j| ddd d  dks+J d S )Nrd   boxesr   r   r,  )rO   rT   r   r   r  r2  )r-   rw  r   r   r   r   test_boxplot_zorder  s   
&ra  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/  rq  rw   rO  rG  zboxplot.meanprops.markerr   rq  rb   T)r#  )ZwhiskersZcapsr_  Zmediansr   r-  r   Zmeans)r   r   r   rO   rT   r  Z
get_marker)r   r   Z	test_dataZ
bxp_handleZ	bxp_linesZ	each_liner   r   r   test_boxplot_marker_behavior  s   



rb  z&boxplot_mod_artists_after_plotting.pngc                  C   sH   g d} t  \}}|j| dd}|D ]}|| D ]}|d qqd S )N)r4  r  r  r  gQ?gQ?gQrw   rI  r   )r   r   r  	set_color)r-   r   r   Zbpkeyr  r   r   r   &test_boxplot_mod_artist_after_plotting  s   re  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  rO   rP   rR   r   rB  r   r   r   r   
      z1test_vert_violinplot_baseline.<locals>.<listcomp>r   r   r  r#  showextremashowmediansr<  )r  r#  rm  rn  r'  )rO   rP   rQ   r   r   r  
violinplotr   )r'  r   r   r   r   r   test_vert_violinplot_baseline  s   
rp  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   rf  rg  ri  rj  r   r   r   r     rk  z2test_vert_violinplot_showmeans.<locals>.<listcomp>r   r   r   rl  r   r  rO   rP   rQ   r   ro  r   r'  r   r   r   test_vert_violinplot_showmeans     
rs  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   rf  rg  ri  rj  r   r   r   r   %  rk  z4test_vert_violinplot_showextrema.<locals>.<listcomp>r   r   r   rl  rq  rr  r   r   r    test_vert_violinplot_showextrema   rt  ru  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   rf  rg  ri  rj  r   r   r   r   /  rk  z4test_vert_violinplot_showmedians.<locals>.<listcomp>r   r   r   rl  rq  rr  r   r   r    test_vert_violinplot_showmedians*  rt  rv  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   rf  rg  ri  rj  r   r   r   r   9  rk  z0test_vert_violinplot_showall.<locals>.<listcomp>r   r   r   ?r   rD  r   r  r  r  )r  r#  rm  rn  	quantilesrq  rr  r   r   r   test_vert_violinplot_showall4  s   
ry  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   rf  rg  ri  rj  r   r   r   r   D  rk  z8test_vert_violinplot_custompoints_10.<locals>.<listcomp>r   r   rd   r  r#  rm  rn  pointsrq  rr  r   r   r   $test_vert_violinplot_custompoints_10?     
r}  z$violinplot_vert_custompoints_200.pngc                  C   rz  )NiYqVc                 S   rf  rg  ri  rj  r   r   r   r   N  rk  z9test_vert_violinplot_custompoints_200.<locals>.<listcomp>r   r   r  r{  rq  rr  r   r   r   %test_vert_violinplot_custompoints_200I  r~  r  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   rf  rg  ri  rj  r   r   r   r   X  rk  z2test_horiz_violinplot_baseline.<locals>.<listcomp>r   Fr   r  r  r#  rm  rn  rq  rr  r   r   r   test_horiz_violinplot_baselineS  r~  r  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   rf  rg  ri  rj  r   r   r   r   b  rk  z5test_horiz_violinplot_showmedians.<locals>.<listcomp>r   Fr   r   r  rq  rr  r   r   r   !test_horiz_violinplot_showmedians]  r~  r  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   rf  rg  ri  rj  r   r   r   r   l  rk  z3test_horiz_violinplot_showmeans.<locals>.<listcomp>r   Fr   r   r  rq  rr  r   r   r   test_horiz_violinplot_showmeansg  r~  r  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   rf  rg  ri  rj  r   r   r   r   v  rk  z5test_horiz_violinplot_showextrema.<locals>.<listcomp>r   Fr   r   r  rq  rr  r   r   r   !test_horiz_violinplot_showextremaq  r~  r  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   rf  rg  ri  rj  r   r   r   r     rk  z1test_horiz_violinplot_showall.<locals>.<listcomp>r   Fr   r   rw  r   rD  r   r  r  r  )r  r  r#  rm  rn  rx  rq  rr  r   r   r   test_horiz_violinplot_showall{  s   
r  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   rf  rg  ri  rj  r   r   r   r     rk  z9test_horiz_violinplot_custompoints_10.<locals>.<listcomp>r   Fr   rd   r  r  r#  rm  rn  r|  rq  rr  r   r   r   %test_horiz_violinplot_custompoints_10     
r  z%violinplot_horiz_custompoints_200.pngc               	   C   r  )	Ni<9!c                 S   rf  rg  ri  rj  r   r   r   r     rk  z:test_horiz_violinplot_custompoints_200.<locals>.<listcomp>r   Fr   r  r  rq  rr  r   r   r   &test_horiz_violinplot_custompoints_200  r  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   rf  rg  ri  rj  r   r   r   r     rk  z1test_violinplot_bad_positions.<locals>.<listcomp>r   r   r  
r   r  rO   rP   rQ   r   r{   r|   r  ro  rr  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   rf  rg  ri  rj  r   r   r   r     rk  z.test_violinplot_bad_widths.<locals>.<listcomp>r   ra  )r  r4  r  rr  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   rh  r   r   r   r  rx  
r   r  rO   rP   rQ   rR   r{   r|   r  ro  rr  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   rh  )r   r   r   ?r  )gr   r   r  r  rr  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   rh  )r   r   rw  r  )rO   rP   rQ   rR   r   ro  )r3   r4   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  rw  rh  )r  r  rw  r
  r   r   rz   r  r  Z	ABCDEFGHIr  )	rO   rP   rQ   r  rR   r  r   ro  values)r3   r4   r  s1s2Z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   rd   rz   r  r   r   F)r  Zmanage_ticks)
r   r   r:  rp   rO   rP   rQ   rR   r  r   )rB  r   Zold_xlimr
  r   Z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   rz   r  r   )r  r  c                 S      g | ]}|  qS r   get_textr  r   r   r   r     r  z+test_boxplot_not_single.<locals>.<listcomp>r   5)r   r   r  rO   rP   r   r  r   rp   r  
get_xticksrI  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   r   z	font.sizeP   raw)rM  format)r   rn   r=   r   updateioBytesIOsavefigrY  r   r   r   test_tick_space_size_0  s   r  )errorbar_basicZerrorbar_mixedr  c            
   	   C   s  t dd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}	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 yr   T)r  r  r  r  rw   r  r   zVert. symmetricZnbinsrH  )r  r   r  zHor. symmetric w/ alphar  z--or  r  r   zH, V asymmetricr  r   r!  r  )r  r  r   ecolorZcapthickzMixed sym., log yzVariable errorbarsr   r-   r*   r  r  r'  )rO   rT   r  r  r   r   r   ro  r  r   locator_paramsr  ZmaximumZsuptitle)
r-   r*   r  r  r   r   r	  ZylowerZ
yerr_lowerr'  r   r   r   test_errorbar  sB   


$




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 )Nrd   r   r  r   r  C1C2)	r   r   rO   rT   ro  rn   r  to_rgba	get_color)ro  r   r-   r*   rq  rB  rr  Zln1r   r   r   test_errorbar_colorcycle2  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   )rO   r  r   r  r-   r   r   r   D  r?  z.test_errorbar_cycle_ecolor.<locals>.<listcomp>)r  r  r  r  r  rh  rw   r  )r  ry   r]   r  )r  ry   r]   r   r  )rO   rT   r   r   r9  ro  )r3   r4   r*   ZaxtZaxryir   r   r  r   test_errorbar_cycle_ecolorA  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   r   rw   r  )r  r   r  )r   r   r   rO   rT   r  r  vstackrR  r{   r|   r  ro  )r   r   r-   r*   Zyerr1r  r  r   r   r   test_errorbar_shapeP  s    "r  Z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)r   r   r  r   )r  r  uplimsr  r   )r   r   r  r   rX  )r  r  lolimsr  r   r   rw   r  magenta)r]   r  r  r  r  r  r  r   r   rz   r
   @r   r   Zcyan)r]   r  r  r  xlolimsxuplimsr  r  r  r  capsizer   )r   r  zErrorbar upper and lower limits)
rO   rT   r  r   r   ro  Z
zeros_likeZ	full_liker:  r  )r-   r*   r  r  r  r   r   r  r  r  r  r   r   r   test_errorbar_limitsb  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   r   r   )r  r  r   r  )rO   rT   r   ro  r  r  r  r  )r-   r*   plotlinerB  ZbarlinesZerrbarr   r   r   test_errobar_nonefmt  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   r   Noner/  rK   z	steps-midroundr:  r;  Zbevel)
r  r  r  r]   r  r?  Zdash_capstyleZdash_joinstyler>  r=  )rO   rT   r   ro  Zget_fillstyleZget_drawstyle)r-   r*   r  rB  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)r   r   rd   r   r   ri  r/  r  )r-   r*   r  r  r]   r   )r   rz   r   z	tab:greenr  r*   )r-   r*   r  r   r  r]   r   r   r   r
  ztab:bluerP  rw   r    )r-   r*   r  r   r  r]   r   r   r  )ri  r  rP  )r/  r/  rw   )r  r*   r    )rY  r  rk  )r  r]   r   r   r  )Z
prop_cycle)r-   r*   r  )r-   r*   r  r   )r-   r*   r  r   )r   ro  r:  r   r   rc)r3   r4   r   Z_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   r   r   r
  znot a tuple of two integersre  ra  )
errorevery)r  rz   znot a valid NumPy fancy indexFTznot a recognized valueZfoobar)	rO   r   r   r   r   r{   r|   r  ro  )r-   r*   r  r   r   r   r   test_errorbar_every_invalid  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   r   r   r
  Zrgbk)r   r   r   rw  r  r   )r  r  r    r  )r    r,  r   )r  r    r   r   )r   r   rw   )rz  r  r   r`  )r  r   r   r   rz   r  )r  rz  r  r    r   )FTFr   r  )rO   r   r   r9  ro  rn   r  )	r3   r4   r-   r*   r  rX   rW   r   shiftr   r   r   test_errorbar_every  sB   (*
,,
,,
r  
elinewidthra  c                 C   s"   t jg dg dg d| d d S )Nra  )r  r  )r   ro  )r  r   r   r   test_errorbar_linewidth_type  s   "r  Z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 )Nr   rz   r  r   rd   r  r  Tr  r-   )r
  r  r'  rO   r   r   r   r  )d1d2r   r   r'  r   r   r   test_hist_stacked_stepfilled  s   r  Zhist_offsetc                  C   sH   t ddd} t ddd}t \}}|j| dd |j|d	d d S )
Nr   rd   r  r   rz   r  r   r  r   r  r  r  r   r   r   r   r   test_hist_offset  s
   r  zhist_step.pngc                  C   sD   t ddd} t \}}|j| dd |dd |dd	 d S )
Nr   rz   r  r  r	  r   rd   rb   r   )rO   r   r   r   r  r;  r:  r  r   r   r   r   r   test_hist_step  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   rd   r  r   rz   r  r  r9   )r
  r;   r  r  r   r   r   test_hist_step_horiz(     r  Z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   rd   r  r   rz   r  r!  rS  r  r   r  T)Zweightsr
  r  r  )r  r  Zw1Zw2r   r   r   r   r   test_hist_stacked_weighted1  s   r  use_line_collectionzw/ line collectionzw/o line collectionzstem.pngc              	   C   sJ   t ddt j d}t \}}|j|t |dddd| d |  d S )	Nr   r   r   C2-.zk+C1-. )linefmt	markerfmtbasefmtr!   r  )rO   r   r   r   r   stemr   r/   r  r-   r   r   r   r   r   	test_stem<  s   r  c                  C   s`   t  \} }ttd}ttd}|| ||| |j||dd |j||ddd d S )Nrd   r   )r  zb--)r  r  )r   r   r  r   r  )r   r   r-   r*   r   r   r   test_stem_argsK  s   
r  c                  C   s@   t dd\} }tjdtjdg}ddg}||| d S )Nr   z2013-9-28 11:00:00z2013-9-28 12:00:00r   r  )r   r   dateutilparserparser  )r   r   xsysr   r   r   test_stem_datesX  s   

r  zstem_orientation.pngc              	   C   sB   t ddt j d}t \}}|j|t |ddd| dd d S )	Nr   r   r  r  Zkxr  r9   )r  r  r  r  r;   )rO   r   r   r   r   r  r   r  r   r   r   test_stem_orientation`  s   
r  Zhist_stacked_stepfilled_alphac                  C   sB   t ddd} t ddd}t \}}|j| |fddd	d
 d S )Nr   rz   r  r   rd   r  r  Tr   )r
  r  r  r  r  r   r   r   "test_hist_stacked_stepfilled_alphal  s   r  Zhist_stacked_stepc                  C   r  )
Nr   rz   r  r   rd   r  r  Tr  r  r  r   r   r   test_hist_stacked_stepu  r  r  Zhist_stacked_normedc                  C   s@   t ddd} t ddd}t \}}|j| |fddd d S )	Nr   rz   r  r   rd   r  T)r  r!  r  r  r   r   r   test_hist_stacked_density~  r  r  zhist_step_bottom.pngc                  C   s4   t ddd} t \}}|j| t ddd d S )Nr   rz   r  rd   r  )r7   r
  )rO   r   r   r   r  rT   r  r   r   r   test_hist_step_bottom  s   r  c                  C   s   g d} g d}t j|| d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ddgddgddgddgddgg}t| | d S )	Nr   r   r   rz   r   r   r   r   r   r   r  r  r
  r   r   r   rz   r   r  r   get_xyr  r'  rB  polygonr)  r   r   r   test_hist_stepfilled_geometry  s   *$r  c               	   C   sn   g d} g d}t j|| d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}t| | d S )	Nr  r  r  r  r   r   r   rz   r  r  r   r   r   test_hist_step_geometry  s   4r  c                  C   s   g d} g d}t j|| g dd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d
dgd
d
gdd
gddgddgg}t| | d S )Nr  r  r   r   r   r  r  r7   r
  r   r   rz   r   r   r  r   r  r  r   r   r   $test_hist_stepfilled_bottom_geometry  s   *$r   c               	   C   st   g d} g d}t j|| g dd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}t| | d S )Nr  r  r  r  r  r   r   rz   r   r   r  r   r  r  r   r   r   test_hist_step_bottom_geometry  s   4r  c                  C   s  g d} g d}g d}t j||g| ddd\}}}t|dks"J |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ddgddgd	dgd	dgddgg}t| | |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dd	gdd
gd	d
gd	dgddgg}t| | d S )Nr  r  r   Tr  r  r  r
  r   r   r   rz   r   r   r  r  r   r  r  Zdata_1Zdata_2rB  r   r  r)  r   r   r   %test_hist_stacked_stepfilled_geometry  s&   

*$
*$r  c               	   C   s   g d} g d}g d}t j||g| ddd\}}}t|dks"J |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}t| | |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}t| | d S )Nr  r  r   Tr  r  r   r   r   rz   r   r  r  r   r   r   test_hist_stacked_step_geometry  s   

4
4r  c                  C   s"  g d} g d}g d}t j||g| dg ddd\}}}t|dks%J |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ddgddgd
dgd
d
gd	d
gg}t| | |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ddgddgd
dgd
dgd	dgg}t| | d S )Nr  r  r   Tr  r  r  r  r7   r
  r   r   r   rz   r   r  r   r   r
  rS  r  r  r   r   r   ,test_hist_stacked_stepfilled_bottom_geometry  s(   

*$
*$r  c               	   C   s   g d} g d}g d}t j||g| dg ddd\}}}t|dks%J |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}t| | |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}t| | d S )Nr  r  r   Tr  r  r  r   r   r   rz   r   r  r   r   r
  rS  r  r  r   r   r   &test_hist_stacked_step_bottom_geometry  s    

4
4r	  Z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  r  r   6  r  )r     r  r  r   i  )r  r  r  r=  r=  rq     )rN  rN  rN  rN  r  r  r   r   r   r   rd   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  ))gXxn?r   r  )r  r   g `p?)r  r   gǟ?)r  gˢ`?r   )gC?r  r   )gK?r  r   )r   r  z yellowr  r  rd   r  mid)r  r
  r  r   r!   zupper right)r   r   r   )r%   Zbbox_to_anchorZncol)r   r   r  r/   )r<  r1  labelsr   r   r   r   r   test_hist_stacked_bar  s   r  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 )Nrd   r  r   r   r   r  )r
  r7   )rO   r   r   r  tolistr  r   r   r   %test_hist_barstacked_bottom_unchanged#  s   r  c                  C   s,   t  \} }|jg tdtdgdd d S )Nrd   r  r	  )r   r   r  r   r   r   r   r   test_hist_emptydata)  s    r  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   r   r|  r  Z
_nolegend_Z00)r   r   r  rA   )r   r   rB  barsr   r   r   test_hist_labels.  s   r  Ztransparent_markersc                  C   s:   t jd t jd} t \}}|j| dddd d S )Nr   r  rb  r   r   )r   rc   rO   rP   rQ   r   r   rn   r"  r   r   r   test_transparent_markers=  s   r  Z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 )Nr   r  r  r  r   )r   r   r   r   r   r   r  )r  r  r   rw   )r   r  r  rz   r-   )r  r  )rb   r   r   r   )r   r   r  r  rn   r  )r   r	  ZrcolorsZbcolorsZalphaskwr   r  r  Zrcolorr  Zbcolorr   r   r   r   test_rgba_markersF  s&   *r  Zmollweide_gridc                  C   s    t  } | jdd}|  d S )N	mollweider  )r   r   r   r  r   r   r   r   test_mollweide_gridW  s   r  c                  C   s   t  } | jdd}ttj tjd}ttj d tjd d}t||\}}t| | fj	}|j
|}|j
 |}tj||d d S )Nr  r  rd  r   r  rz   )r   r   r   rO   r   r   r  r  rB  rR  transProjectionr   invertedr   r   )r   r   ZlonZlatllr)  Z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 )Nr  r  r   r     rz   )r   r   r   rO   r   r  r  rB  rR  r  r  r   r   r   )r   r   r-   r*   r)  r  xy2r   r   r   &test_mollweide_inverse_forward_closurev  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  r  r  rd   )r   r   rc   r   r   r   )r   r   rc   r   r  r   r   r
  r  r  r"  r   r   r   r$    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   rZ   r  r
  )r   r   r  r  r  r  )r   r  r   )r   r   r   )r   r   r   r3  gQ?)ir#  r   r   r
  rd   )r   r   r   r   rz   r   )r1  lineoffsetslinelengths)r   r    lor  r   r    r(  r  )r1  r&  r'  r'  )
rO   rP   rQ   r  r  rT   r   r   r   r%  )r{  data2r'  Znum_datasetsZcolors1Zcolors2r1  Zlineoffsets1Zlineoffsets2r&  Zlinelengths1Zlinelengths2r'  r   axobjZcollsZnum_collectionsr   r   r   test_eventplot  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   rZ   r  r
  N)rO   rP   rQ   r  r   r   r   r%  )r{  r)  r'  r   r*  r   r   r   test_eventplot_defaults  s   r,  r1  ))r  )z
tab:orangeztab:pinkztab:cyanZbLacK)rX  r  N)r   r   r   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   r   r   rz   c                 S   s   g | ]
}|d ur
|ndqS )Nr  r   r   r    r   r   r   r     r?  z)test_eventplot_colors.<locals>.<listcomp>r   r1  N)r  rO   Zbroadcast_tor  Zto_rgba_arrayr   r   r%  r9  r   r  )r1  r'  expectedr   r   r  r  r   r   r   r   test_eventplot_colors  s   r0  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   r  rd   rk  rY  r    rx  r   r   rx   Zdasheddashdotr  )r1  r   r  r   rx  ry   rz   c                 s   s    | ]	}t |jtV  qd S r   )
issubclasscategoryr   )r   Zwir   r   r   r    s    z0test_eventplot_problem_kwargs.<locals>.<genexpr>N)
rO   rP   rQ   r  r   r   r   r%  r  r  )Zrecwarnr{  r)  r'  r   r*  r   r   r   test_eventplot_problem_kwargs   s$   r4  c                  C   s.   t dd\} }|jg gdgd t   d S )Nr   )r  r  r  r  r.  r   r   r%  r   r   r   r   r   test_empty_eventplot  s   r6  r'  r;   )NZverticalr9   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.Nr;   r   r5  )r'  r;   Zoptsr   r   r   r   r   test_eventplot_orientation#  s   r7  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 )	Nr  r   r   rz   r   r+  r   )r&  )r\  r   r%  )r3   r4   Zts_1Zts_2r   r   r   r   test_eventplot_units_list-  s   r8  zmarker_styles.pngc               	   C   sv   t  \} }tttjjjdd d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>)rd  r   r   rd   r   )ry   r]   rc   r!   )r   r   r  sortedr=   markersrv  rn   rO   rT   ri   )r   r   r*   r]   r   r   r   test_marker_styles<  s   .r;  zrc_markerfill.pngc                  C   sP   t  \} }td}tg dD ]\}\}}|tjd< |j|| |d qd S )Nrw  ))r#   r/  )r7   rw   )r   rG  zmarkers.fillstylerg  )r   r   rO   rT   r  r=   r   rn   )r   r   r-   idxr   r]   r   r   r   test_markers_fillstyle_rcparamsE  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 )Nrd   ))rb   rb   r}  r  r~  r   r  r  rb   rY  )r  r   r   r   rn   r:  r;  )r'  Zmarker_as_tupleZmarker_as_listr   r   r   r   r   test_vertex_markersO  s   r>  Zvline_hline_zorderZerrorbar_zorderZx86_64c                  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 )Nrd   r   )r   r,  r   rX  )r   r   r,  r   rw  rx  r   r  rz   zaxvline and axhline zorder testrk  )r  r,  r   r   )r   r   r,  zerrorbar zorder test)r  r   r   r   r   rn   rV   axvliner  rO   rT  ro  )r-   r   r   r*   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   rd   )r  r   r  r   r  z.-)r   r  )rd   r!  r  r  r   rd   r  r!  )r   ro   r  axliner3   r4   r   r   r   r   test_axline_loglogy  s   rC  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 )Nr~  r  r  r  r  r  rl  )r   r   r   r   r  )gffffffr  r   r  )sloper   )r   r  r  ZC4)r  r   infr  rb   r   r   )r   ro   rA  r  rn   rV   r?  rB  r   r   r   test_axline  s    rG  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 )Nr~  r  r  r   rE  r   rD  r  rE  r   r   r   r   r   r  )r   r   )r   r   r  )r   r   rb   rl  )r   ro   rA  r   rn   rB  r   r   r   test_axline_transaxes  s   rK  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 )Nr~  r  r  r   rH  rJ  r   r  rI  r   r  )r   r   r  rz   r   rd   rl  )r   ro   rA  r   set_size_inchesrn   rB  r   r   r   test_axline_transaxes_panzoom  s   rM  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  Nr  r   )rE  r   r  )
r   r   r{   r|   r}   rA  r  r  r  r   r   r   r   r   test_axline_args  s*   




"rN  )Zvlines_basicZvlines_with_nanZ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)r   rz   r   r   rw  )r   rz   r  r   r   r  r   r1  r   )r   rz   r   r   r
  rw  r   rQ  rz   r  )r   r  )r  r  rk  ri  )r1  r   ry   r  rd   )r   r   r
  r  rd   r3  )r   r   rb   r   r   r   )rc  r4  r   r+     r  r  r   )	r   r   rU   rO   r  r:  rq   r   masked_equal)r  r
  r  r   r   r   r  r   r   rc  x3r  x4y4r   fig3ax5Zx5Zymin5Zymax5r   r   r   test_vlines  2   rZ  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.colorrX  r   r   r   )r   r   r   r   rU   r1  r  r  r   r   rA  r   r   r   test_vlines_default  
   "r_  )Zhlines_basicZhlines_with_nanZhlines_maskedc                  C   rO  rP  )	r   r   hlinesrO   r  r;  rp   r   rT  )r
  r  r  r   r   r   r  r   r   rc  r  rU  rW  rV  r   rX  rY  Zy5Zxmin5Zxmax5r   r   r   test_hlines  r[  rb  c                  C   r\  r]  )r   r   r   r   ra  r1  r  r  r^  r   r   r   test_hlines_default-  r`  rc  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)rX  r   r   r$  r  r   r   r   r   rR  )r   r   rz   r   )rX  r   r   r  )r   rU   ra  )r3   r4   r'  Ztest_colorsZ	expect_xyZexpect_colorr   r   r   test_lines_with_colors4  s   
rd  Zstep_linestylec               	   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 )Nrd   r   )rh  ri  rP  r  r   Zpre)r   ry   r  r   r  rH  rb   rw  )r  Y0Y1Y2r  re  )r   ry   r  r'  rf  rg  )	rO   rT   r   r   rB  r9  r  r:  r;  )r-   r*   r   ax_lstZ	ln_stylesr   r  r'  r   r   r   test_step_linestyleF  s*   ri  Z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)r  r  r   r   r   )radiusr   r   T)Zmatch_originalr   r  r   r   r6  ii)r   r   r   r   )r   r   r   r   r  r+  )r   r   r/  Circler   r  ZPatchCollectionZset_offsetsZset_linewidthsZset_edgecolorsr   rc  r  r:  r;  )r   r   r    p1p2r   r   r   test_mixed_collectiond  s   

rn  c                  C   s:   t tdtdd} |   |   dksJ d S )Nr   r   )r   r   r   r   )r   r   rO   Zint32Zint64r   Zget_subplotspecZget_geometryr  r   r   r   test_subplot_key_hash~  s   ro  )zspecgram_freqs.pngzspecgram_freqs_linear.pngzspecgram_noise.pngzspecgram_noise_linear.pnggQ?)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  r  rU  r   r   r  @ffffff@'@rd   r   r   r   rB  r  onesidedtwosidedrz   )NFFTFsnoverlappad_tosidesr  )rv  rw  rx  ry  rz  r   r  N)r=   r   intrO   r$  rT   r  r   r   multiplyouterr  rP   rQ   r  r   ceillog2r9  r   r   r   specgramr1  LogNorm)r  rw  fstims
NFFT_freqsr-   y_freqs
NFFT_noisey_noise	all_sidesr*   rv  rx  ry  r   rz  r   r   r   test_specgram  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  r  rU  r   r   r  rp  rq  rr  r   r   r   r   rB  Nrb   rd   rs  rz   Z	magnituderv  rw  rx  ry  rz  rL   r  )rv  rw  rx  ry  rz  rL   r   r  )r=   r   r{  rO   r$  rT   r   r   r|  r}  r  rE  rP   rQ   r  r  r   r~  r  r9  r   r   r   r  r1  r  )r  rw  r  r  r-   r*   r  r  r  r  rv  rx  ry  r   rz  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  r  rU  r   r   r  rp  rq  rr  rd   r   r   r   rB  Nrb   rs  )rm  Zphaserz   r  dB)rv  rw  rx  ry  rz  rL   r   )r=   r   r{  rO   r$  rT   r   r   r|  r}  r  rE  rP   rQ   r  r  r   r~  r  r9  r   r   r   r  r{   r|   r  )r  rw  r  r  r-   r*   r  r  r  r  rv  rx  ry  rL   r   rz  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  )rw  r   rZ   `   )r   r  rO   ri   Z
get_extent)specfreqsr.  r  rM   rN   Zfreq0Z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)rO   rT   r   r   r   r   r  )r3   r4   r.  signalr   r   r   test_specgram_origin_rcparam  s   



r  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)rO   rT   r   r{   r|   r}   r   r  )r.  r  r   r   r   test_specgram_origin_kwarg  s
   

"r  )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  rp  rq  rr  r  r   r   r   rB  rz  r  rt  F)rz  return_lineru  Trz   )rv  rw  rx  ry  r  r   ZxlabelZylabel)r{  rO   r$  rT   r   r   r|  r}  r  rP   rQ   r  r   r~  r  r9  r   r   r   psdr  r  r  ro   csd)r  rw  r  r  r-   Zys_freqsr  Zys_noiseZ
all_kwargsr  rv  rx  ry  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?  r  r   r   r  r  r   r   r   rd   rz   rB  r   rs  )rw  ry  rz  r   r  r   r  )r{  r$  rO   r~  r  rT   r   r   r}  r  rP   rQ   rE  r  r   r9  r   r   r   Zmagnitude_spectrumro   Zangle_spectrumZphase_spectrum)r  rw  Zfstims1rv  ry  r-   r  r  r  r=  r*   r   rz  r  r  r0  r   r   r   test_spectrum7  s<   ("
r  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 )	Nr   T)r  r  r  r   r   r   )
r   r   r   rC   ro   rO   r  r  rT  r  )r	  r   r   r   r   test_psd_csd_edge_casesa  s   "r  c                 C   sH   |   }| }| }|  |  |  }|j  |j  d S r   )r   r   r   remover@   Ztick_bottomrC   Z	tick_left)r3   r4   rW   Zax_twinxZax_twinyrX   r   r   r   test_twin_removek  s   
r  ztwin_spines.pngc            	      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)Zset_frame_onr   set_visiblespinesr  r   r   r   make_patch_spines_invisible}  s   
z5test_twin_spines.<locals>.make_patch_spines_invisible)r   rz   r  r  )r'   )r  r  Tr   zb-)r   rz   r   zr-)r  r   r   zg-r   r   r   r   A   r   )r  r4  r*   )r  r1  r  r-   r   )r   r   rE  r   r   r  r'   set_positionr  rn   r:  r;  rC   r!   rc  r  r  tick_params)	r  r   ZhostZpar1Zpar2rl  rm  Zp3Ztkwr   r   r   test_twin_spinesz  s0   r  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   r  r   )r  iL  i  i  )r  i-  rd  r  r   g     @@z#BEAED4rl  rD  )r   r  z#7FC97Fr   )r   r  r   r  )r   r'  )r   r  r'  )
r=   r   r   r   r   rO   r   r   rn   r  )r   r   r'  r   r   r   r   test_twin_spines_on_top  s(   


r  z(grid_which, major_visible, minor_visible))rh   TT)majorTF)rX  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)rr  zaxes.grid.whichc                 3       | ]
}|j   kV  qd S r   gridliner   r   tick)major_visibler   r   r        z*test_rcparam_grid_minor.<locals>.<genexpr>c                 3   r  r   r  r  )minor_visibler   r   r    r  )r   r   r  r   r   r  r   r  r@   
majorTicks
minorTicks)Z
grid_whichr  r  r   r   r   )r  r  r   test_rcparam_grid_minor  s   
r  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ZvisibleT)	r   r   r  r  r   r@   r  r  r   r   r   r   r   	test_grid  s   



r  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 )	NTr  rd   )resetrZ  	labelsizer   rX  rl  rr  )
r   r   r  r@   r  r  r   r  r   clearr   r   r   r   test_reset_grid  s   "r  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inrd   r   r  r3  r4  r  re  r  rD  rz   ri  )	directionlengthr4  r   padr  
labelcolorlabelrotation
grid_color
grid_alphagrid_linewidthgrid_linestyle)r   r  r  Zdraw_without_renderingr  r@   Zreset_ticksrC   )r3   r4   r   r   r   r   r   test_reset_ticks  s   




r  c                  C   s:   t  } |  }|d |g d t| d d S )Nr   )r  r   r   r   )r  r   )r   r   r   r?  rn   r   rq   r   r   r   r   test_vline_limit  s
   
r  zfv, fh, args)r   r  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 )	Nr   z"ymin must be a single scalar valuere  )yminymaxz"ymax must be a single scalar valuez"xmin must be a single scalar value)rM   rN   z"xmax must be a single scalar value)r=   r  Znum2dater{   r|   r  )ZfvZfhargsZ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 )
Nr   r   T)r  r  r  r   r   ra  )r   r   r
  rz   r
  )r   r   rn   rp   rq   )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datalimr   T)r  r  r   )
adjustablesharer   r  rQ  )r   r   
set_aspect
get_aspectZ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 )	Nr   T)r  r  r   r   )r  r   rz   r   )r   r   r  rn   r   rp   rq   r  r   r   r   test_shared_with_aspect_27  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  r   T)r  r   r   r  r   r   rz   r   )r   r   r  rn   r   rp   rq   Zbbox_inchesro  r4  get_positionviewLimr  r  )	r  r   r	  Z
fig_aspectr   r   
box_aspectZ
lim_aspectr/  r   r   r   test_shared_with_aspect_3B  s$   r  c                 C   sH   t  \}}t|d|  }|d |d t|jj|jj d S )Nztwin{}r   r   )r   r   r   r  r  r   bboxextents)r   r   r   r   r   r   r   test_twin_with_aspectV  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  rU  )      $r  )r  r  FT)Zvisible_only)
r=   r   r   r   rn   rp   rq   r  Zrelimr   )r  r
  r   r   r   r   r0  r   r   r   test_relim_visible_onlya  s&   

r  c                  C   s,   t  } |  }|jdd |jdd dS )z
    tests for issue #1172
    Zlarge)r  outr  N)r   r   r   r  r   r   r   r   test_text_labelsizez  s   r  zpie_default.pngc               	   C   sD   d} g d}g d}d}t jdd\}}|j||| |ddd	d
 d S )NFrogsZHogsDogsLogsr   r   re  rd   ZyellowgreenZgoldZlightskyblueZ
lightcoralr   r   r   r   )r  r
  r  %1.1f%%TrZ  )exploder  r1  autopctshadow
startangle)r   r   pie)r  sizesr1  r  r  r   r   r   r   test_pie_default  s   
r  )pie_linewidth_0r  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  TrZ  r   r   r  r  r1  r  r  r  
wedgepropsrn  )lr/  r    exr/  r  r  r    )r  r  r1  r  r  r  r  r'  )r   r  r  r   r   )r  r  r1  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  TrZ  r   r   r   r   )	r  r  r1  r  r  r  r  r?   rj  zCenter point)r   r  ->Zarc3)
arrowstyleZconnectionstylesquareZ	lightgrey)Zboxstyler   )r)  r+  rJ  r  rn  )r   r  r-  r  r  r  r  r1  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  TrZ  r   r   r  rn  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  TrZ  )r  r  r1  r  r  r  Zcounterclockrn  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  TrZ  r   r   r  )	r  r  r1  r  r  r  r  framer?   rb   )r   r   )rz   r   rn  r  r  r   r   r   test_pie_frame_grid  s(   r  zpie_rotatelabels_true.pngc               
   C   r  )
N)ZHogwartsr  r  r  r  r  r  r  TrZ  )r  r  r1  r  r  r  rotatelabelsrn  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  TrZ  )r  r  r1  r  r  r  Zlabeldistancer  rn  g333333r  )r   r  r  rg   r/   r  r   r   r   test_pie_nolabel_but_legend  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)r  "   re  )zLong name 1zLong name 2zLong name 3r?   r#   rZ  Zanchorr3  rX  )horizontalalignmentverticalalignmentrn  rotation_moder  r   %.2f)r  r  	textpropsr  r  rn  r  r  r   )
r  r   r   r  get_haget_vaget_rotationZget_rotation_modeget_sizer  )r'  r  r
  rB  textsr  Ztxr   r   r   test_pie_textprops  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_values1  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)r-   	normalizer   rb   g     v@r  F)r   r   r  r&  rg  )r   r   r-   t1t2r   r   r   test_normalize_kwarg_pie8  s   "r  zset_get_ticklabels.pngc                  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 )Nr   rR   zset_x/yticklabelsr   rd   r   rs   rY  r    r<  r
  r   )Z11Z12Z13Z14)r   r   rn   rO   rT   r  
set_xticksr   
set_yticksset_xticklabelsset_yticklabelsr  Z
get_yticksrI  get_yticklabels)r   r   r+   r   r   r   test_set_get_ticklabelsA  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)

    )r   r   r   r
  r  Zbold)Z
fontweight)r   rz   r   r   r   T)rX  r_  r`  N)r   r  r  r  r  rB  r   r   r   test_set_ticks_with_labels[  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 )
Nrd   r   r   r  c                 S   s   g | ]}|qS r   r   r-  r   r   r   r   w  r.  z.test_subsampled_ticklabels.<locals>.<listcomp>Z
bcdefghijkc                 S   r  r   r  r  r   r   r   r   y  r  )rY  r<  ro  r  r  )r   r   rn   rO   rT   r@   	set_ticksr  set_ticklabelsr   get_ticklabels)r   r   r  r   r   r   test_subsampled_ticklabelsq  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 )Nrd   r   r  )rs   rY  r    )r   r   rn   rO   rT   r@   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 )Nr   r   )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   rb   r   Fr  r*   rh   r   r  rZ  r  )r   r   rn   Zsetpr  r  r   r   r   r   test_retain_tick_visibility  s   r%  c                  C   st   t  \} }dd }|jt | |g d |dd |jj	  dd |j
 D }|g dks8J d S )	Nc                 S   s   | dkrdS dS )Nr   
unit valuer   r   r   r   r   r   formatter_func  s   z.test_tick_label_update.<locals>.formatter_func)rb   r   r   r   rz   r  r  c                 S   r  r   r  r  r   r   r   r     r  z*test_tick_label_update.<locals>.<listcomp>)r   r   r&  r   r   )r   r   r@   r   r   r  r:  r   r  r   r   )r   r   r'  Z
tick_textsr   r   r   test_tick_label_update  s   r(  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   r   r   r   rw   )r  rd   rf  rz   r   )r   r   r   r   rn   rO   ri   r   )r   r   r  r   r   r   test_o_marker_path_snap  s   
""r*  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 )Nr   rd   r  r   r  r   r   )r   r   皙ɿr   )r+  r   )	r  r$  r'  r   r   rn   r   rp   rq   )r'  rM   rN   r  r  r  r   r  r   rX  r   r   r   r   test_margins  sF   








r,  c                  C   sN   t jd t \} }|ddgddg |jddd | dks%J d S )Nr  r   r   r   r   )r  Zxmarginr   )r   r   r   r   r   rn   ro   rp   r   r   r   r   test_set_margin_updates_limits  s
   r-  c                  C   s&   t  \} }|d |dg d S Nr   )r   r   r  r   r   r   r   test_length_one_hist  s   
r/  c                  C   s8   dgd } t dd\}}|| |  |t  d S )Nrd   r   r   )r   r   r  r  r  r  )Zmylistr   r   r   r   r   test_pathological_hexbin  s   
r0  c                  C   s(   t  \} }|jddgddgd d d S )Nr   r   rl  rb  r   r   r   r   test_color_None  s   r1  c                  C   s6   t  \} }|jddgddd }d| ksJ d S )Nr   r   r  r  )r   r   rn   r  )r   r   r0  r   r   r   test_color_alias  s   r2  c                  C   s4   t  \} }|jtdgd tdd |  d S )Nr   r   r|  )r   r   r  r   r/   r   r   r   r   test_numerical_hist_label  s   r3  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   rz   r|  )r   r   decoder  r   r/   )r   r   rs   rY  r  r   r   r   test_unicode_hist_label   s   r5  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 )	Nrd   g^ 9^;r8   r   r   r'   r#   r7   )rO   rP   r   r   rn   r  r   rC   Z
offsetTextr  rB   Z
tick_rightr@   Zget_verticalalignmentrm   )r'  r   r   ZbeforeZafterr   r   r   test_move_offsetlabel  s*   



$



$r6  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)r=   r   r   r   )rL  r   r   r   test_rc_spines(  s   
"r7  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 )NTrh   )rr  zaxes.grid.axisr-   r*   r   rz   )r   r   r  r=   r   r   )r   Zrc_dict0Zrc_dict1Zrc_dict2rQ  r   rL  r   r   r   test_rc_grid3  s"   
r8  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)xtick.bottom	xtick.top
ytick.leftytick.rightr  r   tick1Ontick2Onr   r   r   r   r@   rC   Z_major_tick_kwZ_minor_tick_kwr<  r   r   ZxaxZyaxr   r   r   test_rc_tickI  s"   "rB  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)r:  r<  r9  r;  zxtick.minor.bottomzxtick.major.bottomzytick.major.leftzytick.minor.leftr=  r   r>  r?  r@  rA  r   r   r   test_rc_major_minor_tick]  s&   "rC  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   r7  mor  r   T)originalg      ?r   rw  rw  F)g333333?r   g      ?rw  )rO   rT   r   r   r   rn   r  rp   rq   rX  r  r   r  r  )r-   r*   r   r   rf   rg   r   r   r   test_square_plott  s   

rG  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 )Nr  rI  r   r}  rb   r   )	r{   r|   r  r   rn   rO   rT  rT   rj   r   r   r   r   test_bad_plot_args  s   *"rH  zxy, clsr   ))rz   rw  )r   r
  )r   r   r   r  r+  r  )r   r   r3  )rz   r   c                 C   s0   t  \}}t|jg | |R  |ksJ d S r   )r   r   r  
pcolorfast)r)  r'  clsr   r   r   r   r   test_pcolorfast  s   $rK  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 )Nr   Tr  r   r  r   r  r  )r   r   r  r  r  Z
get_yscaleZ
get_xscaler   r	  r   r   r   r   test_shared_scale  s   

rN  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  r   )r{   r|   r}   r   r   r   r   r   r   test_shared_bool  s   "rO  c                   C   s   t tddg dS )z4Violin plot should handle point mass pdf gracefully.r   N)r   ro  rO   r   r   r   r   r   test_violin_point_mass  s   rP  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 )
Nr-   r   r*   errr   )r   r   r   r   r   )r   r   r  r  )r   rO   rT   ri   )Zbase_xyZ
err_cyclerZxerr_cyZyerr_cyemptyZ	xerr_onlyZ	yerr_onlyZboth_errr   r   r   generate_errorbar_inputs  s*   $

rS  r=  c                 C   s$   t  }|jdi | }|  d S )Nr   )r   r   ro  r  )r=  r   Zebr   r   r   test_errorbar_inputs_shotgun  s   rT  Zdash_offsetc                  C   sT   t  \} }tdd}t|}tdddD ]}|j||| |dfddd qd S )	Nr   rd   r   r   rd   rd   r   r  )r  r   r   )r   r   rO   r   Z	ones_liker   rn   )r   r   r-   r*   r  r   r   r   test_dash_offset  s   
rV  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  )r  r}  g?r  zaxes.titlepadg      R@)r   r   r  ZtitleOffsetTransZ
get_matrixrM  r=   r   )r   r   rx  r   r   r   test_title_pad  s   


$rX  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 )	Nr?   zaxes.titlelocationrW  r8   r$   r'   rG  Zfail)r   r   r   r  Z	get_titler{   r|   r  r   r   r   r   test_title_location_roundtrip  s   

"rY  z
loglog.pngc                  C   sN   t  \} }tdd}|j||d dd |jddd |jd	dd
d d S )Nr   r  rz   r   r   rf  r   )r  r4  r   rX  )r  r4  rZ  )r   r   rO   rT   r  r  )r   r   r-   r   r   r   test_loglog  s
   rZ  ztest_loglog_nonpos.pngc                  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 )Nrz   r   r  g      rw  r  r   )rP  Zclipr   r   )rF  )r   nonpositiver   r   )r[  )
r   r   rO   rT   r9  r   r  r  r  r  )r   r	  r-   r*   ZmcyZ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 )Nr  r   )r   r   r   r   rU  r  rd   )r  g      #@)r   r   rn   Z
get_yboundr  r  Z
get_xboundr  rO   rT  rI  r  rT   rl   r   r   r   r   test_axes_margins&  s0   r]  r-   r*   )r  c                 C   s    dd }dd }||d| j  S )Nc                 S   <   |   }|  | dd | j  }|d dksJ d S Nr   r   rb   r4  )r   r  r:  r@   get_major_locatorr   r   rk  r   r   r   	_helper_xJ  
   z&shared_axis_remover.<locals>._helper_xc                 S   r^  r_  )r   r  r;  rC   r`  ra  r   r   r   	_helper_yQ  rc  z&shared_axis_remover.<locals>._helper_yr   )param)requestrb  rd  r   r   r   shared_axis_removerH  s   rg  )r   r   subplots_sharedr   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 )
Nr   r   rh  r   r  rL  r   r   r   r   rD  rD  )re  r   r   r   r   r   )rf  r   r   rh  r   r   r   shared_axes_generator[  s   
	


rj  c                 C   s   | \}}|| d S r   r   )rj  rg  r   r   r   r   r   test_remove_shared_axesl  s   rk  c                  C   s`   t jddddd\} }|d d }|d d  }|  |dd t|d d  | d S )Nr   r  rL  r   r   r   )r   r   rp   r  r:  r   )r   rh  r   Z	orig_xlimr   r   r   test_remove_shared_axes_relimr  s   rl  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 )NirZ  r  r   r   TrL  r  )rm  r  )rO   rT   rP   Zrandom_sampler  r   r   r:  r;  r  r   r   rp   rq   )r  r.  r   r   r   r   r   r   test_shared_axes_autoscale{  s   $(rn  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 )NrQ  r  r   r  rn  r   )r   r   rC   r`  Z
set_paramsr:  r  r  r   r  r;  r   r   r   r   test_adjust_numtick_aspect  s   


ro  zauto_numticks.pngc                   C   s   t dd d S )Nr   )r   r   r   r   r   r   test_auto_numticks  s   rp  zauto_numticks_log.pngc                  C   s.   t  \} }dtjd< |ddgddg d S )NZround_numbersaxes.autolimit_modeg#B;g     j@gؗҜ<rd   )r   r   r=   r   r  r   r   r   r   test_auto_numticks_log  s   
rr  c                  C   s   t  \} }|g d d S )N)r   r   )r   r   r  r   r   r   r   test_broken_barh_empty  r  rs  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  r  r   r   r  r   r  r   r   gUUUUUU?N)	r   r   r\  r  r:  r  Zverticesr^  r_  )r   r   Zd0ppr   r   r   test_broken_barh_timedelta  s
    (rv  c                 C   sD   | j ddd}td}tjdd}t \}}|||| d S )Nz
2000-01-01rd   )r  r  r  r  )r  rO   rT   rP   r   r   r   r  )r  timedepthr'  r   r   r   r   r   test_pandas_pcolormesh  s
   
ry  c                 C   sp   t jdddd}t t tt|}| ||d}t }|t |j	d dk 
 }|jdd	|d
 d S )Nz2005-02z2005-03rV  r  )r  r  r   r   r  r  rt  )rO   rT   r   r   r   r  r  r   r   r  rj  rn   )r  r  r  r  r   Zwithout_zero_indexr   r   r   test_pandas_indexing_dates  s   rz  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   rh  )r-   r*   xeyer^  )columnsr  r-   r*   r{  r|  r  )r  rO   rP   uniformr   r   ro  r  r   r   r   test_pandas_errorbar_indexing  s   r  c                 C   s:   |  g dg dd}t \}}||j|d  d S )Nr  )rw  r   r   )ZXXYYr  )r  r   r   rn   r  r  r   r   r   test_pandas_index_shape  s   r  c                 C   s8   | j g dd}|jdd  }t \}}|| d S )N)
r   r   r   rz   rz   r   r   r   r   r   rt  r   )r  Zilocr   r   r  )r  Zser_1Z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 )Nr   rs  r   rs   rY  r?   r  )r  r   r   r   r  r%   r  r   r  r   r   r   test_pandas_bar_align_center  s   r  c                  C   s   dd l m}  t }| |d}tjtdd |d W d    n1 s'w   Y  | 	|d}tjtdd |d W d    d S 1 sJw   Y  d S )Nr   z;The apply_tickdir function was deprecated in Matplotlib 3.5re  r  )
Zmatplotlib.axisr  r   r  ZXTickr{   rm  r   Zapply_tickdirZYTick)Zmaxisr   r  r   r   r   #test_tick_apply_tickdir_deprecation  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   rX  r  )r  r  r  r   r  r  r  )
r   r   rC   set_tick_paramsr  _size	tick1liner  label1r  )Z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 )NrY  r   r   r1  )r  r  r  r  r   rP  )r   r   r  r@   rC   r  r  r  Zget_linewidthZ	get_alphaget_linestyle)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 )NFTr  )Z	labelleftZ
labelrightrZ  rX  r   )	r   r   r  rC   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 )NTFr  )labeltoplabelbottomrZ  rX  r   )	r   r   r  r@   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 )NrZ   rj  rh  )r   r   r   rn   r   r  )r   lnr   r   r   test_none_kwargs3  s   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 )Nr  r  )r   rz   r   r   r  r  r   )r   r  rO   r   Zuint8r9  r   r)  )r  rY  r   r-   r   r   r   test_bar_uint89  s
    r  zdate_timezone_x.pngr   c                  C   sj   dd t dD } tjdd tddd tj| dgd dd	 tddd tj| dgd d
d	 d S )Nc              
   S   (   g | ]}t j d dd|tjddqS   r      Canada/Easternr   Ztzinfor\  r  tzZgettzr   r   r   r   r   C  
    

z(test_date_timezone_x.<locals>.<listcomp>rz   r  r3  r  r   r   r  )r  UTCr   r   r   r   rz  Z
time_indexr   r   r   test_date_timezone_x@  s   r  zdate_timezone_y.pngc                  C   sr   dd t dD } tjdd tddd tjdgd | dd	d
d tddd tjdgd | dd	d
d d S )Nc              
   S   r  r  r  r   r   r   r   r   T  r  z(test_date_timezone_y.<locals>.<listcomp>rz   r  r  r   r   r  FT)r  Zxdateydater  r  r  r   r   r   test_date_timezone_yQ  s   r  zdate_timezone_x_and_y.pngc                     sn   t jj  fddtdD } tjdd tddd tj| | dd	d
 tddd tj| | dd	d
 d S )Nc              	      s    g | ]}t j d dd| dqS )r  r   r  r  r\  r   r  r   r   r   g  s    z.test_date_timezone_x_and_y.<locals>.<listcomp>rz   r  r  r   r   r  T)r  r  z
US/Eastern)r\  timezoneZutcr   r   r   r   rz  r  r   r  r   test_date_timezone_x_and_yc  s   
r  zaxisbelow.pngc                  C   s   t  jdddd} d}t| |D ]9\}}|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 || qd S )Nrz   T)r  r  r   )Fr0  Tr  rd   rx  r%  r(  rk  rl  r    rh  )r   ry   r   F)r#   r7   r8   r'   )r   r   r   r9  rn   r/  rk  r.  r  r  r  r  Zset_axisbelow)r	  Zsettingsr   ZsettingZcircr   r   r   test_axisbelowt  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   r)  )rM  Xlabel2Titlerz   )r   r   r   r   r   r   r  r  r   get_rendererget_window_extentr  r
  rO   Zisclose)	r   r   r   Zxlabel2r  rendererZbbox_y0_titleZbbox_y1_xlabel2Z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 )NrD  )r#   ZXlabelr  r  )r   g(\?)r   r   rE  r   r   r  r  r  r  r  Z_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 topr   p=
ף?)r   r   r   r   r@   rr   r  r  r   r  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  r-   Tr  r7   r#   r  r  r  r   r  )
r   r   r   r   r  r  r  r   r  r  r   r   r   r   test_title_xticks_top_both  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  rD  r   r-   Tr  Boor   r   )
r   r   r   r   r   r  r  r  r   r  )r   r   ttr   r   r   test_title_no_move_off_page  s   


r  c                  C   sB   t  \} }|g d |jjdd |j  dksJ d S )Ng   @Ag   eAg   FArX  )r  )r   r   rn   rC   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)Zlabel1Onlabel2On)r  )r   r   rn   rC   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 )Nr   r   g-q=gqىE)r   r   rn   rO   r   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&  rw  r   r  r  r  r   r   r   r     r   z#test_barb_units.<locals>.<listcomp>r   rN  rd   r   r  )r   r   r   rO   r   r  Zbarbsr   r   r  r*   uvr   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   rN  rd   r   r  )r   r   r   rO   r   r  Z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   rz   )	r  to_rgbr   r   r   rn   r  r  r  )r  r   r   r  r  Z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 )Nr   r   r   rh   r   )rZ  rn  rZ  rY  r-   5   )r  r  r*   r   )r  rn  r  )
r   r   rn   r@   r  rC   rI  r  r  r  )rw  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 )	Nrz   zC{}r
  r  )r   Z
facecolorsr   r   r  )start)r   r   r   r  r  r  r  tupler  Zsqueezer  r  r  )	r   r   r  ZcctargetZalZedge_targetelZface_targetr   r   r   test_fillbetween_cycle  s(   
 
 r  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'  rq  r  rd   r   )
r   r   r   Zset_xmarginr  rp   r@   r  r   r   )r   r   ZmarginZxlim0Zxlim1r   Zxlim0tZxlim1tZx0tZx1tr  r   r   r   test_log_margins'  s   


r  c                  C   s   d} t | t | }}t | d }t \}}tt |j|||d W d    n1 s2w   Y  d}|j|||d |j|||g|  d d S )Nr   r   r  )r   r   r   )rO   rT   r   r   r{   r|   r  r   )r  r-   r*   r1  r   r   Zc_rgbr   r   r   test_color_length_mismatch5  s   r  c                   C   s   t jdgdd t   d S )Nr   ZLabelr|  )r   r%  r/   r   r   r   r   test_eventplot_legendA  s   r  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   r   r   )r8   ro  )r   r   r   r  r  )r   r   r  r   r  r  )r   r   Zrect1rect2r   r   r   test_bar_broadcast_argsF  s
   (r  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 )Nr   r   )
r   rn   r{   r|   r  rf   rO   r  rF  rg   r   r   r   r   test_invalid_axis_limitsQ  s   "r  r  r  r   r  c                 C   s6   t  }|g d ||  || |  d S )Nr  )r   r   rn   r  r  r  )r  r  r   r   r   r   test_minorticks_on^  s
   

r  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 )Nr   r   r   r   )r   r   r  r   rn   r   r  Z	intervalx)r   r   Zxtwinr  r   r   r   r   test_twinx_knows_limitsh  s   r  c                   C   s   t jddgddgddd d S )Nr   r   ri  )r  r   )r   rn   r   r   r   r   test_zero_linewidthw  s   r  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   ro  r/   r   r   r   r   test_empty_errorbar_legend|  s   r  c                 C   s\   t ddd}dd |D }dd |D }dd |D }|  || | || d S )Nr  rd   r   c                 S   s0   g | ]}d |d  d|d   d|  d qS )g@rz   r  r   g\(\@      @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  )rO   rT   r   rn   )r3   r4   r  r  r-   r*   r   r   r   test_plot_decimal  s   r  c                 C   s0   |   jddddd |  jddddd d S )Nr   rw   r   r   )r   r  r   )r   rn   )r3   r4   r   r   r   test_markerfacecolor_none_alpha  s   r  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  zxtick.directionzytick.directionoffN)	r   r   r   get_tightbboxr  r  r  r  r  r   r   bbbb2r   r   r   test_tick_padding_tightbbox  s   


r  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\}
}|du sxJ |j  t ddgd	dgg}t |
  |ksJ dS )z:
    Ensure that inset_ax argument is indeed optional
    r  r  r   r   rd   Nrb   r   r   333333@r   r  r   )boundsr   )rO   mgridr  r   r   r   r   r  r  apply_aspectindicate_insetr  r   r   r  get_bbox
get_points)dxdyr*   r-   rM  r   r   rf   rg   r  rec
connectorsxxr   r   r   
test_inset  s&   
*
,

r  c                  C   sz  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|	dksJ |j  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 )Nr  r   r   rd   rb   r   )r  r  ffffff?r  r   r  r   r  r   r   g=
ףp=?gKXc'?g(\?g?g-C6?Zrtol)rO   r  r  r   r   r   r   r  r  r  
inset_axesr:  r;  r  Zindicate_inset_zoomr  r  r   r   r  r  r  r   r   r  )r  r  r*   r-   rM  r   r   Zaxin1r  r  r  r   r   r   test_zoom_inset  s6   
*




r  
x_inverted
y_invertedc                 C   s   t dd\}\}}td}|||d | r|  |r"|  |g d|\}}|\}}	}
}| r6dnd}|r<dnd}||
jd |jd   dksNJ ||jd |	jd   dks^J ||	jd |jd   dksnJ ||jd |
jd   dks~J dS )	zR
    Test that the inset lines are correctly located with inverted data axes.
    r   r   rd   rw   )r   r   r   r   rb   r   N)	r   r   rO   rT   rn   r  r  r  r"  )r  r  r   r   r   r-   r  r  Z
lower_leftZ
upper_leftZlower_rightZupper_rightZsign_xZsign_yr   r   r   test_indicate_inset_inverted  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  rh   rY  r   r   rE  r4  Zactive)r   r   r  r  rO   rN  r  r4  r   r   r   r   test_set_position  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)r
  r
  r!  r!  r  Fgư>r  )r   r   r  r7   r  r  r  Z_set_view_from_bboxrO   r   r   r  r  r   r   r   !test_spines_properbbox_after_zoom  s   r  c                  C   sH   t  \} }t|  |tjdd | j }tj	|
|jd d S )Nr   r  i  )r   r   r   r  rO   rS  r  r  r   r   r  r4  )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 )Nrd   r  r  rh  r  )r   r   r   r   r   r   r   r   )r   r  rO   rP   r~  r   r   )r  r  r-   r*   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   r   r   r   r   r   test_scatter_empty_data"  s   r  zannotate_across_transforms.pngc               	   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   rd   r  )gQ@rz   r  )r  r   r   r   r   Fr   rq  r  r  )r  )r)  r*  r+  r,  rJ  )rO   r   r  r   r   r   rn   r  r  r@   r  rC   r-  Z	transDatar   r  )r-   r*   r   r   Zaxinsr   r   r   test_annotate_across_transforms(  s   

r   zsecondary_xy.pngc                  C   s   t jddddd\} }dd }t|D ]A\}}|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 qd S )Nr   r   rd   r   T)r  rl  c                 S   8   t jdd d|  W  d    S 1 sw   Y  d S Nr  r  r   rO   r  r  r   r   r   invert=     $z!test_secondary_xy.<locals>.invertr  r   r   Z	functionsr  c                 S   s   d|  S Nr   r   r  r   r   r   r+  I      z#test_secondary_xy.<locals>.<lambda>c                 S   s   | d S r  r   r  r   r   r   r+  I  r	  r  c                 S      | d S r  r   r  r   r   r   r+  J  r	  c                 S   r
  )Nr   r   r  r   r   r   r+  J  r	  rD  )r   r   r  rn   rO   rT   secondary_xaxissecondary_yaxis)r   r	  r  nnr   secaxr   r   r   test_secondary_xy9  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    d S 1 sbw   Y  d S )	Nr   r  r   c                 S   s   d|  S r.  r   r  r   r   r   r+  R  r	  z%test_secondary_fail.<locals>.<lambda>r  r'   r7   )
r   r   rn   rO   rT   r{   r|   r  r  r  r   r   r   r   test_secondary_failN  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  r   r  c                 S   r  r  r  r  r   r   r   r  ]  r  z%test_secondary_resize.<locals>.invertr#   r  )rw  r   rF  )r   r   rn   rO   rT   r  r  r   rL  r   r  r  )r   r   r  r   r   r   test_secondary_resizeY  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  r   r  c                 S   r  r  r  r  r   r   r   r  k  r  z'test_secondary_minorloc.<locals>.invertr#   r  r   r  )r   r   rn   rO   rT   r  r   Z_axisZget_minor_locatorr   ZNullLocatorr  ZAutoMinorLocatorr  r   Z
LogLocator)r   r   r  r  r   r   r   test_secondary_minorlocg  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  r@   r   r   ZScalarFormatterr  r   r   r   )r   r   r  r   r   r   test_secondary_formatter  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
    r  r*   r   Nrz   )r4  ro  r   r   r   r   r,  )r7   r#   r8   r'   r   r   r  r   rX  r   )r  r   r  r  r@   rC   r  r   r0  r  r  r4  ro  r   r  r  )r   r   r  bbaxisr  Zaxxr  Zaxisrbbspinesrs   Zspinerr  bbaxr  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
  rM  r  )g     ^@g8R@g     @g     @@)g8`U@gX@g     J@g     @r   )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   r   r  r  r  Bboxfrom_boundsr   r  r  ZtransformedtransFigurer  )r   r   r  r  r  r  r  r  rY  targetbbZaxbbr   r   r   test_normal_axes  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  ZxticklabelsZyticklabelsr  r"  r#  r&  r!  Zatolr'  )r        ^@r   r%  )r   r   r   r  r   ro   r  r  r  r(  r)  r   r  )
r   r   r  r  r  r  r  rY  r  r+  r   r   r   test_nodecorator  s,   
r0  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  r1  r   g     @)r   r   r   ro   r  r7   r  r  r   r  r9  r  r(  r)  r   r  )
r   r   r  r  r  r  targetsr  Zbbspiner+  r   r   r   test_displaced_spine  s*   
r3  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   r5  )g88b@r  r!  r%  )r  r  Zinoutr  Fr  r  r  r  r   r   r!  r.  N)r  r   r   r   r  r  r   r  r  r(  r)  r   r  )r2  Zdnumdirsr   r   r  r  r  r  r  r   r+  r   r   r   test_tickdirs  s2   
r7  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  rh   r  )rZ  r  )r  g88[@r   r5  )gCiq\a@r$  r4  r%  r   r!  r.  rX  r   r$  )r  gPUUUUUB@r   gVUUUUT@)gz6P@r/  goTT@r%  )r   r   r   r  r   r  r  r   r  r(  r)  r   r  r  )	r   r   r  r  r  r  r2  r  r+  r   r   r   test_minor_accountedfor-  s>   


"r8  c                 C   sR   |  dd | dd |  d}|d |d | dd d S )Nr   Fr  r   TZon)r   r  rB  r   r   r   test_axis_bool_argumentsL  s   


r9  c                  C   sr   t  \} }d}d}d}d}|||||g}t|||||fks#J ||f| ks-J ||f| ks7J d S )Nr   rd   r   r  )r   r   r  r  rp   rq   )r   r   rM   rN   r  r  r  r   r   r   test_axis_extent_argX  s   r:  c                  C   s~   t dd tddD } t g d}t j|d}t \}}|| | t	t 
d}| d| d	| fks=J d S )
Nc                 S   s   g | ]	}t  d d|qS )r&  r   r  r  r   r   r   r   k  rk  z(test_datetime_masked.<locals>.<listcomp>r   r
  r^  r   rx  g    H&Ag    d&A)rO   r   r   r   r  r   r   rn   r^  r_  ry  rp   )r-   r*   rx  r   r   rT  r   r   r   test_datetime_maskedh  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 )	Nra  )rz   r   r   r
  rQ  r*  r   r   rb   r
  )r   r  rB  r  r   r   r   test_hist_auto_binsu  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 )Nr   ra  r  r  )r   r   rO   r  r  r  r   r   )
r   r   r   r'  Znan_datar  rL  rB  ZnanbinsZ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 )	Nrd   rQ  rH  T)r   r!  r   rb   r   )r   r  rO   rP   r   r<  r   r   r   test_hist_range_and_density  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 )Nra  r   rz   )r-   ro  r  r  r,  )r   r   r  ro  rA  r   r,  )
r   r   r-   ZbarcontZ	data_lineZcaplinesZbarlinecolsr  ZcaplineZ
barlinecolr   r   r   test_bar_errbar_zorder  s   
r@  c                  C   s6   t  \} }|  |ddg | dksJ d S )Nr   r  r  )r   r   r  r  rp   r   r   r   r   test_set_ticks_inverted  s   rA  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 )NrU  r  r  r  r   r  logit)F]tE?g'^P?r   r  r  rf   r  rg   re   r  r   r   )r   r   r   rn   ro   r   r*  Ztransform_bboxr  ro  r4  r{   approx)r   r   r   r   r   r   $test_aspect_nonlinear_adjustable_box  s   
rF  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 )NrU  r  ri  r  r  r   )r   r   rB  )gbeF?rC  r   r  rD  r   gS[:XL	@g'In?@)r   r   r   rn   ro   r   r  rp   r{   rE  rq   r   r   r   r   (test_aspect_nonlinear_adjustable_datalim  s   
rG  c                  C   s   t  \} }| }|ddg |d t  \}}|d |ddgddg |jdd	d
 | j  |j  |	 }|	 }|	 }t
|j|j t
|j|j d S )Nr3  iX  r   r   r   r
  r  rn  r  r  )r   r   r   rn   set_box_aspectr   r  r  r   r  r   r  )r  r   Zaxtwinr  r   bb1Zbbtr  r   r   r   test_box_aspect  s   



rJ  c                  C   s   t  \} }|g d | j  |d t  \}}|d |j  |g d | j  |j  | }| }t|j|j d S )N)r   r   rw  r   r   )	r   r   r  r  r   rH  r  r   r  )r  r   r  r   rI  r  r   r   r   test_box_aspect_custom_position  s   





rK  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 )Nr   rz   r   )r  T)r  rl  r   )r   r   r  r  r   r  r  r  rt  r4  ro  r   )r   r	  r  r  r   r  r   r   r   test_bbox_aspect_axes_init  s   




rL  c                  C   s4   t dd\} }|g d | j  |  d S )Nr   ra  )r   r   rn   r  r   Zredraw_in_framer   r   r   r   test_redraw_in_frame  s   
rM  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  Zinaxesr  r   r   r   r   test_invisible_axes	  s   
rN  c                  C   <   dt jd< t  } | j }|D ]}|j dksJ qd S Nr  zlines.markeredgecolor)r   r   r  r@   get_major_ticksr  get_markeredgecolorr   Zticksr  r   r   r   "test_xtickcolor_is_not_markercolor     

rT  c                  C   rO  rP  )r   r   r  rC   rQ  r  rR  rS  r   r   r   "test_ytickcolor_is_not_markercolor  rU  rV  r  rQ  )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   Zget_autoscaleZ_onr   ZlimZget_)r  r   )rQ  )
r   r   rO   rT   r   r   r   r  r   r   )
r  rQ  r   r   r-   r*   Zget_autoscale_onZset_limZget_limZ	post_autor   r   r   test_unautoscale!  s   

rW  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  r  r   r   r   r   e   )r   rn   rO   r   Zget_pathZ_interpolation_stepsr   )r3   r4   r  r   r   r   )test_polar_interpolation_steps_variable_r5  s
   $
rY  c                  C   s6   t  \} }|dd | j  | dksJ d S )Nr   g&.>)r   g	>)r   r   r  r  r   rq   r   r   r   r   test_autoscale_tiny_sticky=  s   
rZ  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 )Nrl  zxtick.colorr   zxtick.labelcolor)r   r   r  r@   rQ  r  r  r  rS  r   r   r   &test_xtickcolor_is_not_xticklabelcolorE     


r\  c                  C   r[  )Nrl  zytick.colorr   zytick.labelcolor)r   r   r  rC   rQ  r  r  r  rS  r   r   r   &test_ytickcolor_is_not_yticklabelcolorO  r]  r^  r  c                 C   s   g | ]}|d ur|qS r   r   )r   r  r   r   r   r   Y  s    
r   )r  rd   r3  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 )Nzxtick.labelsizezytick.labelsizer-   r*   )r   r   r   r   r  r   r9  r@   rC   rQ  r  r  Z_get_tick_label_size)r  r   r   namer  r  r   r   r   test_relative_ticklabel_sizesY  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 )Nr   r   r  r  r  )r   rz   r   rz   r  r   r   r  )r   r   r   r   r  rp   rO   rN  )r   r   r   rf   r   r   r   test_multiplot_autoscaleg  s   ra  c                  C   sX   t  } | d}| jg d|d}| }| jdd |  | k s*J d S )Nr@  )r  r  r   r   r  r   )r8   )r   r   r   r   r  rE  r  r  )r   r  r   Zinit_posr   r   r   $test_sharing_does_not_link_positionsp  s   
 rb  c                 C   sz  t 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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   r   r   rb   r  rz   rd   rl  r   r}  )
rO   r   r   rn   rj   r   r  r  rT   r  )r3   r4   r   r	  r   r   r   test_2dcolor_ploty  s   
"""&
"""*rc  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  r   r!  TrL  )rO   rT   r   r   r   r  rn   r  )r3   r4   r-   r*   r	  r   r   r   r   test_shared_axes_clear  s   


rd  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 )	Nr   r  rL  r   zo-r  )r  r   r   r   )r  r   r   r  )r   r   r  rn   r  rp   r  rq   rM  r   r   r   test_shared_axes_retick  s   


re  r+   )r8   r?   r'   c                 C   sD   t  }| }|jdd| d |jd |j  | ks J d S )Ntestr   r)   r'   )r   r   r   rC   Zset_label_positionrA   r  )r+   r   r   r   r   r   test_ylabel_ha_with_position  s
   rg  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 )	Nr   r   rz   r   r?   r7   r#   r   r   r  	bar_labelr)  r  r  r   r  Zheightsr  r  r   r   r    test_bar_label_location_vertical     
rl  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 )	Nr   r   rz   rh  r   r8   r?   r'   r   r   r  rj  r)  r  r  r   r  r4  r  r  r   r   r   "test_bar_label_location_horizontal  rm  rp  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 )Nr   r   rz   rh  r?   )Z
label_typer   rn  ro  r   r   r   test_bar_label_location_center  s   ""rq  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 )
Nr   r   rz   rh  )r  r   r?   r7   r#   ri  rk  r   r   r   !test_bar_label_location_errorbars  s   
""rr  c                  C   sV   t  } | ddgddg}| j|dd}|d  dksJ |d  d	ks)J d S )
Nr   r   rz   rh  r	  )r   r   z3.00z-4.00r   r   r  rj  r  r   r  r  r   r   r   test_bar_label_fmt  s
   ru  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 )	Nr   r   rz   rh  r_  r`  )r  r   rs  rt  r   r   r   test_bar_label_labels  s
   rv  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 )Nr   rz   r   c                 S   r  r   r  )r   r  r   r   r   r     r  z,test_bar_label_nan_ydata.<locals>.<listcomp>r   1r   rt  r7   )r   r   r  rO   r  rj  r)  r  )r   r  r  r   r   r   test_bar_label_nan_ydata  s   
rx  c               	   C   sp   t  \} }|t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   rb   r  rN  r  r   gffffff?r   r  r   gr  r+  )r   r   r.  r/  ZWedgerO   r   r   r   Zassert_array_almost_equal_nulpr   ZdataLimr  )r   r   Zbotr   r   r   test_patch_bounds   s   "ry  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/edgecolorsre  r   r  r!  rk  rY  )r]   r/  r   r   )r{   rm  rn  r   r   r  r   r   r    test_warn_ignored_scatter_kwargs  s   "rz  c                     s  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  tjtdd  jd	= W d    n1 sw   Y  tjtdd  jd	d = W d    n1 sw   Y  tjtdd  jdd = W d    n1 sw   Y  tjtdd  jd= W d    n	1 sw   Y  |   jrJ |   jr'J |   jr1J |   jr;J  jrAJ  jrGJ tjtdd  j|d  W d    n	1 sbw   Y  t j|d gkstJ tjtdd  j|d	  W d    n	1 sw   Y  t j|dd  ksJ tjtdd  jd|d  W d    n	1 sw   Y  t j|d |d |d	 gksJ tjtdd |d  jd< W d    n	1 sw   Y  t j|d |d |d	 gksJ tjtdd |dd  jdd< W d    n	1 s&w   Y  t j|ks5J  jg d g |dddksIJ g d j dddg|ksZJ d S )Nc              	      s&   g | ]}  t||d  d qS )r   r   )rn   rO   rT   r  r  r   r   r     s   & z(test_artist_sublists.<locals>.<listcomp>r
  r   )r   r   r  r   rG  rb   zout of rangere  r   z'modification of the Axes.lines propertyr2  ra  r   rz   )r   r   r   r   rO   rT   rl   rT  r.  r/  r0  r  r  r  ZimagesrA  r   Ztablesr  r{   r|   
IndexErrorr  rm  r   r  r  insert)r   rA  r  r  r   r  r   r  r   test_artist_sublists  s   


&&(&r}  c                  C   sh   t d} t d}t \}}|| |}t|dksJ t \}}|g g }t|dks2J d S )Nrd   )rd   r   r   r   )rO   ri   r   r   rn   r  )r-   r*   rB  r   r0  r   r   r   test_empty_line_plotsa  s   

r~  c                  C   s\   t   } | j| j| j| jfD ]}d}|ddgddgg|dj}|j|j	f|ks+J qd S )N)rw  r  r   r   r   rz   )clim)
r   r   r   rl   r  r  rI  r  r  r  )r   Zplot_methodr  r  r   r   r   	test_climo  s   r  r  (  r  r   r\  r  r   r  	itertoolsr   platformr  r   Zdateutil.tzr  ZnumpyrO   r   r   r{   r=   r   Zmatplotlib.testing.decoratorsr	   r
   r   Zmatplotlib.colorsr1  r  Zmatplotlib.datesr  r^  Zmatplotlib.figurer   Zmatplotlib.font_managerZfont_managerZmfont_managerZmatplotlib.markersr:  ru  Zmatplotlib.patchesr   r/  Zmatplotlib.pyplotZpyplotr   Zmatplotlib.tickerZtickerr   Zmatplotlib.transformsZ
transformsr  Znumpy.testingr   r   r   r   Zmatplotlib.cbookr   r   r6   r<   r>   rD   rY   rv   r~   r   r   r   r   r   r   r  r  r   r   r   r  r  r  r   r  r  r  r  r  r1  r@  rC  rF  rL  rd  rp  ru  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rT  rT   r  r  r  r  r	  r  r  r  r)  r2  rN  rO  rU  rZ  r]  r^  r`  rc  rj  rs  ry  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/  r1  ZPatchr,  rA  rD  r3  rG  rJ  rR  rS  rV  rX  rZ  r\  r]  rk  rp  rr  rv  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)  r*  r+  r,  r.  r0  r2  r5  r7  r9  r;  r<  rF  rH  rJ  rK  rM  rR  rV  r[  r]  r^  ra  rb  re  rp  rs  ru  rv  ry  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  r  r  r  r	  r  r  r  r  r  r  r  r   r#  r$  r+  r,  r0  r4  r6  r7  r8  r;  r=  r>  machiner@  rC  rG  rK  rM  rN  rZ  r_  rb  rc  r  rT  rd  ri  rn  ro  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r?  rV   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  r2  r3  r5  r6  r7  r8  rB  rC  rG  rH  ZimageZ	AxesImager   ZPcolorImagerP   ZQuadMeshrj   r   rK  rN  rO  rP  rS  rT  rV  rX  rY  rZ  r\  r]  Zfixturerg  rj  rk  rl  rn  ro  rp  rr  rs  rv  ry  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  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r  r  r  r  r,  r0  r3  r7  r8  r9  r:  r;  r=  r>  r?  r@  rA  rF  rG  rJ  rK  rL  rM  rN  rT  rV  rW  rY  rZ  r\  r^  Zfont_scalingsr`  ra  rb  rc  rd  re  rg  rl  rp  rq  rr  ru  rv  rx  ry  rz  r}  r~  r  r   r   r   r   <module>   s   











'
*

2


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

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



















!


!




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



+


  *	;

		
	
>


	

	
	
	


	
	
	
	
	
	


				

	
	2
.
+
	
	


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

,




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



!
	
.'





	
	
	
	O