
    G/Ph"                     X    d dl Z d dlZd dlmZmZmZmZmZm	Z	 d dl
Z
 G d d          ZdS )    N)assert_assert_equalassert_array_equalassert_array_almost_equalassert_raises_assert_valid_refcountc                       e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd ZdS )TestRegressionc                     t          t          j        dg          t          j        ddg          z
  t          j        ddg                     d S )N   r   )r   nppoly1dselfs    _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numpy/lib/tests/test_regression.pytest_poly1dzTestRegression.test_poly1d   sN    RYs^^biA&7&77YAw''	) 	) 	) 	) 	)    c                     t           j                            d          }|                                }t          j        |d           t          j        |d           t	          ||           d S )N)   r   T)rowvarF)r   randomcopycovr   r   xys      r   test_cov_parametersz"TestRegression.test_cov_parameters   sd    IV$$FFHH
q
q1a     r   c                     t          d          D ]4}t          j        g dddg           t          j        g dddg           5d S )Nd   )r      r      r   r   )r   r   r!   r   r"   )ranger   digitize)r   is     r   test_mem_digitizez TestRegression.test_mem_digitize   sX    s 	1 	1AKq!f---K!Q0000	1 	1r   c                 n    t          g t          j        t          j        g                                d S N)r   r   uniquearrayr   s    r   test_unique_zero_sizedz%TestRegression.test_unique_zero_sized   s*    2ry"6677777r   c                 b   t          j        d           } |t          j        d          t          j        d          t          j        d                      |t          j        d          t          j        d          t          j        d          t          j        d                     d S )Nc                      | S r(    )argss    r   <lambda>z3TestRegression.test_mem_vectorise.<locals>.<lambda>%   s     r   )r   r!   r   )r!   r   r   )r   r   r!   r!   r!   )r   	vectorizezeros)r   vts     r   test_mem_vectorisez!TestRegression.test_mem_vectorise#   s    \,,--
28I 3 3RXi5H5HIII
28I 3 3RX ? 6 68F##	% 	% 	% 	% 	%r   c                     t          t          j        ddd         dg           t          t          j        dd         g            d S )Nr   y              ?)r   r   mgridr   s    r   test_mgrid_single_elementz(TestRegression.test_mgrid_single_element*   sA    28AaF+aS11128AaC="-----r   c                 R    d }t          j        |          }t          |           d S )Nc                     dS )N{   r.   )r   r   s     r   pz1TestRegression.test_refcount_vectorize.<locals>.p1   s    3r   )r   r2   r   )r   r<   vs      r   test_refcount_vectorizez&TestRegression.test_refcount_vectorize/   s2    	 	 	LOOq!!!!!r   c                     t          j        t           j        t           j        dgd          }t          t           j        j        t          |d           d S )Nr   F)rr@   )r   r   nanr   linalgLinAlgErrorgetattr)r   r<   s     r   test_poly1d_nan_rootsz$TestRegression.test_poly1d_nan_roots6   s@    Irvrvq)U333bi+Wa=====r   c                 2    t          j        g dg           d S )Ng      ?)r   polymulr   s    r   test_mem_polymulzTestRegression.test_mem_polymul;   s    

