
    Bwg                     H    d dl Z d dlZd dlmZmZmZmZmZm	Z	  G d d      Z
y)    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y)TestRegressionc                     t        t        j                  dg      t        j                  ddg      z
  t        j                  ddg             y )N   r   )r   nppoly1dselfs    V/var/www/horilla/myenv/lib/python3.12/site-packages/numpy/lib/tests/test_regression.pytest_poly1dzTestRegression.test_poly1d   s8    RYYs^biiA&77YYAw'	)    c                     t         j                  j                  d      }|j                         }t        j                  |d       t        j                  |d       t	        ||       y )N)   r   T)rowvarF)r   randomcopycovr   r   xys      r   test_cov_parametersz"TestRegression.test_cov_parameters   sG    IIV$FFH
q
q1a r   c                     t        d      D ]6  }t        j                  g dddg       t        j                  g dddg       8 y )Nd   )r      r      r   r   )r   r   r!   r   r"   )ranger   digitize)r   is     r   test_mem_digitizez TestRegression.test_mem_digitize   s9    s 	1AKKq!f-KK!Q0	1r   c                 h    t        g t        j                  t        j                  g                    y N)r   r   uniquearrayr   s    r   test_unique_zero_sizedz%TestRegression.test_unique_zero_sized   s    2ryy"67r   c                 d   t        j                  d       } |t        j                  d      t        j                  d      t        j                  d              |t        j                  d      t        j                  d      t        j                  d      t        j                  d             y )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"   sr    \\,-
288I 3RXXi5HI
288I 3RXX ? 688F#	%r   c                     t        t        j                  ddd   dg       t        t        j                  dd g        y )Nr   y              ?)r   r   mgridr   s    r   test_mgrid_single_elementz(TestRegression.test_mgrid_single_element)   s1    288AaF+aS1288Aa="-r   c                 J    d }t        j                  |      }t        |       y )Nc                      y)N{   r.   )r   r   s     r   pz1TestRegression.test_refcount_vectorize.<locals>.p0   s    r   )r   r2   r   )r   r<   vs      r   test_refcount_vectorizez&TestRegression.test_refcount_vectorize.   s    	LLOq!r   c                     t        j                  t         j                  t         j                  dgd      }t        t         j                  j
                  t        |d       y )Nr   F)rr@   )r   r   nanr   linalgLinAlgErrorgetattr)r   r<   s     r   test_poly1d_nan_rootsz$TestRegression.test_poly1d_nan_roots5   s9    IIrvvrvvq)U3bii++Wa=r   c                 2    t        j                  g dg       y )Ng      ?)r   polymulr   s    r   test_mem_polymulzTestRegression.test_mem_polymul:   s    


2tr   c                 Z    t        j                  g       }t        j                  |d       y )Nzasdasd	asdasd)r   r*   append)r   r   s     r   test_mem_string_concatz%TestRegression.test_mem_string_concat>   s    HHRL
		!%&r   c                     t        j                  g d      }t        j                  g d      }t        j                  ||      \  }}t        ||z  |z   |       y )Nr   r!   r   )r   r!   r   r"      )r   r   polydivr   )r   ur=   qr@   s        r   test_poly_divzTestRegression.test_poly_divC   sD    IIi IIo&zz!Q1QqS1Wa r   c                     t        j                  g d      }t        j                  ddg      }t        ||k7         t        ||k(         y )NrM   r   r"   )r   r   r   r   s      r   test_poly_eqzTestRegression.test_poly_eqJ   s8    IIi IIq!fQQr   c                 b    g d}g d}g d}t        j                  ||d      }t        ||       y )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_buildQ   s5    00: Aq!$!#v.r   c                 P   d}t        j                  dt              }t        j                  ||      \  }}t	        |j
                  t        k(  |       d}t        j                  dt              }t        j                  ||      \  }}t	        |j
                  t        k(  |       y 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_typef   sz    -GGAW%zz!Q17"C(+GGAS!zz!Q15 #&r   c                 l    t        t        t        j                  t        j                  d      d       y )N)r   
   i   )bins)r   
