
    gY
                     *    d dl mZ d dlmZ d Zd ZdS )    Prufer)raisesc                  ~   t          ddgddgddgddggd          j        dk    sJ t          ddgddgddgddgg          j        dk    sJ t          ddgddgddgddgg          } | j        dk    sJ | j        dk    sJ | j        g dk    sJ t          ddgddgddgddgddgg          } | j        dk    sJ | j        d	k    sJ | j        ddgddgddgddgddggk    sJ | j        g d
k    sJ t          j        g dg dg d          ddgddgddgddgddgdd	ggdfk    sJ t          dgdz            j        t          d	gdz            j        cxk    rdk    sn J g d}d |D             }t          |          j        |k    sJ t          t          t          |                    j                  t          |          k    sJ t          t          d            t          t          d            t          t          j        ddgddg           j        ddgk    sJ t          t          d            t          t          d            t          t          d            t          ddgddgddgg          } |                                 }|j        ddgddgddggk    sJ |j        dk    sJ d S )Nr                  )r   r   r   i     )r
   r   r
   r   )r   r   r   r	   )r   r
   r   )r   r
   r      i  ))r   r   )r   r   )r   r	   )r   r   )r   r   )r
   r   )r   r
   c                 ,    g | ]}t          |          S  )list).0ts     a/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/combinatorics/tests/test_prufer.py
<listcomp>ztest_prufer.<locals>.<listcomp>   s    (((a$q''(((    c                  ,    t          ddgddgg          S )Nr   r   r	   r
   r   r   r   r   <lambda>ztest_prufer.<locals>.<lambda>       v1v1v&677 r   c                  ,    t          ddgddgg          S )Nr   r	   r
   r   r   r   r   r   ztest_prufer.<locals>.<lambda>    r   r   c                  4    t          j        ddgddg          S )Nr   r	   r
   r   edgesr   r   r   r   ztest_prufer.<locals>.<lambda>"   s    v|	
AA    r   c                  4    t          j        ddgddg          S )Nr   r   r   r   r   r   r   r   r   ztest_prufer.<locals>.<lambda>$   s    v|QFQF;; r   c                  "    t          g g          S )Nr   r   r   r   r   ztest_prufer.<locals>.<lambda>%   s    vrd|| r   )r   nodesrankprufer_repr	tree_reprr   sizesortedsetr   
ValueErrornext)atree
tree_listsbs       r   test_pruferr,      so   Aq6Aq6Aq6Aq62A66<AAAAAq6Aq6Aq6Aq62339Q>>>>AAAA/00A6Q;;;;7a<<<<=III%%%%AAAAA788A6S====7a<<<<;Aq6Aq6Aq6Aq6Aq6BBBBB=LLL((((<iii;;a&1a&1a&1a&1a&1a&	91=> > > >1#a%==A!3;;;;t;;;;;; DCCD((4(((J$<<!Z////&T##-..&2D2DDDDD
:77888
:778886<AA//0<AFFFF
:     
:;;<<<
:++,,,AAA'((A	A;Aq6Aq6Aq6222226Q;;;;;;r   c                     d }  | ddggg             | g dgdg            | g dgdg            | g dgdg            | g dddggddg            | g d	gddg            | g d
ddggddg            | g dgddg            | g dgddg            | g dgddg            | g dgddg            | g dgddg            | g dgddg            | g dddggddg            | g dgddg            | g dddggddg            | g dgddg            | g dgddg            | g dgddg            | g dg dddggg d           d S )Nc                 :   t          j        |  \  }}t          ||          } t          | j                  }d |D             }| j        |k    sJ t          t          |          j                  |k    sJ t          j        | j        |          j        |k    sJ d S )Nc                     g | ]}|d z
  S )r   r   )r   is     r   r   z1test_round_trip.<locals>.doit.<locals>.<listcomp>2   s    qQUr   )r   r   r$   r"   r!   unrankr    )r   r+   enr(   s        r   doitztest_round_trip.<locals>.doit.   s    |Q11aLL1;A}!!!!fQii)**a////}QVQ''3q888888r   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   r   r
   r	   )r   r   r	   r
   )r   r   r   r
   )r   r	   r      r   )r   r   r   r	   r	   r   r   )r4   s    r   test_round_tripr6   -   sk   9 9 9 	D1a&2D)))qcD)))qcD)))qcD)))aV	q!f%%%D,,,!Q   D)))aV	q!f%%%D,,,!Q   D,,,!Q   D,,,!Q   D,,,!Q   D,,,!Q   D,,,!Q   D)))aV	q!f%%%D,,,!Q   D)))aV	q!f%%%D,,,!Q   D,,,!Q   D,,,!Q   D,,,q!f	-/A/A/ABBBBBr   N)sympy.combinatorics.pruferr   sympy.testing.pytestr   r,   r6   r   r   r   <module>r9      sW    - - - - - - ' ' ' ' ' '% % %PC C C C Cr   