
    Owg                         d dl Zd dlZd dlZd dlmZ d dlm	Z	 d dl
mZ ej                  d        Zej                  d        Z G d de      Zy)    N)BooleanArray)ComparisonOpsc                      t        j                  ddgdz  t        j                  gz   ddgdz  z   t        j                  gz   ddgz   d      S )z;Fixture returning boolean array with valid and missing dataTF   ,   booleandtype)pdarraynpnan     b/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/arrays/boolean/test_comparison.pydatar   
   sQ     88	uRVVH$e}r'99RVVHDe}T r   c                  *    t        j                         S )zFixture returning BooleanDtype)r   BooleanDtyper   r   r   r
   r
      s     ??r   c                   |    e Zd Zd Zd Zej                  j                  dddej                  g      d        Z
d Zy)	TestComparisonOpsc                 *    | j                  ||d       y )NT)_compare_other)selfr   comparison_ops      r   test_compare_scalarz%TestComparisonOps.test_compare_scalar   s    D-6r   c                 F   t        j                  dgt        |      z  d      }| j                  |||       t	        j                  dgt        |      z        }| j                  |||       t        j
                  dgt        |      z        }| j                  |||       y )NTr   r	   )r   r   lenr   r   Series)r   r   r   others       r   test_compare_arrayz$TestComparisonOps.test_compare_array   s    $#d)+9=D-7$#d)+,D-7		4&3t9,-D-7r   r   TFc                 4    t        j                  | |||       y )N)r   test_scalar)r   r   r   r
   s       r   r"   zTestComparisonOps.test_scalar%   s    !!$}eDr   c                 8   |}t        j                  dgdz  dgdz  z   d gdz  z   d      }t        j                  g ddz  d      } |||      } ||j                  |j                        }|j                  |j                  z  }t	        ||      }t        j                  ||       d |d<   t        j                  |t        j                  dgdz  dgdz  z   d gdz  z   d             t        j                  |t        j                  g ddz  d             y )NT   Fr   r	   )TFNr   )r   r   _data_maskr   tmassert_extension_array_equal)	r   r   opabresultvaluesmaskexpecteds	            r   
test_arrayzTestComparisonOps.test_array)   s   HHdVaZ5'A+-
:)LHH(1,I>AqAGGQWW%ww -
''9 q	
''rxx
eWq[0D6A:=YO	
 	''rxx+a/yA	
r   N)__name__
__module____qualname__r   r    pytestmarkparametrizer   NAr"   r0   r   r   r   r   r      sC    78 [[WtUBEE&:;E <E
r   r   )numpyr   r4   pandasr   pandas._testing_testingr'   pandas.arraysr   !pandas.tests.arrays.masked_sharedr   fixturer   r
   r   r   r   r   <module>r?      sS        & ;    
#
 #
r   