2tr   c                 X    t          j        g           }t          j        |d           d S )Nzasdasd	asdasd)r   r*   append)r   r   s     r   test_mem_string_concatz%TestRegression.test_mem_string_concat?   s(    HRLL
	!%&&&&&r   c                     t          j        g d          }t          j        g d          }t          j        ||          \  }}t          ||z  |z   |           d S )Nr   r!   r   )r   r!   r   r"      )r   r   polydivr   )r   ur=   qr@   s        r   test_poly_divzTestRegression.test_poly_divD   s[    Iiii  Iooo&&z!Q1QqS1Wa     r   c                     t          j        g d          }t          j        ddg          }t          ||k               t          ||k               d S )NrM   r   r"   )r   r   r   r   s      r   test_poly_eqzTestRegression.test_poly_eqK   sJ    Iiii  Iq!fQQr   c                 j    g d}g d}g d}t          j        ||d          }t          ||           d S )N)g1$ͱgHPB?g{p#gpI#@g
K<ls)QZ   [   \   ]   ^   _   `   a   b   c   r    e   f   g   h   i   j   k   l   m   n   o   p   q   r   s   t   u   v   w   x   y   z   r;   |   }   ~                                                                                                                                       )Q      "@      @      @      @r          @      @      &@r   r   r         @r   r   r   r   r         $@r   r   r   r   r   r   g      *@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   polyfitr   )r   refr   r   testeds        r   test_polyfit_buildz!TestRegression.test_polyfit_buildR   s`    0 0 00 0 0: : : Aq!$$!#v.....r   c                 R   d}t          j        dt                    }t          j        ||          \  }}t	          |j        t          k    |           d}t          j        dt                    }t          j        ||          \  }}t	          |j        t          k    |           d S NzWrong type, should be complexr   dtypezWrong type, should be float)r   onescomplexrO   r   r   intfloat)r   msgr   rQ   r@   s        r   test_polydiv_typez TestRegression.test_polydiv_typeg   s    -GAW%%%z!Q17"C(((+GAS!!!z!Q15 #&&&&&r   c                 n    t          t          t          j        t          j        d          d           d S )N)r   
   i   )bins)r   
