o
    GZh0\                     @   sN  d dl Z d dlZd dlZd dlmZ d dlmZmZmZ d dl	m
Z
mZ d dlmZ d dlmZmZmZmZ d dlmZmZmZmZmZmZmZmZmZmZ d dlmZm Z m!Z!m"Z"m#Z#m$Z$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/m0Z0m1Z1 d dl2m3Z3 d dl4m5Z5 d dl6m7Z8 d dl9m:Z: d dlm;Z; d dlm<Z< d dl=m>Z> e>dZ?ddhZ@ddhZAdhZBg ddfddZCdd ZDd d! ZEd"d# ZFd$d% ZGd&d' ZHd(d) ZId*d+ ZJd,d- ZKd.d/ ZLd0d1 ZMd2d3 ZNd4d5 ZOd6d7 ZPd8d9 ZQd:d; ZRd d<lSmTZT d=d> ZUd d?lVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d@dA Zd dBlmZ d dClmZ d dDlmZmZ d dElmZmZmZmZ d dFlmZmZmZ dGdH Zd dIlmZ dJdK Zd dLlmZ dMdN Zd dOlmZmZ dPdQ Zd dRlmZ dSdT Zd dUlmZ d dVlmZ dWdX ZedYdZ Zed[d\ Zd d]lmZ d d^lmZ d d_lmZ d d`lmZ dadb Zdcdd Zededf Zdgdh Zdidj Zdkdl Zdmdn Zdodp Zdqdr Zdsdt Zd dulmZ d dvlmZmZ d dwlmZ d dxlmZmZ d dylmZ d dzlmZ d{d| Zed}d~ Zedd Zedd Zd dlmZ d dlmZ dd Zd dlmZ d dlmZ dd Zdd Zdd Zdd ZdS )    N)meter)XFAILraisesignore_warnings)AtomBasic)SingletonRegistry)StrDummySymbolWild)
EIpioozoonanIntegerRationalFloatAlgebraicNumber)EqualityGreaterThanLessThan
RelationalStrictGreaterThanStrictLessThan
Unequality)Add)Mul)Pow)
DerivativeFunctionFunctionClassLambdaWildFunction)Interval)	vectorize)gmpy)SymPyDeprecationWarning)S)symbols)import_modulecloudpickleZ_assumptionsZ_mhashZis_EmptySetZexpr_free_symbolsZ_sage_T c           	         s   t  trdD ]tt fdd qdtjtjddg}tr%|tg |D ]U|v r.q'tr=t  t	r8q' }nt
rK }n	tt }t }t|}t|t|ksfJ |siq'fdd}| || || | q'd	S )
z2 Check that pickling and copying round-trips.
    )r      c                      s   t  S N)pickledumpsr.   )aprotocolr.   R/var/www/auris/lib/python3.10/site-packages/sympy/utilities/tests/test_pickling.py<lambda>8   s    zcheck.<locals>.<lambda>         c              	      s   |D ]q}|t v r	q|tv rt| |rt||sJ |q|tv s$| v rGtt t| |t||ks7J |W d    n1 sAw   Y  qt| |sMqt| |}t|dsst||s`J |t|||kssJ dt|||f qd S )N__call__z%s != %s, protocol: %s)dont_check_attrsnot_equal_attrshasattrdeprecated_attrsr   r)   getattr)r3   bdiattr)
deprecatedr4   r.   r5   cS   s(   




&zcheck.<locals>.cN)
isinstancer   r   NotImplementedErrorcopydeepcopyr-   extendcallabletypeinspectismoduleloadsr2   r1   dirset)	r3   exclude
check_attrrD   	protocolsr@   Zd1Zd2rE   r.   )r3   rD   r4   r5   check2   s2   



rU   c                  C   s&   t t  tt ttfD ]} t|  q
d S r0   )r   r   r   r*   rU   rE   r.   r.   r5   test_core_basicm   s   
rW   c                   C   s   t td d S Nx)rU   r	   r.   r.   r.   r5   test_core_Strq   s   rZ   c                  C   s6   t t dddttdddttdfD ]} t|  qd S )NrY   F)ZcommutativeZ_issue_3130)r
   r   r   rU   rV   r.   r.   r5   test_core_symbolt   s
   
r[   c                  C   sL   t dtddtdfD ]} t|  qtttdfD ]} t| dd qd S )Nr7   r8   z1.2FrS   )r   r   r   rU   r   sqrtrV   r.   r.   r5   test_core_numbers}   s
   
r^   c                  C   s   t dd } t|  d S )NrY   g      ?)r   rU   )yr.   r.   r5   test_core_float_copy   s   r`   c                  C   sl   t d} t d}tt| |tt| |tt| |tt| |tt| |tt| |tt| |fD ]}t| q-d S )NrY   r_   )	r   r   r   r   r   r   r   r   rU   )rY   r_   rE   r.   r.   r5   test_core_relational   s   
ra   c                  C   (   t d} tt| dfD ]}t| qd S NrY   r9   )r   r   rU   rY   rE   r.   r.   r5   test_core_add      
re   c                  C   rb   rc   )r   r   rU   rd   r.   r.   r5   test_core_mul   rf   rg   c                  C   rb   rc   )r   r    rU   rd   r.   r.   r5   test_core_power   rf   rh   c                  C   s.   t d} tt| ttttfD ]}t| qd S rX   )r   r!   r"   r#   r$   r%   rU   rY   fr.   r.   r5   test_core_function   s   
rk   c                  C   s   t d} t|  d S )Nrj   )r"   rU   )rj   r.   r.   r5   test_core_undefinedfunctions   s   rl   c                  C   s    t d} td}t||  d S )NZ_long_unique_name_1Z_long_unique_name_2)r   r"   rU   ri   r.   r.   r5   test_core_appliedundef   s   rm   c                  C   s    t t ddfD ]} t|  qd S )Nr   r7   )r&   rU   rV   r.   r.   r5   test_core_interval   s   
rn   c                  C      t t dfD ]} t|  qd S )Nr   )r'   rU   rV   r.   r.   r5   test_core_multidimensional      
rp   c                  C   s   g d} t j t jg}|dd | D 7 }tr|dd g7 }tdtdtdtdd	ttttt t	t
tjtjtjtjtjtjfD ]}|D ]
}|||u sMJ qCq?d S )
N)r   r/   r7   r8   r9   c                    s   g | ]  fd dqS )c                    s   t t |  S r0   )r1   rO   r2   rY   protor.   r5   r6      s    z,test_Singletons.<locals>.<listcomp>.<lambda>r.   ).0r.   rs   r5   
<listcomp>   s    z#test_Singletons.<locals>.<listcomp>c                 S   s   t t | S r0   )r-   rO   r2   rr   r.   r.   r5   r6      s    z!test_Singletons.<locals>.<lambda>r   r/   r7   )rH   rI   r-   r   r   r   r   r   r   r   r   r*   ZGoldenRatioZTribonacciConstantZ
EulerGammaZCatalanZEmptySetZIdentityFunction)rT   Zcopiersobjfuncr.   r.   r5   test_Singletons   s    rz   )	FreeGroupc                   C   s   t tddd d S )Nzx, y, zFr\   )rU   r{   r.   r.   r.   r5   test_free_group   s   r|   );	Piecewise
lowergammaacosh
chebyshevu
chebyshevtlnchebyshevt_rootlegendre	Heaviside	bernoullicothtanhassoc_legendresignargasin
DiracDeltarerfAbs
uppergammabinomialsinhcoscotacosacotgammabellhermiteharmonicLambertWzetalog	factorialasinhacothcoshdirichlet_etaEijkloggammaerfceilingim	fibonacci
tribonacci	conjugatetanchebyshevu_rootflooratanhr]   sinatanfflucasatan2	polygammaexpc                  C   sp  g t ttttttttt	t
ttttttttttttttttttttttt t!t"t#t$t%t&t't(t)R } t*t+t,t-t.t/t0t1t2t3t4f}t5d\}}}t6t7t8|||t9d|dk f|d |dkf|d dft:f}| D ]}t;| ||}t;| q|D ]}t;| |||}t;| q|D ]}t;| qd S )Nx,y,zr   rw   r7   r/   r8   T)<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   rU   )Zone_varZtwo_varrY   r_   zZothersclsrE   r.   r.   r5   test_functions   s   
&



