o
    ho                     @   s   d dl mZmZmZ d dlmZ d dl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 d dlmZ ed\ZZZ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 )    )FloatIntegerRational)symbols)betaEizetaMaxMinsqrt
riemann_xifrac)CXX98CodePrinterCXX11CodePrinterCXX17CodePrintercxxcode)log1pzx y u vc                  C   s|   t  ttddv sJ t  ttdttdksJ t  } | jdks'J | jdks.J d| jv s5J d| jvs<J d S )N   )zstd::max(x, 3)zstd::max(3, x)z&std::min(3, std::min(x, std::sqrt(x)))C++zC++98templatealignas)	r   doprintr	   xr
   r   languagestandardreserved_words)cxx98printer r   S/var/www/html/ai/venv/lib/python3.10/site-packages/sympy/printing/tests/test_cxx.pytest_CXX98CodePrinter   s    r   c                  C   sh   t  ttdksJ t  } | jdksJ | jdksJ d| jv s$J d| jv s+J d| jvs2J d S )Nstd::log1p(x)r   zC++11operatornoexceptconcept)r   r   r   r   r   r   r   )cxx11printerr   r   r   test_CXX11CodePrinter   s   r%   c                  C   s,   G dd dt } |  ttdksJ d S )Nc                   @   s   e Zd Zdd ZdS )z-test_subclass_print_method.<locals>.MyPrinterc                 S   s   dd t| j|j S )Nzmy_library::log1p(%s)z, )joinmap_printargs)selfexprr   r   r   _print_log1p"   s   z:test_subclass_print_method.<locals>.MyPrinter._print_log1pN)__name__
__module____qualname__r,   r   r   r   r   	MyPrinter!   s    r0   my_library::log1p(x)r   r   r   r   )r0   r   r   r   test_subclass_print_method    s   r3   c                  C   sL   G dd dt } t  }|  }|ttdksJ |ttdks$J d S )Nc                   @   s   e Zd ZdZdS )z1test_subclass_print_method__ns.<locals>.MyPrinterzmy_library::N)r-   r.   r/   _nsr   r   r   r   r0   )   s    r0   r    r1   r2   )r0   pmypr   r   r   test_subclass_print_method__ns(   s
   r7   c                   C   s~   t  tttdksJ t  ttdksJ t  ttdks%J t  ttdks1J t  ttdks=J d S )Nzstd::beta(x, y)zstd::expint(x)zstd::riemann_zeta(x)z(x - std::floor(x))z^((1.0/2.0)*std::pow(M_PI, -1.0/2.0*x)*x*(x - 1)*std::tgamma((1.0/2.0)*x)*std::riemann_zeta(x)))	r   r   r   r   yr   r   r   r   r   r   r   r   test_CXX17CodePrinter3   s
   r9   c                   C   s.   t tttd dt ddgksJ d S )N      ?*z0.5zstd::sqrt(x))sortedr   r   r   splitr   r   r   r   test_cxxcode=   s   .r>   c                   C   sH   t tttttttdksJ t tttttttdks"J d S )Nz(std::max(std::min(u, v), std::min(x, y))z(std::min(std::max(u, v), std::max(x, y)))r   r	   r
   r   r8   uvr   r   r   r   test_cxxcode_nested_minmax@   s   rA   c                  C   sj   G dd dt } |  }tdD ]}||d| ksJ q|tddks(J |td dks3J d S )	Nc                       s$   e Zd Z fddZdd Z  ZS )z.test_subclass_Integer_Float.<locals>.MyPrinterc                    s   dt  | S )NzbigInt("%s"))super_print_Integer)r*   arg	__class__r   r   rC   H   s   z=test_subclass_Integer_Float.<locals>.MyPrinter._print_Integerc                 S   s,   t |}d| t|j| t|jf S )NzbigFloat(%s, %s))r   r(   r   r5   q)r*   rD   ratr   r   r   _print_FloatK   s
   z;test_subclass_Integer_Float.<locals>.MyPrinter._print_Float)r-   r.   r/   rC   rI   __classcell__r   r   rE   r   r0   G   s    r0      zbigInt("%d")r:   z"bigFloat(bigInt("1"), bigInt("2"))g      z$bigFloat(bigInt("1"), bigInt("1"))/x)r   ranger   r   r   )r0   r5   ir   r   r   test_subclass_Integer_FloatF   s   rN   N)"sympy.core.numbersr   r   r   sympy.core.symbolr   sympy.functionsr   r   r   r	   r
   r   r   r   sympy.printing.cxxr   r   r   r   sympy.codegen.cfunctionsr   r   r8   r?   r@   r   r%   r3   r7   r9   r>   rA   rN   r   r   r   r   <module>   s    (

