
    ^j5                     h   d 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m	Z	 ddl
Z
ddlZddl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mZ d
 Zd Zd Z  edgddd      d        Z!d Z"d Z#d Z$d Z%ejL                  jO                  d      d        Z( edgdd ejR                         dk(  rdnd      d        Z* e       d        Z+d Z,d Z- ed gdd ejR                         dk(  rdnd!      d"        Z. ed#gdd$      d%        Z/d& Z0 ed'gd()      d*        Z1d+ Z2 e       d,        Z3e
jh                  jk                  d-d.d/g       e       d0               Z6d1 Z7d2 Z8 ed3gdd$      d4        Z9 e       d5        Z:d6 Z; e       d7        Z< e       d8        Z=d9 Z>d: Z?y);z%
Tests specific to the lines module.
    N)SimpleNamespace)cycler)assert_array_equal)_path)MarkerStyle)Path)image_comparisoncheck_figures_equalc            	          d\  } }t        j                  ddg      t        j                  ddg      }}d}t        t        j                  | ||||      dg       y)zTest a problematic case.)i)  i  g     H@g     W@g     @g(\@r   N)nparrayr   mlinessegment_hits)cxcyxyradiuss        f/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/matplotlib/tests/test_lines.pytest_segment_hitsr      sR    FB88T4L!288S$K#8qAFv**2r1a@1#F    c                      t        j                         \  } }t        j                  j	                  d       |j                  t        j                  j                  dd      dg       y Nr         r   )   r   )
linestylespltsubplotsr   randomseedcontourrandnfigaxs     r   test_set_line_coll_dashr)   !   sD    llnGCIINN1 JJryyr2&K=JAr   c                  L   t        j                  t        d      5  t        j                  dg        d d d        t        j                  t        d      5  t        j                  g d       d d d        t        j                  g g       } t        j                  t        d      5  | j                  d       d d d        t        j                  t        d      5  | j                  d       d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   WxY w# 1 sw Y   y xY w)Nzxdata must bematchr   zydata must be   z	x must bez	y must be)pytestraisesRuntimeErrorr   Line2D	set_xdata	set_ydatalines    r   test_invalid_line_datar6   )   s    	|?	; a	|?	; b! ==R D	|;	7 q	|;	7 q     s/   C6D&DD6C?DDD#line_dashesT_classic_testg~jth?)remove_textstyletolc                  l    t        j                         \  } }|j                  t        d      dd       y )N
   r      	linestylelw)r    r!   plotranger&   s     r   test_line_dashesrD   6   s)     llnGCGGE"IG3r   c                  ~   t        j                         \  } }|j                  t        d      d       |j                  t        d      d       |j                  t        d      d       |j                  t        d      d       |j                  t        d      d       | j                  j                          y )Nr=   none)colorrz.3)r-   r   r   r-   )r-   r   r   r    r!   rB   rC   canvasdrawr&   s     r   test_line_colorsrL   ?   s    llnGCGGE"IVG$GGE"ISG!GGE"ITG"GGE"I\G*GGE"IYG'JJOOr   c                      t        j                  g g       } t        j                  t              5  | j                  d       d d d        y # 1 sw Y   y xY wNfoobar)r   r1   r.   r/   
ValueError	set_colorr4   s    r   test_valid_colorsrR   I   s?    ==R D	z	" !x ! ! !   AAc                      t        j                         \  } }ddddddddd	d
dg ffD ]  }|j                  t        d      |         | j                  j                          y )N-solid--dashedz-.dashdot:dotted)r   N)r    r   r=   )r@   rI   )r'   r(   lss      r   test_linestyle_variantsr^   O   se    llnGCGT8YX'Ar7 ) 	b	R(	)
 JJOOr   c                      t        j                  g g       } t        j                  t              5  | j                  d       d d d        y # 1 sw Y   y xY w)Naardvark)r   r1   r.   r/   rP   set_linestyler4   s    r   test_valid_linestylesrb   Y   sA    ==R D	z	" ':&' ' 'rS   mpl20c                      t        j                         \  } }|j                  ddgddgdd       | j                          y )Nr   r-   rW   )r]   rA   )r    r!   rB   draw_without_renderingr&   s     r   /test_zero_linewidth_dashed_uses_solid_gc_dashesrf   _   s9    llnGCGGQFQFtG* r   zdrawstyle_variants.pngx86_64gQ?c                      t        j                  d      \  } }g d}t        |j                  |      D ]4  \  }}|j	                  t        d      |       |j                  dd       6 y )N   )defaultz	steps-mid	steps-prez