r   )GeometryEntity)Point)CircleEllipse)LineLinearEntityRaySegment)PolygonRegularPolygonTrianglec                  C   s   t dd} t dd}t dd}t dd}tt t | tt| dtt| ddtt| |tt| |tt| |tt| |tt| |||t	t	| ddt
t
| ||fD ]}t|dd qMd S )	Nr/   r7   r8   r   r9      Fr\   )r   r   r   r   r   r   r   r   r   r   r   rU   )p1p2Zp3Zp4rE   r.   r.   r5   test_geometry   s   



r   )Integralc                  C   s&   t d} tt| fD ]}t| q
d S rX   )r   r   rU   rd   r.   r.   r5   test_integrals  s   
r   )Logicc                  C   ro   Nr/   )r   rU   rV   r.   r.   r5   
test_logic  rq   r   )MatrixSparseMatrixc                  C   s>   t t g dttddgddggfD ]
} t| ddgd qd S )	Nr/   r7   r8   r/   r7   r8   r9   Z_smatZ_matrD   )r   r   rU   rV   r.   r.   r5   test_matrices"  s   (r   )Sievec                  C   s   t t  fD ]} t|  qd S r0   )r   rU   rV   r.   r.   r5   test_ntheory*  s   
r   )Pauli)Unitc                  C   s"   t tttdfD ]} t|  qd S r   )r   r   r   rU   rV   r.   r.   r5   test_physics3  s   
r   c                  C   s  ddl m} m} ddlm} ddlm}m} ddlm	}m
}m}m} ddlm}	 ddlm}
 ddlm} dd	lm} dd
