U
|
®ý°d÷ ã @ sº d dl Zd dlZd dlmZ d dlmZ d dlm Z ej
|
di ddiddig¡dd Z ej
|
d
|
e d ¡d d Zdd Zdd Zej
|
dddg¡dd Zej
|
dddg¡dd ZdS ) é N)Úarray)Ú SparseArrayÚkwargsÚ check_exactFTc C sh t ddg}t ddg}| dd¡rTd}tjt|d tj||f| W 5 Q R X ntj||f| d S )Ng{kAÆ¿g0Ûw½ÙÕ?g0Ûw½ÙÕ?r FzµExtensionArray are different
|
|
ExtensionArray values are different \(50\.0 %\)
|
\[left\]: \[-0\.17387645482.*, 0\.341414801642.*\]
|
\[right\]: \[-0\.17387645482.*, 0\.341414801642.*\]©Úmatch)r ÚgetÚpytestÚraisesÚAssertionErrorÚtmÚ assert_extension_array_equal)r Úarr1Úarr2Úmsg© r újd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/util/test_assert_extension_array_equal.pyÚ+test_assert_extension_array_equal_not_exact s
|
r Údecimalsé
|
c C sr dd| }t ddg}t ddg}| dkr^d}tjt|d tj|||d W 5 Q R X ntj|||d d S ) Ng à?r g¿¶~úÏ¿?guWvÁà¿?é zExtensionArray are different
|
|
ExtensionArray values are different \(50\.0 %\)
|
\[left\]: \[0\.5, 0\.123456\]
|
\[right\]: \[0\.5, 0\.123457\]r )Úrtol)r r r
|
r r r )r r r r r r r r Ú.test_assert_extension_array_equal_less_precise$ s r c C sx d}d| i}t tj|dd}t tj|dd}| rdd}tjt|d tj||f| W 5 Q R X ntj||f| d S )Nr Ú check_dtypeÚint64©ZdtypeÚint32zyExtensionArray are different
|
|
Attribute "dtype" are different
|
\[left\]: Sparse\[int64, 0\]
|
\[right\]: Sparse\[int32, 0\]r )r ÚnpÚaranger r
|
r r r )r Úendr r r r r r r Ú0test_assert_extension_array_equal_dtype_mismatch8 s r c C sT t tjddtjg} t tjdddg}d}tjt|d t | |¡ W 5 Q R X d S )Né é é z®ExtensionArray NA mask are different
|
|
ExtensionArray NA mask values are different \(25\.0 %\)
|
\[left\]: \[True, False, False, True\]
|
\[right\]: \[True, False, False, False\]r )r r Únanr r
|
r r r )r r r r r r Ú0test_assert_extension_array_equal_missing_valuesM s
|
r% ÚsideÚleftÚrightc C s\ t d¡}t|}| d}| dkr,||fn||f}tjt|d tj| W 5 Q R X d S )Nr z is not an ExtensionArrayr' r )r r r r r
|
r r r )r& Z numpy_arrayZextension_arrayr Úargsr r r Ú5test_assert_extension_array_equal_non_extension_array\ s
|
|
ÿ
|
ýr* Ú right_dtypeZInt32r c C s8 t dddgdd}t dddg| d}tj||dd d S )Nr! r" r# ZInt64r F)r )r r r )r+ r' r( r r r Ú7test_assert_extension_array_equal_ignore_dtype_mismatchl s r, )Únumpyr r Zpandasr Zpandas._testingZ_testingr Zpandas.core.arrays.sparser ÚmarkZ parametrizer Úranger r r% r* r, r r r r Ú<module> s( ýþ
|
|
|
|
