a
    Pf                     @   sl   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d Zejdd ZG dd	 d	eZdS )
    N)BooleanArray)ComparisonOpsc                   C   s:   t jddgd tjg ddgd  tjg ddg ddS )z;Fixture returning boolean array with valid and missing dataTF   ,   booleandtype)pdarraynpnan r   r   d/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/arrays/boolean/test_comparison.pydata
   s    .r   c                   C   s   t  S )zFixture returning BooleanDtype)r	   ZBooleanDtyper   r   r   r   r      s    r   c                   @   sB   e Zd Zdd Zdd Zejdddej	gdd	 Z
d
d ZdS )TestComparisonOpsc                 C   s   |  ||d d S )NT)_compare_other)selfr   comparison_opr   r   r   test_compare_scalar   s    z%TestComparisonOps.test_compare_scalarc                 C   sn   t jdgt| dd}| ||| tdgt| }| ||| t dgt| }| ||| d S )NTr   r   )r	   r
   lenr   r   ZSeries)r   r   r   otherr   r   r   test_compare_array   s    z$TestComparisonOps.test_compare_arrayr   TFc                 C   s   t | ||| d S )N)r   test_scalar)r   r   r   r   r   r   r   r   %   s    zTestComparisonOps.test_scalarc           	      C   s   |}t jdgd dgd  d gd  dd}t jg dd dd}|||}||j|j}|j|jB }t||}t|| d |d< t|t jdgd dgd  d gd  dd t|t jg dd dd d S )NT   Fr   r   )TFNr   )r	   r
   _dataZ_maskr   tmZassert_extension_array_equal)	r   r   opabresultvaluesmaskexpectedr   r   r   
test_array)   s    (

(zTestComparisonOps.test_arrayN)__name__
__module____qualname__r   r   pytestmarkZparametrizer	   ZNAr   r#   r   r   r   r   r      s
   
r   )numpyr   r'   Zpandasr	   Zpandas._testingZ_testingr   Zpandas.arraysr   Z!pandas.tests.arrays.masked_sharedr   Zfixturer   r   r   r   r   r   r   <module>   s   