lm} ddlm}m}m}m}m}m}m}m} ddl m!} ddl"m#} ddl$m%} | | dd||||||||||	|
||||||||||||||fD ]}t&| qd S )Nr   )ColorGradientColorScheme)ManagedWindow)Plot
ScreenShot)PlotAxesPlotAxesBasePlotAxesFramePlotAxesOrdinate)
PlotCamera)PlotController)	PlotCurve)PlotInterval)PlotMode)Cartesian2DCartesian3DCylindricalParametricCurve2DParametricCurve3DParametricSurfacePolar	Spherical)
PlotObject)PlotSurface)
PlotWindowg?g?)'&sympy.plotting.pygletplot.color_schemer   r   Z(sympy.plotting.pygletplot.managed_windowr   sympy.plotting.plotr   r   #sympy.plotting.pygletplot.plot_axesr   r   r   r   Z%sympy.plotting.pygletplot.plot_camerar   Z)sympy.plotting.pygletplot.plot_controllerr   Z$sympy.plotting.pygletplot.plot_curver   Z'sympy.plotting.pygletplot.plot_intervalr   Z#sympy.plotting.pygletplot.plot_moder   Z$sympy.plotting.pygletplot.plot_modesr   r   r   r   r   r   r   r   Z%sympy.plotting.pygletplot.plot_objectr   Z&sympy.plotting.pygletplot.plot_surfacer   Z%sympy.plotting.pygletplot.plot_windowr   rU   )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   rE   r.   r.   r5   test_plotting;  s.   (



r   c                  C   sN   ddl m}  ddlm} ddlm} t| d t|ddd t|  d S )	Nr   )r   )r   )r   rainbowr/   F)visible)r   r   r   r   r   r   rU   )r   r   r   r.   r.   r5   test_plotting2V  s   r   )ZZ)QQ)lex)Polyc                  C   sN   ddl m}  td}tt||fD ]}t| q| | |fD ]}t| qd S )Nr   )PurePolyrY   )sympy.polys.polytoolsr   r   r   rU   )r   rY   rE   r.   r.   r5   test_pickling_polys_polytoolss  s   

