a
    Pf|                     @   s2   d dl Z d dlZd dlZd dlZG dd dZdS )    Nc                
   @   s   e Zd Zdd ZddddejededgZ	e
jde	d	d
ed	ed	g dd Ze
jde	ddedg dd Ze
jde	dg dd ZdS )TestSetitemValidationc                 C   s  dt | d|j }t|}tjt|d ||d< W d    n1 sL0    Y  tjt|d ||d d < W d    n1 s0    Y  tjt|d ||dg< W d    n1 s0    Y  t|}tjt|d ||d< W d    n1 s0    Y  d S )NzInvalid value 'z' for dtype )matchr   )	strdtypereescapepytestZraises	TypeErrorpdZSeries)selfarrinvalidmsgZser r   a/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/arrays/masked/test_indexing.py_check_setitem_invalid
   s    
&*(

z,TestSetitemValidation._check_setitem_invalidy      ?       @True1z1.0NaTr      g      ?c                 C   s"   t jg ddd}| || d S )N)TFNbooleanr   r
   arrayr   )r   r   r   r   r   r   #test_setitem_validation_scalar_bool-   s    z9TestSetitemValidation.test_setitem_validation_scalar_boolTg      ?c                 C   s"   t jg d|d}| || d S N)r      Nr   r   )r   r   Zany_int_ea_dtyper   r   r   r   "test_setitem_validation_scalar_int4   s    z8TestSetitemValidation.test_setitem_validation_scalar_intc                 C   s"   t jg d|d}| || d S r   r   )r   r   Zfloat_ea_dtyper   r   r   r   $test_setitem_validation_scalar_float9   s    z:TestSetitemValidation.test_setitem_validation_scalar_floatN)__name__
__module____qualname__r   r
   r   npZ
datetime64Ztimedelta64Z_invalid_scalarsr   markZparametrizeZint64Zfloat64r   r   r   r   r   r   r   r   	   s"   


r   )r   numpyr"   r   Zpandasr
   r   r   r   r   r   <module>   s   