ValueErrorr   histogramddr   r   s    r   test_histogramdd_too_many_binsz-TestRegression.test_histogramdd_too_many_binsq   s    j".."'''2BOr   c                 8   d}t        j                  dt              }t        t        j                  |      j
                  t        k(  |       d}t        j                  dt              }t        t        j                  |      j
                  t        k(  |       y r   )r   r   r   r   polyintr   r   r   )r   r   r   s      r   test_polyint_typez TestRegression.test_polyint_typeu   sf    -GGAW%

1##w.4+GGAS!

1##u,c2r   c                 h    t        t        j                  t        j                  g g                   y r(   )listr   ndenumerater*   r   s    r   test_ndenumerate_crashz%TestRegression.test_ndenumerate_crash~   s     	R^^BHHbTN+,r   c                     t        j                  t         j                        j                  dz  }t	        d|z  dz  dz         fd}fd}t        t        |       t        t        |       y )N   r!   g?r   c            	          d} t        j                  | fdz        }t         j                  j                  d|       }d|t        j                  |||||      <   y Nr   rN   r   )sizer   r   r   randintix_nar%   thesizes      r   dpz4TestRegression.test_large_fancy_indexing.<locals>.dp   sQ    AQA		!!!QW!5A'(AbffQ1a#$r   c                      d} t        j                  | fdz        }t         j                  j                  d|       }|t        j                  |||||          y r   r   r   s      r   dp2z5TestRegression.test_large_fancy_indexing.<locals>.dp2   sO    AQA		!!!QW!5AbffQ1a#$r   )r   r   intpitemsizer   r   r   )r   nbitsr   r   r   s       @r   test_large_fancy_indexingz(TestRegression.test_large_fancy_indexing   sX    !**Q.q%x7+A-.	)	% 	j"%j#&r   c                     t        j                  ddg      }t        j                  d|      }t        t         j                  ||f   j                  |k(         y )N)r   f4)bi4)r   )r   r   r3   r   r_)r   dtr   s      r   test_void_coercionz!TestRegression.test_void_coercion   sD    XX{K01HHT2ad!!R'(r   c                     t        j                         g}|D ])  }t        t        |t                     t        |dk7         + y )N )r   get_includer   
isinstancestr)r   include_dirspaths      r   test_include_dirsz TestRegression.test_include_dirs   s=     ()  	 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
                               y )Nr   r   )r   r   r   polyderr   ndarrayr   s    r   test_polyder_return_typez'TestRegression.test_polyder_return_type   s    
2::biina8"))DE
2::qc1-rzz:;
2::biina8"))DE
2::qc1-rzz:;r   c                 f   ddl m} t        j                  g dt        j                        }g d}t        j
                  d      j                  t        j                        }t        j                  t        j                  t        j                  g}	  |||||       y # 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   sz    8xx	2vvay)RXXrxx0	!$tU3 	!  	!s   B   B0c                 .   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
                  ||             y )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   s    n-JJx	*"5Q"2!32>?(+678JJx	*"5Q"3!4B?@!"JJx	*"5Q"5!6bAB$%JJx ;<BGIJQR01r   c                     t        j                  dt              }	 t        j                  |       y # 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   s7    HHQd#	!IIaL 	!  	!s	   3 Ac                      G d d      }t        t        j                  d      }	 t        j                   |       |       	 |j                          y # t
        $ r t        w xY w# |j                          w xY w)Nc                       e Zd ZdZy))TestRegression.test_py3_compat.<locals>.Cz3Old-style class in python2, normal class in python3N)__name__
__module____qualname____doc__r.   r   r   Cr     s    Er   r  w)output)openosdevnullr   infoAttributeErrorr   close)r   r  outs      r   test_py3_compatzTestRegression.test_py3_compat   sa    	 	 2::s#	GGAC$ IIK  	!  	! IIKs   A A$$A' 'A9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8%.
">
'
!/*	'P3-
'()
 <!2*!r   r
   )r  numpyr   numpy.testingr   r   r   r   r   r   r
   r.   r   r   <module>r$     s!    	  Y Yr   