r   c                  C   s   ddl m} m}m} | | tdgtdgtdggtfD ]	}t|dgd q||tdtdgtdtdgftfD ]}t| q<||tdtdgtdtdtdgtfD ]}t| qZd S )Nr   )DMPDMFANPr/   r7   r8   repr   )Zsympy.polys.polyclassesr   r  r  r   rU   r   )r   r  r  rE   r.   r.   r5   test_pickling_polys_polyclasses  s   *.
2
r  c                  C   s^   ddl m}  | dtt}| |fD ]
}t|ddgd q|j|jfD ]}t|ddgdd q!d S )Nr   )PolyRingr   r/   )rR   F)rR   rS   )Zsympy.polys.ringsr  r   r   rU   Zdtypeone)r  ringrE   r.   r.   r5   test_pickling_polys_rings  s   r  c                   C   s   d S r0   r.   r.   r.   r.   r5   test_pickling_polys_fields  s   r	  c                  C   s,   ddl m}  | | ddfD ]}t| qd S )Nr   )PythonRationalr/      )Z"sympy.polys.domains.pythonrationalr
  rU   )r
  rE   r.   r.   r5   test_pickling_polys_elements  s   
r  c                  C   s  ddl m}  ddlm} | |  fD ]}t|dd q|| fD ]}t|dd qtd urTddlm} ddlm	} || fD ]}t|dd q=|| fD ]}t|dd qKddl
m} dd	lm} ||ttd
fD ]}t|dd qi|| fD ]}t|dd qwd S )Nr   )PythonIntegerRing)PythonRationalFieldFr\   )GMPYIntegerRing)GMPYRationalField)AlgebraicField)ExpressionDomainr8   )Z%sympy.polys.domains.pythonintegerringr  Z'sympy.polys.domains.pythonrationalfieldr  rU   _gmpyZ#sympy.polys.domains.gmpyintegerringr  Z%sympy.polys.domains.gmpyrationalfieldr  Z"sympy.polys.domains.algebraicfieldr  Z$sympy.polys.domains.expressiondomainr  r   r]   )r  r  rE   r  r  r  r  r.   r.   r5   test_pickling_polys_domains  s(   

r  c                  C   s   ddl m} m}m}m} | |  fD ]}t| q|| fD ]}t| q|| fD ]}t| q)|||  fD ]}t| q7d S )Nr   )LexOrderGradedLexOrderReversedGradedLexOrderInverseOrder)sympy.polys.orderingsr  r  r  r  rU   )r  r  r  r  rE   r.   r.   r5   test_pickling_polys_orderings   s   


	
r  c                  C   s^   ddl m} m} td\}}}| | dfD ]}t| q||d|||ffD ]}t| q&d S )Nr   )MonomialOpsMonomialr   r8   r   )Zsympy.polys.monomialsr  r  r+   rU   )r  r  rY   r_   r   rE   r.   r.   r5   test_pickling_polys_monomials  s   

r  c                  C   s"  ddl m} m}m}m}m}m}m}m}m	}m
}	m}
m}m}m}m}m}m}m}m} | |  fD ]}t| q/|| fD ]}t| q;|| fD ]}t| qG|| fD ]}t| qS|| fD ]}t| q_|| fD ]}t| qk|| fD ]}t| qw|| fD ]}t| q|| fD ]}t| q|	|	 fD ]}t| q|
|
 fD ]}t| q|| fD ]}t| q|| fD ]}t| q|| fD ]}t| q|| fD ]}t| q|| fD ]}t| q|| fD ]}t| q|| fD ]}t| q|| fD ]}t| qd S )Nr   )HeuristicGCDFailedHomomorphismFailedIsomorphismFailedExtraneousFactorsEvaluationFailedRefinementFailedCoercionFailedNotInvertibleNotReversibleNotAlgebraicDomainErrorPolynomialErrorUnificationFailedGeneratorsErrorGeneratorsNeededUnivariatePolynomialErrorMultivariatePolynomialErrorOptionError	FlagError)Zsympy.polys.polyerrorsr  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  r*  r+  r,  r-  r.  r/  r0  rU   )r  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  r*  r+  r,  r-  r.  r/  r0  rE   r.   r.   r5   test_pickling_polys_errors"  sP   T

















