
    -i                     8   S SK rS SKrS SKJrJr  \" SS \" SS5      \" SS5      5      r\" SS \" SS\" SS5      \" SS\" SS5      \" SS	5      5      5      \" SS\" SS
5      \" SS5      5      5      r\R                  R                  S\S4\S4/5      S 5       r
g)    N)Treebuchheim                         ztree, n_nodes	   c                    ^ U4S jm[        U 5      nT" U5      n[        U5      U:X  d   eSn U Vs/ s H  oUS   U:X  d  M  US   PM     nnU(       d  g [        [        R                  " U5      5      [        U5      :X  d   eUS-  nM\  s  snf )Nc                   > U R                   U R                  4/nU R                   H9  nUR                  U R                  S-   :X  d   eUR                  T" U5      5        M;     [	        U R                  5      (       aF  U R                   U R                  S   R                   U R                  S   R                   -   S-  :X  d   eU$ )Nr   r   r   )xychildrenextendlen)	draw_treereschild	walk_trees      [/var/www/html/venv/lib/python3.13/site-packages/sklearn/tree/tests/test_reingold_tilford.pyr    test_buchheim.<locals>.walk_tree   s    Y[[)*''E77ikkAo---JJy'( ( y!!"" 	 2 21 5 7 7):L:LQ:O:Q:Q QUVVVV
    r   r   )r   r   npunique)treen_nodeslayoutcoordinatesdepthnodex_at_this_depthr   s          @r   test_buchheimr&      s     d^FF#K{w&&& E
/:O{t1g>N747{O299_-.#o2FFFF
 Os   B	B)numpyr   pytestsklearn.tree._reingold_tilfordr   r   simple_treebigger_treemarkparametrizer&    r   r   <module>r/      s      92q$r1+tB{3
	RRDQKb!-	 	QRT"a[)
 K+;k1=M*NO Pr   