steps-poststepsN  	drawstyle)r      )xlimylim)r    r!   zipflatrB   rC   set)r'   axsdssr(   dss        r   test_drawstyle_variantsry   f   s^     ||AHC
LC
 chh$ )B
dr*
F()r   c                 "   | j                         }t        j                  d      }|j                  |dz   |       |j                         }t	        j
                         j                  dd      }|j                  ||||j                  z          y )Nrm   g     @@        )	transform)add_subplotr   arangerB   mtransformsAffine2D	translate	transData)fig_reffig_testr(   r   ts        r   test_no_subslice_with_transformr   t   sr    				B
		$AGGAHa				B((5AGGAqAbllNG+r   c                      t        j                  g g       } t        j                  t              5  | j                  d       d d d        y # 1 sw Y   y xY wrN   )r   r1   r.   r/   rP   set_drawstyler4   s    r   test_valid_drawstylesr      sA    ==R D	z	" %8$% % %rS   c                     t        j                  ddt         j                  z  d      } t        j                  |       }t	        j
                         \  }}|j                  | |      \  }|j                  d       t        |j                         j                        dt        |       z  dz
  k(  sJ |j                  d       t        |j                         j                        t        |       k(  sJ y )Nr   rp   r=   rk   r-   rj   )r   linspacepisinr    r!   rB   r   lenget_pathvertices)r   r   r'   r(   r5   s        r   test_set_drawstyler      s    
Aqw#A
q	AllnGCGGAqMED{#t}}''(Ac!fHQJ666y!t}}''(CF222r   line_collection_dashesg?c                      t        j                         \  } }t        j                  j	                  d       |j                  t        j                  j                  dd      dg       y r   r   r&   s     r   test_set_line_coll_dash_imager      sD     llnGCIINN1JJryyr2&K=JAr   zmarker_fill_styles.png)r9   r:   c                  .   t        j                  g ddddddt        j                  g d      g      } d}t        j                  d	d	g      }t        j                  d
dg      }t	        j
                         \  }}t        d      D ]i  \  }}t        t        j                  j                        D ]>  \  }}	t        |       }
|j                  |dz  |z   ||z   d|dz  z  z   |d||	|	d|
|
d       @ k |j                  d
d       |j                  dd       y )Nr   r   r-   gz#ff0000cmr   )r   r   r   
lightgreenr-   r   	   zov^<>8sp*hHDdPXr=         ?rp   r   r>   )	marker
markersizemarkerfacecoloralt	fillstylelabel	linewidthrG   markeredgecolormarkeredgewidthg      @   )	itertoolscycler   r   r    r!   	enumerater   r1   
fillStylesnextrB   set_ylimset_xlim)colorsaltcolorr   r   r'   r(   jr   ifsrG   s              r   test_marker_fill_stylesr      s   __iic3 hhy13 4FH
!QA
!QAllnGC 01 '	6v}}778 	'EArLEGGAFQJAa!e 4!!'/ $)$%  	'	'' KK3KKCr   c                      t        j                  g dt        dd      d      \  } | j                         dk(  sJ | j	                         dk(  sJ y)z=Test that markerfacecolor does not override fillstyle='none'.r-   r   rp   orF   )r   red)r   markerfacecolorN)r    rB   r   get_fillstyleget_markerfacecolor)ls    r   test_markerfacecolor_fillstyler      sL    	)Kv$F"'
)BA??&&&  "f,,,r   scaled_linesrj   )r:   c            	      |   t        j                  dd      } t        j                         \  }}g d}t	        t
        j                  d         }t        t        ||            D ]W  \  }\  }}t        j                  ddd      D ]5  } |j                  | |t        j                  d      z  d|z  z   f||d	| 7 Y y )
Nr       )rX   r[   rY   axes.prop_cycler   r=   2   皙?r?   )r   r   r    r!   r   
matplotlibrcParamsr   rs   rB   ones)	thr'   r(   lins_stylesr   r   r]   styrA   s	            r   test_lw_scalingr      s    	Q	BllnGC1K	
##$56	7B!#k2"67 M9B++b"b) 	MBBGGB"''"+R/L2"LL	MMr   c                  .   t        j                  t        j                  g d            sJ t        j                  t        j                  dt        j                  dg            sJ t        j                  ddgt        j                  gdz  z   ddgz         rJ t        j                  t        j                  dd	gd
            rJ dt
        j                  j                  z  } t        j                  t        j                  g| z  t        |              y )Nr-   rp   r   r-   r   r>   d   r   rp   i   i   z>i4)r   is_sorted_and_has_non_nanr   r   nanr   r1   _subslice_optim_min_sizer    rB   rC   )ns    r   test_is_sorted_and_has_non_nanr      s    **288I+>???**288QN+CDDD..1v3/F!Q/OPPP..rxx58I5/QRRR	FMM222AHHbffX\58$r   c                     | j                         j                  ddgd       |j                         j                  g dg ddddg       y )Nr   r-   z-or   )r   r-   r-   rp   	markevery)r!   steprB   r   r   s     r   test_step_markersr      sC    aVT*Iy$1a&Ir   parentfigureaxesc                     t         j                  j                  d       t        j                  ddd      }t         j                  j	                  t        |            }d ddg dddgt        d	d
