o
    ú»Îh7  ã                   @   sÐ   d dl mZ d dlmZmZ d dl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mZm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S )é    )Úsymbols)ÚcosÚsin)ÚQQÚZZ)ÚPoly)ÚNotInvertible)ÚFiniteExtension)ÚDomainMatrix)Úraises)ÚxÚyÚtc                     sª  t ttd d tƒƒ‰ ˆ jdksJ ‚tˆ ƒdksJ ‚ˆ j‰ˆ ¡ ˆ u s%J ‚ˆˆ ˆ dƒks/J ‚tt‡fdd„ƒ ˆ j	ˆ j
ˆfksBJ ‚ˆ dƒˆ j
ksKJ ‚ˆd ˆ dƒksUJ ‚ˆd dks]J ‚dˆ dˆ  dˆ kskJ ‚dˆ d ˆ d	ƒkswJ ‚ˆ dƒ ¡ ˆ dƒksƒJ ‚tt‡ fd
d„ƒ t ttd t d tddƒ‰ˆjdks¢J ‚ˆj} ˆj	ˆdƒ| | d fks³J ‚| d | ks»J ‚| d ˆdƒksÅJ ‚| d ˆdƒksÏJ ‚| d | d ksÙJ ‚| d | d ksãJ ‚| d | d |  d ksñJ ‚ˆtd t ƒ ¡ d|  ksJ ‚ˆtd ƒd ˆtd ƒ ¡ ksJ ‚| d | d  ˆdƒks$J ‚| d d| d  d  | d d ks9J ‚| d d| d  d  | d d ksNJ ‚d| d d  | d |  d ksaJ ‚| d d d | d  d kssJ ‚tt‡fdd„ƒ t ttd td  t d tddƒ}|jdks—J ‚t|ƒdks J ‚|j}dtd td  t d  }|t |t   ¡ ||ƒksÂJ ‚|t |t  | |dƒksÓJ ‚d S )Né   é   zZZ[x]/(x**2 + 1)éÿÿÿÿc                      s   ˆ d S )N© r   r   )Úir   úl/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/sympy/polys/agca/tests/test_extensions.pyÚ<lambda>   ó    z&test_FiniteExtension.<locals>.<lambda>é   é   é   c                      ó   ˆ dƒ  ¡ S )Nr   ©Úinverser   )ÚAr   r   r       ó    )Úmodulusé   é   é   é	   é   é   iíÿÿÿc                      r   )Nr   r   r   )ÚFr   r   r   4   r   T)ÚfieldzZZ(x)[t]/(t**2 - x**3 - x + 1))r	   r   r   ZrankÚstrÚ	generatorÚparentr   Ú	TypeErrorZbasisÚoner   ÚNotImplementedErrorr   r   )ÚaÚKr   Úcr   )r   r&   r   r   Útest_FiniteExtension   sP   "&**&$&"&r1   c                  C   s’   t td d ttd} t td d ttd}t| ƒ}t|ƒ}|tt td d ƒƒks,J ‚|tt td d ƒƒks:J ‚t||t| ƒhƒdksGJ ‚d S )Nr   ©Údomain)r   r   r   r   r	   Úlen)Úp1Zp2ÚK1ÚK2r   r   r   Útest_FiniteExtension_eq_hash@   s   r8   c                     s<  t ttd d ttdƒ} | tƒ‰ ˆ d d d | jksJ ‚dˆ d d  | jks+J ‚ˆ d d ˆ d  ˆ d ks;J ‚ˆ d d ˆ d  ˆ d ksKJ ‚ˆ d d ˆ d  | jksZJ ‚tt‡ fdd„ƒ tt‡ fdd„ƒ tt‡ fdd„ƒ t ttd d ttdƒ} | tƒ‰ ˆ d d d | jks“J ‚tt	‡ fd	d„ƒ d S )
Nr   r   r2   r   c                      s   ˆ d d d S )Nr   r   r   r   r   ©Úxfr   r   r   T   s    z*test_FiniteExtension_mod.<locals>.<lambda>c                      s   ˆ g  S ©Nr   r   r9   r   r   r   U   r   c                      s   g ˆ  S r;   r   r   r9   r   r   r   V   r   c                      s   ˆ d d ˆ d  S )Nr   r   r   r   r9   r   r   r   \   s    )
r	   r   r   r   Úzeror   ÚZeroDivisionErrorr+   r   r-   ©r/   r   r9   r   Útest_FiniteExtension_modK   s     r?   c                  C   sJ   t ttd d ttdƒ} | tƒ}|  t¡|ksJ ‚|  |¡tks#J ‚d S )Nr   r   r2   )r	   r   r   r   Z
from_sympyÚto_sympy©r/   r:   r   r   r   Útest_FiniteExtension_from_sympy_   s   rB   c                  C   sJ   t ttd d tddƒ} t ttd d tddƒ}|  t¡|ks#J ‚d S )Nr   r   r   r2   r   )r	   r   r   Z
set_domainr   )ÚKZÚKQr   r   r   Útest_FiniteExtension_set_domaing   s   rE   c                  C   sD   t ttd d ƒƒ} | tƒ}|  |d d |d ¡|d ks J ‚d S )Né   r   r   )r	   r   r   ZexquorA   r   r   r   Útest_FiniteExtension_exquom   s   (rG   c                  C   sŽ   t ttd d ƒƒ} tt }| tƒ|tƒ}}|  |¡|ks J ‚| |¡|ks)J ‚t ttd d tdƒ}| tddƒt¡|jd ksEJ ‚d S ©Nr   r   r2   )r	   r   r   r   ÚconvertZconvert_fromr,   )r6   r7   Úx1Zx2r/   r   r   r   Útest_FiniteExtension_convertt   s   $rK   c                     s²  t ttd d ttdƒ} t ttd d ttdƒ}t ttd d ttt dƒ}t ttd d tt t¡dƒ}| jdu s@J ‚|jdu sGJ ‚|jdu sNJ ‚|jdu sUJ ‚| |||fD ]{‰ ˆ  t¡‰ˆˆ j	 ˆkskJ ‚ˆˆ j	 ˆkstJ ‚ˆˆ j	 ˆ j
ks~J ‚tt‡ ‡fdd„ƒ tt‡ ‡fdd„ƒ tt‡ ‡fd	d„ƒ ˆ jr»ˆˆ ˆ j	ks¨J ‚ˆˆ ˆ j	ks±J ‚ˆˆ ˆ j
ksºJ ‚q[tt‡fd
d„ƒ tt‡fdd„ƒ tt‡fdd„ƒ q[d S )Nr   r   r2   TFc                      s
   ˆˆ j  S r;   ©r<   r   ©r/   ÚxKr   r   r      ó   
 z4test_FiniteExtension_division_ring.<locals>.<lambda>c                      s
   ˆˆ j  S r;   rL   r   rM   r   r   r      rO   c                      s
   ˆˆ j  S r;   rL   r   rM   r   r   r   ‘   rO   c                      s   ˆ ˆ  S r;   r   r   ©rN   r   r   r   —   r   c                      s   ˆ ˆ  S r;   r   r   rP   r   r   r   ˜   r   c                      s   ˆ ˆ  S r;   r   r   rP   r   r   r   ™   r   )r	   r   r   r   r   r   Z
frac_fieldZis_FieldrI   r,   r<   r   r=   r-   )rD   rC   ZKQtZKQtfr   rM   r   Ú"test_FiniteExtension_division_ring€   s0    
ñrQ   c                  C   s  t ttd d ƒƒ} ttt| d}|j| ksJ ‚| ¡ tks J ‚|d  ¡ dks*J ‚t ttd d ttdƒ} t ttd d t| dƒ}t|ƒdksLJ ‚| 	tt ¡}| 
|¡tt ks^J ‚| 
|d ¡ddt t  ksoJ ‚ttt t|d}|d tddt t  t|dks‹J ‚d S )Nr   r2   zQQ[x]/(x**2 - 2)[t]/(t**2 - 2)rF   )r	   r   r   r   r3   Zas_exprr   r   r(   rI   r@   )r/   Úpr7   ZeKr   r   r   Útest_FiniteExtension_Polyœ   s   "*rS   c                     s  t dƒ\} ‰‰tˆƒtˆƒ | tˆƒ tˆƒ |  tˆƒ tˆƒ gtˆƒtˆƒ | tˆƒ tˆƒ | tˆƒ tˆƒ gtˆƒ|  tˆƒ dgg}‡‡fdd„}|t|  ƒ}|tƒ|  }||fD ],‰ ‡ fdd„|D ƒ}t|dˆ ƒ}| ¡ d ˆ j d	  }|ˆ  | d
 tˆƒ ¡ksŒJ ‚q`d S )Nzrho, phi, thetar   c                    sp   t ttˆ ƒd tˆ ƒd  d tˆ ƒ| tˆ ƒ dƒ} t ttˆƒd tˆƒd  d tˆƒ| tˆƒ dƒ} | S rH   )r	   r   r   r   r>   )rR   r   r   r   Úmake_extension¸   s   66z<test_FiniteExtension_sincos_jacobian.<locals>.make_extensionc                    s   g | ]}‡ fd d„|D ƒ‘qS )c                    s   g | ]}ˆ   |¡‘qS r   )rI   )Ú.0Úer>   r   r   Ú
<listcomp>Á   s    zCtest_FiniteExtension_sincos_jacobian.<locals>.<listcomp>.<listcomp>r   )rU   Úrowr>   r   r   rW   Á   s    z8test_FiniteExtension_sincos_jacobian.<locals>.<listcomp>)r   r   r   r   r   )r   r   r   r   r
   Zcharpolyr,   rI   )ÚrÚelementsrT   ZKsc1ZKsc2Z
elements_KÚJZdetr   )r/   rR   r   r   Ú$test_FiniteExtension_sincos_jacobian®   s   64ý ür\   N) Zsympy.core.symbolr   Z(sympy.functions.elementary.trigonometricr   r   Zsympy.polysr   r   Zsympy.polys.polytoolsr   Zsympy.polys.polyerrorsr   Zsympy.polys.agca.extensionsr	   Zsympy.polys.domainmatrixr
   Zsympy.testing.pytestr   Z	sympy.abcr   r   r   r1   r8   r?   rB   rE   rG   rK   rQ   rS   r\   r   r   r   r   Ú<module>   s&    2