rL i!NddlZddlmZddlZddlmZddlm Z ddl m Z m Z ddl mZmZddlmZddlmZdZedgd d Zed gd ZedgddZdZedgdZdZedgdZedgdZdZdZy)N)Mock)Path) CustomCellTable)image_comparisoncheck_figures_equal)Bboxc8ddg}tj|y)Nbr) cellColours)plttable) cellcolorss a/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/matplotlib/tests/test_table.pytest_non_squarerssJII*%ztable_zorder.pngT) remove_textc >ddgddgg}d}dDcgc]}d|z }}g}tjt|}t|D],}||z }|j |Dcgc] }d|d z z c}.tj d d tj zd }tj|tj|d d tj|||ddtj|||dd tjgycc}wcc}w)NiRiبivi)FreezeWind)d2z%d yearz%1.1fg@@rr)lwzordercenter)cellText rowLabels colLabelslocrz upper center) npzeroslenreversedappendlinspacepirplotcosryticks)datar"xr!r yoffrowts r test_zorderr3s  FO FO D#I(121Q2I2H 88C N #D~>  t Aqw$AHHQq a*IIx!! IIx!!   JJrN13 =s DD ztable_labels.pngcd}tjdd|}tjj |}dg|zg|z}tj }|j ddd}|jd|j||d|j ddd }|jd|j||d g|zd |j ddd}|jd|j||d |j ddd}|jd|j||d g|zd y)Nr1roffbest)r rowColoursr#rHeader)r r:r!r#)r colColoursr#)r r<r"r#) r$r)rcmRdYlGnfigure add_subplotaxisr) dimccoloursr figax1ax2ax3ax4s rtest_label_coloursrJ5s: C Aq#AffmmAG }s"H **,C //!Q "CHHUOIIx  //!Q "CHHUOIIx ! S( //!Q "CHHUOIIx  //!Q "CHHUOIIx ! S(rztable_cell_manipulation.pngmpl20)stylec$d}dgt|zgdz}dgt|z}tjt|dt|dzfd\}}t||D]+\}}|j ||d | |j d -y) N horizontalverticalopenclosedTRBLr7r皙?rr6tight)nrowsfigsizelayoutr) colWidthsr r#edgesr8)r&rsubplotsziprrA)text_placeholderscellsr r\_axsaxcells rtest_diff_cell_tablerfZs LEE "#a'HE "I \\E QE 1 4Eg VFAsUOD #!   rc |d}tjtjtjtjtjftjtjtjtjtjftjtjtjtjtjftjtjtjtjtjftjtjtjtjtjftjtjtjtjtjftjtjtjtjtjftjtjtjtjtjff}t ||D]I\}}t d|dd}t d|jjD}||k(rIJy)NrNrrr6) visible_edgeswidthheightc3&K|] \}}| ywN).0rbss r z"test_customcell..zsC41aQCs) rMOVETOLINETO CLOSEPOLYr_rtupleget_path iter_segments)typescodesr2rCrecodes rtest_customcellr{ksu LE dkk4;; T[[I dkk4;; T[[I dkk4;; T[[I dkk4;; T^^L dkk4;; T[[I dkk4;; T[[I dkk4;; T[[I dkk4;; T[[I EE5!1&1EC4==?#@#@#BCCDyyrztable_auto_column.pngc~tjdd\}\}}}}|jd|jddgddggddgd d gd }|j d |j d|j gd|jd|jddgddggddgd d gd }|j d |j d|j d|jd|jddgddggddgd d gd }|j d |j d|j d|j d|j d|jd|jddgddggddgd d gd }|j d |j dy)Nrr6r8zFit Textrz-very long long text, Longer text than defaultArUCol1Col2r)r r!r"r#F )rr6rr)rr^rArauto_set_font_size set_fontsizeauto_set_column_width) rErFrGrHrItb1tb2tb3tb4s rtest_auto_columnr~s # Q 2C #sCHHUO ))q/BAFH*6"   C 5!Rj)HHUO ))q/BAFH*6"   C 5!Rj)HHUO ))q/BAFH*6"   C 5!Rb!a a HHUO ))q/BAFH*6"   C 5!Rrc*tj\}}t|}|jdddd}t |t sJ||dusJt dddd}||d<|d|usJ|j tj|y)Nr6r)r6rrh)ri)rr6)rr^radd_cell isinstancer propertiessetp)rErdrrecell2s rtest_table_cellsrsllnGC "IE >>!Q1 %D dJ '' ' 5;   vq!4 8EE$K ;%   HHUOrpng) extensionsc :ddgddgg}d}d}|Dcgc]}|Dcgc]}|c}}}}|j}|j|||dgd |j} | j|||dtjd d d d  ycc}wcc}}w)Nrr5r)FooBar)AdaBobr)rW皙?皙?g333333?)r r!r"r#bboxrWrg?r)r^rr from_extents) fig_testfig_refr. col_labels row_labelsr1r/ cell_textax_listax_bboxs rtest_table_bboxrs F F D JJ2673#&QQC&&7I7!G MM9&&+  G MM9&&((c3< '7s B B BBcGdd}tj}td|_td|_td|_|tj |<|d|dg|tjd d d |d gg}|jj||jjd d gd d gg|jj|jjtj j|tj j|rJy)NceZdZdZdZy)!test_table_unit..FakeUnitcyrmrn)selfthings r__init__z*test_table_unit..FakeUnit.__init__s rcy)NHellorn)rs r__repr__z*test_table_unit..FakeUnit.__repr__srN)__name__ __module__ __qualname__rrrnrrFakeUnitrs   rrcy)Nrrn)vuas rz!test_table_unit..r) side_effectcyrmrn)rrs rrz!test_table_unit..rrc*tjSrm)munitsAxisInfo)rrs rrz!test_table_unit..s FOOrs .O&!& %&D9:@%&'!(!H01A B  &*+,3-3l$()4(7)7<T" Yr