r1  c                  C   sd   ddl m} m} td}|d | d }| | |dfD ]}t| q|||tfD ]}t| q)d S )Nr   )CRootOfRootSumrY   r8   )Zsympy.polys.rootoftoolsr2  r3  r   rU   r   )r2  r3  rY   rj   rE   r.   r.   r5   test_pickling_polys_rootoftools  s   

r4  )LatexPrinter)MathMLContentPrinterMathMLPresentationPrinter)PrettyPrinter)
prettyForm
stringPict)Printer)PythonPrinterc                  C   s8   t t  ttttttdtt tt fD ]} t|  qd S )Nr3   )	r5  r6  r7  r8  r9  r:  r;  r<  rU   rV   r.   r.   r5   test_printing  s   
r=  c                   C      t t  d S r0   )rU   r6  r.   r.   r.   r5   test_printing1     r?  c                   C   r>  r0   )rU   r7  r.   r.   r.   r5   test_printing2  r@  rA  c                   C   r>  r0   )rU   r8  r.   r.   r.   r5   test_printing3  r@  rB  )Limit)Orderc                  C   s:   t d} t d}tt| |dtt| fD ]}t| qd S )NerY   r/   )r   rC  rD  rU   )rE  rY   rE   r.   r.   r5   test_series  s
   
rF  )Product)Sumc                  C   s>   t d} tt| | ddftt| | ddffD ]}t| qd S )NrY   r7   r9   )r   rG  rH  rU   rd   r.   r.   r5   test_concrete  s   (
rI  c                  C   s   t dddd} t|  d S )Nmessagez1.0zactive-deprecations)Zdeprecated_since_versionZactive_deprecations_target)r)   rU   )wr.   r.   r5   test_deprecation_warning  s   rL  c                   C   s    t t tjtjksJ d S r0   )r1   rO   r2   r*   ZHalfr.   r.   r.   r5   test_issue_18438  s    rM  c                  C   s   d} t | tdksJ d S )Nsi   ^       sympy.core.powerPowsympy.core.numbersIntegerKR}bhHalf)R}bR}b.r7   )r1   rO   r]   )datar.   r.   r5   !test_unpickle_from_older_versions  s   rO  )rM   rH   r1   Zsympy.physics.unitsr   Zsympy.testing.pytestr   r   r   Zsympy.core.basicr   r   Zsympy.core.singletonr   Zsympy.core.symbolr	   r
   r   r   Zsympy.core.numbersr   r   r   r   r   r   r   r   r   r   Zsympy.core.relationalr   r   r   r   r   r   r   Zsympy.core.addr   Zsympy.core.mulr   Zsympy.core.powerr    Zsympy.core.functionr!   r"   r#   r$   r%   Zsympy.sets.setsr&   Zsympy.core.multidimensionalr'   Zsympy.external.gmpyr(   r  Zsympy.utilities.exceptionsr)   r*   r+   Zsympy.externalr,   r-   r<   r>   r;   rU   rW   rZ   r[   r^   r`   ra   re   rg   rh   rk   rl   rm   rn   rp   rz   Zsympy.combinatorics.free_groupsr{   r|   Zsympy.functionsr}   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   Zsympy.geometry.entityr   Zsympy.geometry.pointr   Zsympy.geometry.ellipser   r   Zsympy.geometry.liner   r   r   r   Zsympy.geometry.polygonr   r   r   r   Zsympy.integrals.integralsr   r   Zsympy.core.logicr   r   Zsympy.matricesr   r   r   Zsympy.ntheory.generater   r   Zsympy.physics.paulialgebrar   r   r   r   r   Zsympy.polys.domains.integerringr   Z!sympy.polys.domains.rationalfieldr   r  r   r   r   r   r  r  r	  r  r  r  r  r1  r4  Zsympy.printing.latexr5  Zsympy.printing.mathmlr6  r7  Zsympy.printing.pretty.prettyr8  Z sympy.printing.pretty.stringpictr9  r:  Zsympy.printing.printerr;  Zsympy.printing.pythonr<  r=  r?  rA  rB  Zsympy.series.limitsrC  Zsympy.series.orderrD  rF  Zsympy.concrete.productsrG  Zsympy.concrete.summationsrH  rI  rL  rM  rO  r.   r.   r.   r5   <module>   s    0$;	





<
a


