
    h                        d dl Z d dlZd dlZd dlZd dlmZ d dlmZ	 d Z
d Zej                  j                  d ej                  dg      dddfd ej                  dg      ddfg      d	        Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zej                  j                  dddg      ej                  j                  dddg      d               Zd Zd Zd Zy)    N)	Timestampc                      d} t        j                  t        |       5  t        j                  t        j                  ddg      t        j                  g d             d d d        y # 1 sw Y   y xY w)Nzcnumpy array are different

numpy array shapes are different
\[left\]:  \(2L*,\)
\[right\]: \(3L*,\)match               pytestraisesAssertionErrortmassert_numpy_array_equalnparraymsgs    TD:\jyotish\venv\Lib\site-packages\pandas/tests/util/test_assert_numpy_array_equal.py,test_assert_numpy_array_equal_shape_mismatchr      sK    C 
~S	1
##BHHaV$4bhhy6IJ 
2	1	1   AA((A1c                      d} t        j                  t        |       5  t        j                  dd       d d d        y # 1 sw Y   y xY w)NzExpected typer   r   r   r   r   r   r   r   )expecteds    r   &test_assert_numpy_array_equal_bad_typer      s1    H	~X	6
##Aq) 
7	6	6s	   >Aza,b,klass1,klass2r   ndarrayintc                     d| d| }t        j                  t        |      5  t        j                  | |       d d d        y # 1 sw Y   y xY w)NzHnumpy array are different

numpy array classes are different
\[left\]:  z
\[right\]: r   r   )abklass1klass2r   s        r   ,test_assert_numpy_array_equal_class_mismatchr%      sK    
 X X	C 
~S	1
##Aq) 
2	1	1s   AAc            	      $   d} t        j                  t        |       5  t        j                  t        j                  t
        j                  ddg      t        j                  dt
        j                  dg             d d d        y # 1 sw Y   y xY w)Nznumpy array are different

numpy array values are different \(66\.66667 %\)
\[left\]:  \[nan, 2\.0, 3\.0\]
\[right\]: \[1\.0, nan, 3\.0\]r   r   r
   r   )r   r   r   r   r   r   r   nanr   s    r   -test_assert_numpy_array_equal_value_mismatch1r(   ,   s\    (C 
~S	1
##BHHbffa^$<bhh266ST~>VW 
2	1	1s   ABBc                      d} t        j                  t        |       5  t        j                  t        j                  ddg      t        j                  ddg             d d d        y # 1 sw Y   y xY w)Nzonumpy array are different

numpy array values are different \(50\.0 %\)
\[left\]:  \[1, 2\]
\[right\]: \[1, 3\]r   r   r   r
   r   r   s    r   -test_assert_numpy_array_equal_value_mismatch2r*   7   sO    C 
~S	1
##BHHaV$4bhh1v6FG 
2	1	1r   c            
         d} t        j                  t        |       5  t        j                  t        j                  ddgddgddgg      t        j                  ddgddgddgg             d d d        y # 1 sw Y   y xY w	Nznumpy array are different

numpy array values are different \(16\.66667 %\)
\[left\]:  \[\[1, 2\], \[3, 4\], \[5, 6\]\]
\[right\]: \[\[1, 3\], \[3, 4\], \[5, 6\]\]r   r   r   r
   r   r      r   r   s    r   -test_assert_numpy_array_equal_value_mismatch3r.   B   p    9C 
~S	1
##HHq!fq!fq!f-.1a&1a&1a&9Q0R	
 
2	1	1   AA66A?c                      d} t        j                  t        |       5  t        j                  t        j                  ddg      t        j                  ddg             d d d        y # 1 sw Y   y xY w)Nznumpy array are different

numpy array values are different \(50\.0 %\)
\[left\]:  \[1\.1, 2\.000001\]
\[right\]: \[1\.1, 2.0\]r   g?g7   @g       @r   r   s    r   -test_assert_numpy_array_equal_value_mismatch4r2   O   sO    !C 
~S	1
##BHHc8_$=rxxc
?ST 
2	1	1r   c            
         d} t        j                  t        |       5  t        j                  t        j                  ddgddgddgg      t        j                  ddgddgddgg             d d d        y # 1 sw Y   y xY wr,   r   r   s    r   -test_assert_numpy_array_equal_value_mismatch5r4   Z   r/   r0   c            	          d} t        j                  t        |       5  t        j                  t        j                  ddgddgg      t        j                  ddgddgg             d d d        y # 1 sw Y   y xY w)Nznumpy array are different

