U
|
ý°dD ã @ sÄ d dl Zd dlZd dlmZmZ d dlm mZ d dl
|
Z d dl
|
m Z d dl mZ d dlmZmZmZmZmZ dd Zdd Zd d
|
Zd d Zd d Ze d¡ej deej dg¡dd Z dS )é N)Ú _maybe_upcastÚ na_values)ÚNA)ÚArrowStringArrayÚ BooleanArrayÚ FloatingArrayÚ IntegerArrayÚ StringArrayc C sz t | ¡}t| }t jdd|g|d}t|dd}t dddg¡}t|jt jr^t||d}n t ||d}t
|
||¡ d S )Né é ©ÚdtypeT©Zuse_dtype_backendF©Úmask) Únpr r Úarrayr Ú
|
issubclassÚtypeÚintegerr r ÚtmÚ assert_extension_array_equal)Úany_real_numpy_dtyper Úna_valueÚarrÚresultÚ expected_maskÚexpected© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/parser/test_upcast.pyÚtest_maybe_upcast s
|
r c C s~ | dkrt ¡ tjdddg| d}t|dd}t dddg¡}tt | ¡jtjrbt ||d }n t
|
||d }t ||¡ d S )
|
NÚfloat32r
|
r é r Tr Fr ) ÚpytestÚskipr r r r r r r r r r r )r r r r r r r r Útest_maybe_upcast_no_na' s r% c C s` t j} t| }t jdd|gdd | ¡}t|dd}t dddg¡}t||d}t ||¡ d S ©NTFZuint8r r r ) r Úbool_r r Úviewr r r r ©r r r r r r r r r Útest_maybe_upcaste_bool8 s r* c C sX t j} t jdddgdd | ¡}t|dd}t dddg¡}t||d}t ||¡ d S r& )r r' r r( r r r r )r r r r r r r r Ú test_maybe_upcaste_bool_no_nanD s r+ c C sV t j} t| }t j||g| d}t|dd}t ddg¡}t||d}t ||¡ d S )Nr Tr r )r Úint64r r r r r r r) r r r Útest_maybe_upcaste_all_nanO s r- ÚpyarrowÚvalÚcc C sª dd l }t d|¡ tjdd| gtjd}t|dd}|dkrl| d krLd nt}ttjdd|gtjd}n$| d krxd nd }t | dd|g¡}t
|
||¡ W 5 Q R X d S )
|
Nr zmode.string_storageÚaÚbr Tr Úpythonr0 ) r. ÚpdZoption_contextr r Úobject_r r r r r r )r/ Zstring_storageÚpar r Zexp_valr r r r Útest_maybe_upcast_object[ s r7 ) Únumpyr r# Zpandas._libs.parsersr r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr4 r Zpandas._testingZ_testingr Zpandas.core.arraysr r r r r r r% r* r+ r- Z
|
skip_if_noÚmarkZ parametrizer5 r7 r r r r Ú<module> s
|