d      t        j                  t        |            |dkD     ddg	}g d}|dk(  r|d d }|d d } fd}fd}nY|dk(  rTt         j                  dd      j                        t        j                  dd      j                        fd}fd}|D ]  }	 |||	        |D ]Q  }	t        j                  t        |	            j                  t              j                  t              }
 |||
       S y )N*   r   r-         )rp   r>   )r-   r>      r>   r=   rp   r   333333?)r   皙?)	111111111111111000100010001000100001000010010001000001001000000000000100000101010000011100011101101101101101111001010011011101r   c                V    j                  t        j                  | |d|             y Nr   r   r   
add_artistr   r1   )r   r   r   r   s      r   add_testz test_markevery.<locals>.add_test   s$    a3)DFr   c                V    j                  t        j                  | |d|             y r   r   )r   r   r   r   s      r   add_refztest_markevery.<locals>.add_ref   s$    a3)DFr   r   r   c                B    t              j                  | |d|       y Nz-gDr   r   rB   )r   r   r   axs_tests      r   r   z test_markevery.<locals>.add_test   s    N1eyAr   c                B    t              j                  | |d|       y r   r   )r   r   r   axs_refs      r   r   ztest_markevery.<locals>.add_ref   s    Mq!Ui@r   r   )r   r"   r#   r   randr   slicer~   iterr!   rt   r   listastypeintbool)r   r   r   r   r   
cases_test	cases_refr   r   casemer   r   s   ``         @@r   test_markeveryr     sb    IINN2
Aq"A
		s1vA6:b'5B?))CF#AG,z#JGI _
crN		F	F 
6	))!Q/445w''1-223	B	A  'A&'  $XXd4j!((-44T:1#$r   c                     t        j                         } | j                  t        j                  ddgddgdd             t        j                  t              5  | j                  j                          d d d        y # 1 sw Y   y xY w)Nr   r-   r   r   r   )
r    r   r   r   r1   r.   r/   rP   rJ   rK   )r'   s    r   .test_markevery_figure_line_unsupported_relsizer    s_    
**,CNN6==!Q!QrJK	z	" 

  s   A>>Bc                     t        j                         \  } }|j                  g dt        d            \  }| j                  j                          |j                         dk(  sJ |j                  d       | j                  j                          |j                  t        d             | j                  j                          t        j                  ddgddgdd	gg      }|j                  g d
t        |      d      \  }|j                  g d|d      \  }t        |j                         j                  |j                         t        |j                         j                  |j                         y )N)rp   r   r   D)r   sr   r   r-   r   rp   r      )r   ms)r   rp   r-   )r    r!   rB   r   rJ   rK   
get_marker
set_markerr   _create_closedr   r   )r'   r(   r5   	triangle1line2line3s         r   test_marker_as_markerstyler    s   llnGCGGIk#&6G7EDJJOO??### 	OOCJJOOOOK$%JJOO##b"X2wA$?@IWWY{9'="WEFEWWYyRW8FEu'')22I4F4FGu'')22I4F4FGr   zstriped_line.pngc           	      2   t         j                  j                  d      }t        j                         \  }}|j                  |j                  d      ddddd	       |j                  |j                  d      d
ddddd       |j                  d       y )Ni!N,   )sizeorangebluerW   r>   zblue in orange)rG   gapcolorr@   rA   r   r   black)r   )rp   r>   r   rp   zblack in redr   )rG   r  r@   rA   r   alpha)handlelength)r   r"   default_rngr    r!   rB   uniformlegend)text_placeholdersrng_r(   s       r   test_striped_linesr$  &  s    
))


)CLLNEArGGCKKRK 6q(8  :GGCKKRK 'A^3  PII1Ir   c                     | j                         j                  ddgg d       |j                         j                  ddgg d       y )Nr-   rp   r   )dashes)r-   rp   r   r-   rp   r   )r}   rB   r   s     r   test_odd_dashesr'  1  sC    Ay91v.@Ar   c                  n   t        j                         \  } }t        | j                  j                  dz  | j                  j
                  dz  dz         }|j                  ddgddgd      \  }|j                  |      \  }}|rJ |j                  ddgddgdd	      \  }|j                  |      \  }}|sJ t        |d
   dg       |j                  ddgddgd      \  }|j                  |      \  }}|rJ |j                  d       |j                  |      \  }}|sJ t        |d
   dg       y )Nrp      )r   r   r   r-   T)pickerr   )r*  