numpy array values are different \(25\.0 %\)
\[left\]:  \[\[1, 2\], \[3, 4\]\]
\[right\]: \[\[1, 3\], \[3, 4\]\]r   r   r   r
   r   r   r   s    r   -test_assert_numpy_array_equal_value_mismatch6r6   g   sd    -C 
~S	1
##HHq!fq!f%&1a&1a&1A(B	
 
2	1	1s   A	A00A9c                      d} t        j                  t        |       5  t        j                  t        j                  ddg      t        j                  g d      d       d d d        y # 1 sw Y   y xY w)NzWIndex are different

Index shapes are different
\[left\]:  \(2L*,\)
\[right\]: \(3L*,\)r   r   r   r	   Index)objr   r   s    r   5test_assert_numpy_array_equal_shape_mismatch_overrider:   t   sM    C 
~S	1
##BHHaV$4bhhy6IwW 
2	1	1s   AA**A3c                      d} t        j                  t        |       5  t        j                  t        j                  g d      t        j                  g d             d d d        y # 1 sw Y   y xY w)Nu   numpy array are different

numpy array values are different \(33\.33333 %\)
\[left\]:  \[á, à, ä\]
\[right\]: \[á, à, å\]r   )   á   à   ä)r<   r=      år   r   s    r   test_numpy_array_equal_unicoder@      sL    
!C 
~S	1
##HH'("((3E*F	
 
2	1	1r   c                  8   t        j                  t        d      t        d      g      } t        j                  t        d      t        d      g      }d}t        j                  t
        |      5  t        j                  | |       d d d        y # 1 sw Y   y xY w)Nz
2011-01-01z
2011-01-02znumpy array are different

numpy array values are different \(50\.0 %\)
\[left\]:  \[2011-01-01 00:00:00, 2011-01-01 00:00:00\]
\[right\]: \[2011-01-01 00:00:00, 2011-01-02 00:00:00\]r   )r   r   r   r   r   r   r   r   r!   r"   r   s      r   test_numpy_array_equal_objectrC      so    
)L)9\+BCDA
)L)9\+BCDA?C 
~S	1
##Aq) 
2	1	1s   0BB
other_typesamecopy
check_samec                 d   t        j                  g d      }d }| dk(  r|j                         }n|j                         }|| k7  r	|dk(  rdnd}|=t	        j
                  t        |      5  t        j                  |||       d d d        y t        j                  |||       y # 1 sw Y   y xY w)N)r   r   r
   rE   z0array\(\[1, 2, 3\]\) is not array\(\[1, 2, 3\]\)z,array\(\[1, 2, 3\]\) is array\(\[1, 2, 3\]\)r   )rG   )	r   r   viewrF   r   r   r   r   r   )rD   rG   r!   r   others        r    test_numpy_array_equal_copy_flagrK      s     	A
CVZ V# @@ 	 ]]>5''5ZH 65 	##AuD 65s   +B&&B/c                     t        j                  ddg      } t        j                  dt        j                  gt              }d}t        j                  t        |      5  t        j                  | |       d d d        y # 1 sw Y   y xY w)NTFdtypez{numpy array are different

numpy array values are different \(50.0 %\)
\[left\]:  \[True, False\]
\[right\]: \[True, <NA>\]r   )
r   r   pdNAobjectr   r   r   r   r   rB   s      r   "test_numpy_array_equal_contains_narR      s_    
$A
$f-A!C 
~S	1
##Aq) 
2	1	1s    B  B	c                 0   t        j                  | gt              }t        j                  ||       t        | d      r| j                         }nt        j                  |       }t        j                  |gt              }t        j                  ||       y )NrM   rF   )r   r   rQ   r   r   hasattrrF   )nulls_fixturer!   rJ   r"   s       r   #test_numpy_array_equal_identical_narV      sn    
-/A1% }f%""$		-(
%'A1%    c                  <   t        j                  t         j                  gt              } t        j                  t        j
                  gt              }d}t        j                  t        |      5  t        j                  | |       d d d        y # 1 sw Y   y xY w)NrM   znnumpy array are different

numpy array values are different \(100.0 %\)
\[left\]:  \[nan\]
\[right\]: \[<NA>\]r   )r   r   r'   rQ   rO   rP   r   r   r   r   r   rB   s      r   #test_numpy_array_equal_different_narY      sa    
"&&(A
"%%'AC 
~S	1
##Aq) 
2	1	1s   2BB)rF   numpyr   r   pandasrO   r   pandas._testing_testingr   r   r   markparametrizer   r%   r(   r*   r.   r2   r4   r6   r:   r@   rC   rK   rR   rV   rY    rW   r   <module>ra      s        K* bhhsmQ	5)Axrxx}eY+OP*	*XH

U



X
"* '78'78E 9 9E.*&*rW   