ValueErrorr   histogramddr   r   s    r   test_histogramdd_too_many_binsz-TestRegression.test_histogramdd_too_many_binsr   s+    j"."''2B2BOOOOOOr   c                 :   d}t          j        dt                    }t          t          j        |          j        t          k    |           d}t          j        dt                    }t          t          j        |          j        t          k    |           d S r   )r   r   r   r   polyintr   r   r   )r   r   r   s      r   test_polyint_typez TestRegression.test_polyint_typev   sz    -GAW%%%
1#w.444+GAS!!!
1#u,c22222r   c                 n    t          t          j        t          j        g g                               d S r(   )listr   ndenumerater*   r   s    r   test_ndenumerate_crashz%TestRegression.test_ndenumerate_crash   s,     	R^BHbTNN++,,,,,r   c                     t          j        t           j                  j        dz  }t	          d|z  dz  dz             fd}fd}t          t          |           t          t          |           d S )N   r!   g?r   c            	          d} t          j        | fdz            }t           j                            d|           }d|t          j        |||||          <   d S Nr   rN   r   )sizer   r   r   randintix_nar%   thesizes      r   dpz4TestRegression.test_large_fancy_indexing.<locals>.dp   sY    AQA	!!!QW!55A'(AbfQ1a##$$$r   c                      d} t          j        | fdz            }t           j                            d|           }|t          j        |||||                    d S r   r   r   s      r   dp2z5TestRegression.test_large_fancy_indexing.<locals>.dp2   sZ    AQA	!!!QW!55AbfQ1a##$$$$r   )r   r   intpitemsizer   r   r   )r   nbitsr   r   r   s       @r   test_large_fancy_indexingz(TestRegression.test_large_fancy_indexing   s    !!*Q.q%x7+A-..	) 	) 	) 	) 	)	% 	% 	% 	% 	% 	j"%%%j#&&&&&r   c                     t          j        ddg          }t          j        d|          }t          t           j        ||f         j        |k               d S )N)r   f4)bi4)r   )r   r   r3   r   r_)r   dtr   s      r   test_void_coercionz!TestRegression.test_void_coercion   sM    X{K011HT2ad!R'(((((r   c                     t          j                    g}|D ]7}t          t          |t                               t          |dk               8d S )N )r   get_includer   
isinstancestr)r   include_dirspaths      r   test_include_dirsz TestRegression.test_include_dirs   sZ     (()  	  	 DJtS))***DBJ	  	 r   c           	      &   t          t          t          j        t          j        dg          d          t          j                             t          t          t          j        dgd          t          j                             t          t          t          j        t          j        dg          d          t          j                             t          t          t          j        dgd          t          j                             d S )Nr   r   )r   r   r   polyderr   ndarrayr   s    r   test_polyder_return_typez'TestRegression.test_polyder_return_type   s    
2:binna88")DDEEE
2:qc1--rz::;;;
2:binna88")DDEEE
2:qc1--rz::;;;;;r   c                 \   ddl m} t          j        g dt          j                  }g d}t          j        d                              t          j                  }t          j        t          j        t          j        g}	  |||||           d S # t          $ r t                      w xY w)Nr   )append_fieldsrM   r   )r   r   cr   )
numpy.lib.recfunctionsr   r   r*   int32eyeastypefloat64	ExceptionAssertionError)r   r   basenamesdatadlists         r   test_append_fields_dtype_listz,TestRegression.test_append_fields_dtype_list   s    888888x			222vayy))RXrx0	#M$tU33333 	# 	# 	# """	#s   B B+c                 X   ddl m} ddg}t          j         |d          |          }t	          |t          j        dg|                     dd	d
gfg}t          j         |d          |          }t	          |t          j        dg|                     dg}t          j         |d          |          }t	          |t          j        dg|                     dg}t          j         |d          |          }dg}t	          |t          j        ||                     d S )Nr   )StringIO)r   u1r!   )r   r  r!   z0 1 2 3r   r   r   r!   r   r   )r   r  )r   r   )r   r  )))r   r   r!   r   )r   r  r1   )r  )r   r  )r!   r   r!   z0 1 2 3 4 5 6 7 8 9 10 11)))r  r  )r"   rN   ))      )r   	   )r      )ior   r   loadtxtr   r*   )r   r   r   r   r   s        r   test_loadtxt_fields_subarraysz,TestRegression.test_loadtxt_fields_subarrays   sP   n-Jxx	**"555Q"2!32>>>???(+678Jxx	**"555Q"3!4B???@@@!"Jxx	**"555Q"5!6bAAABBB$%Jxx ;<<BGGGIJQR00011111r   c                     t          j        dt                    }	 t          j        |           d S # t          $ r t                      w xY w)Nr!   r   )r   r3   boolnansumr   r   )r   r   s     r   test_nansum_with_booleanz'TestRegression.test_nansum_with_boolean   sS    HQd###	#IaLLLLL 	# 	# 	# """	#s	   3 Ac                 (    G d d          }t          t          j        d          }	 t          j         |            |           n# t
          $ r t                      w xY w	 |                                 d S # |                                 w xY w)Nc                       e Zd ZdZdS ))TestRegression.test_py3_compat.<locals>.Cz3Old-style class in python2, normal class in python3N)__name__
__module____qualname____doc__r.   r   r   Cr     s        EEDr   r  w)output)openosdevnullr   infoAttributeErrorr   close)r   r  outs      r   test_py3_compatzTestRegression.test_py3_compat   s    	 	 	 	 	 	 	 	 2:s##	GAACC$$$$$ 	# 	# 	# """	# % IIKKKKKCIIKKKKs   A A; A!!A; ;BN)r  r  r  r   r   r&   r+   r5   r8   r>   rE   rH   rK   rR   rT   r   r   r   r   r   r   r   r   r   r   r  r  r!  r.   r   r   r
   r
      s       ) ) )
! ! !1 1 18 8 8% % %. . .
" " "> > >
  ' ' '
! ! !  / / /*	' 	' 	'P P P3 3 3- - -
' ' '() ) )
     < < <# # #2 2 2*# # #    r   r
   )r  numpyr   numpy.testingr   r   r   r   r   r   pytestr
   r.   r   r   <module>r%     s    				                    Y Y Y Y Y Y Y Y Y Yr   