pickradiusind)
r    r!   r   bboxwidthheightrB   containsr   set_pickradius)r'   r(   mouse_eventl0foundindicesl1l2s           r   test_pickingr8  7  s5   llnGC!CHHNNa$7$'HHOOq$82$=?K ''1a&1a&'
.CB[[-NE79 ''1a&1a&"'
=CB[[-NE7L5wu~s+ ''1a&1a&'
.CB[[-NE79b[[-NE7L5wu~s+r   c                    t        j                  ddd      }| j                         j                  ||d      \  }t	        d      |d d  |j                  d       |j                         j                  g dg ddd       y )	Nr   ri   rp   z.-r   rl   )r   rp   r   rn   )r   r~   r}   rB   rC   r   )r   r   r   r   s       r   test_input_copyr:  Q  sn     			!QA					$	$Q4	0BA8AaDOOGy)TWMr   c                    dddg dddgt        ddd	      d
ddddg}t        j                  d   } |t        j                  ddt        |                  }t        j                  dd      }d|dz  z  }|j                         }t        |      D ]   \  }}	|j                  ||z
  d|	||          " t        ||      t        j                  d<   | j                         }
t        |      D ]  \  }}|
j                  ||z
  d        y)z*Test that we can set markevery prop_cycle.N   )r   r<  )      r   r   r   r      r   r   r   g      ?)r{   r   )g?r   jet皙?皙?r-   r>   rp   zo-)r   rG   r   )r   mpl	colormapsr   r   r   r}   r   rB   r   r   r   )r   r   casescmapr   r   r   rv   r   r   r(   r#  s               r   test_markevery_prop_cyclerG  \  s    1g|aW3Qc3&E ==D"++c3E
34F
BA	AqDA



C!%( D9Q	CD .4e:@.BJ)* 
			B%  1
Atr   c                     t        j                         \  } }|j                  dd      }|j                  dd      }|j                  d       |j	                  d       |j                  d       |j                  d       |j                         dk(  sJ |j                         dk(  sJ |j                         dk(  sJ |j                         dk(  sJ t        j                  t        j                        5  |j                  d	d
       d d d        t        j                  t        j                        5  |j                  dd       d d d        t        j                  t        d      5  |j                  d	d
       d d d        t        j                  t        d      5  |j	                  d       d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   WxY w# 1 sw Y   y xY w)N)r   r   333333?slope)rB  r   )rA  r   g333333@)r   rA  )rI  rB  rA  r   rB  z.Cannot set an 'xy2' value while 'slope' is setr+   z-Cannot set a 'slope' value while 'xy2' is setr   )r    r!   axlineset_xy1	set_slopeset_xy2get_xy1	get_slopeget_xy2r.   warnsrC  MatplotlibDeprecationWarningr/   rP   )r'   r(   line1r  s       r   test_axline_settersrV  u  s   llnGCIIhcI*EIIh)E 
MM(	OOC	MM(	MM( ==?h&&&??###==?h&&&==?h&&&	c66	7 b"	c66	7 b" 
zM
O b" 
zL
N      s0   8F=6G	,G"G!=G	GG!G*c                      t        j                  dd      } | j                         j                  d      }| j                         j                  d      }|d   |d   z
  }|dkD  sJ |dk  sJ y)	z@Test that small slopes are not coerced to zero in the transform.)r   r   g+=rJ  )r-   r-   r-   r   g-=N)r    rL  get_transformtransform_point)r5   p1p2dys       r   test_axline_small_sloper]    sl    ::fE*D					-	-f	5B					-	-f	5B	AAB6M6::r   )@__doc__r   platformtypesr   r   numpyr   numpy.testingr   r.   r   rC  r   matplotlib.lineslinesr   matplotlib.markersr   matplotlib.pathr   matplotlib.pyplotpyplotr    matplotlib.transforms
transformsr   matplotlib.testing.decoratorsr	   r
   r   r)   r6   rD   rL   rR   r^   rb   r:   contextrf   machinery   r   r   r   r   r   r   r   r   r   markparametrizer  r  r  r$  r'  r8  r:  rG  rV  r]  r\   r   r   <module>rp     sb     !   ,     ! *    + OGB
 =/t?PUV4 W4!' 7! ! +,$o+8++-9atE	)E	) , ,%
3 +,$g+8++-9atEBEB +,$oV W>- >")4M 5M% J J
 Hf#56)$  7)$XH( %&DH I B B
,4 N N  0>r   