a
    Lf                     @   sV   d Z ddlZddlm  mZ ddlmZm	Z	m
Z
mZ G dd dZG dd dZdS )zTests for polyutils module.

    N)assert_almost_equalassert_raisesassert_equalassert_c                   @   sD   e Z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 )TestMiscc                 C   s8   dg}t dD ]$}tdgdg|  }t|| qd S )N      r   )rangeputrimseqr   )selftgtZnum_trailing_zerosres r   ^/var/www/ai-form-bot/venv/lib/python3.9/site-packages/numpy/polynomial/tests/test_polyutils.pytest_trimseq   s    zTestMisc.test_trimseqc                 C   s.   g t jg t jdfD ]}tt|| qd S )N)dtype)nparrayZint32r   r
   r   )r   Z	empty_seqr   r   r   test_trimseq_empty_input   s    z!TestMisc.test_trimseq_empty_inputc                 C   s   t ttjg g t ttjddggg t ttjdgdgg g d}tt|D ]l}t|D ]^}td|| }td|| }t||g\}}t|j	j
|j	j
k t|j	j
|| k q\qPd S )Nr      a)idO)r   
ValueErrorr
   Z	as_seriesr	   lenr   Zonesr   r   char)r   typesr   jcicjZresiZresjr   r   r   test_as_series   s    zTestMisc.test_as_seriesc                 C   sb   g d}t ttj|d tt||d d  tt|d|d d  tt|ddg d S )N)r   r   r   r#   r   r   r   )r   r   r
   Ztrimcoefr   )r   Zcoefr   r   r   test_trimcoef&   s
    zTestMisc.test_trimcoefc                 C   s>   t ttjdddg t ttjdddg t ttjddg  d S )Nr   r   r      Z   gV@)r   r   r
   Z
_vander_ndr   r   r   r   test_vander_nd_exception/   s    z!TestMisc.test_vander_nd_exceptionc                 C   s   t ttjtjddg d S )Nr&   r   )r   ZeroDivisionErrorr
   _divr)   r   r   r   test_div_zerodiv7   s    zTestMisc.test_div_zerodivc                 C   s   t ttjdg ddd d S )Nr   r&   r      )r   r   r
   Z_powr)   r   r   r   test_pow_too_large;   s    zTestMisc.test_pow_too_largeN)
__name__
__module____qualname__r   r   r"   r%   r*   r-   r/   r   r   r   r   r      s   	r   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )
TestDomainc                 C   sL   g d}ddg}t |}t|| g d}ddg}t |}t|| d S )N)r   
   r'   r#   r#   r4   )y      ?      ?      ?      r   r   y                    @      ?)r
   	getdomainr   )r   xr   r   r   r   r   test_getdomainA   s    


zTestDomain.test_getdomainc                 C   s   ddg}ddg}|}t |||}t|| ddg}ddg}|}|}t |||}t|| ddg}ddg}t||g}t||g}t |||}t|| G d	d
 d
tj}ddg}ddg}t||g|}t |||}tt|| d S )Nr   r.   r   r'                 r6   r   c                   @   s   e Zd ZdS )z,TestDomain.test_mapdomain.<locals>.MyNDArrayN)r0   r1   r2   r   r   r   r   	MyNDArrayg   s   r<   )	r
   Z	mapdomainr   r   r   Zndarrayviewr   
isinstance)r   dom1dom2r   r   r8   r<   r   r   r   test_mapdomainN   s.    


zTestDomain.test_mapdomainc                 C   s`   ddg}ddg}ddg}t ||}t|| ddg}dd	g}d
dg}t ||}t|| d S )Nr   r.   r   r'   g      ?r:   r6   r;   r   y            ?r5   )r
   Zmapparmsr   )r   r?   r@   r   r   r   r   r   test_mapparmsp   s    
zTestDomain.test_mapparmsN)r0   r1   r2   r9   rA   rB   r   r   r   r   r3   ?   s   "r3   )__doc__numpyr   Znumpy.polynomial.polyutilsZ
polynomialZ	polyutilsr
   Znumpy.testingr   r   r   r   r   r3   r   r   r   r   <module>   s
   4