o d<@sddlmZmZddlZddlZddlZddlmZddlm m Z ddl Z ddl mZmZmZmZmZmZddlmZddlmZmZe jZGdddZdd Zd d ZdS) )datetime timedeltaN)PerformanceWarning) Categorical DataFrameSeries Timestamp date_rangeoption_context) NumericBlock ObjectBlockc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#S)$TestDataFrameBlockInternalscCsxtdddd}|d}td|i}|djjdusJtj|jd<|djjdus+J|jdks2J|d|ks:JdS) N20130101 US/Eastern)periodstzB)rrD)r r_valuesfreqpdNaTiloc)selfdtitsdfrl/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/frame/test_block_internals.py,test_setitem_invalidates_datetime_index_freq&s  zHTestDataFrameBlockInternals.test_setitem_invalidates_datetime_index_freqcCsXt|jtd}t|jtd}t||t|jtjd}t|jtjd}t||dS)Ndtype)r_mgrint_seriestmassert_frame_equalnpint32)r float_framecastedexpectedrrr test_cast_internals7s  z/TestDataFrameBlockInternals.test_cast_internalscCsd|d<|}t|jjdksJ|}||usJt||d|d<t|jjdks0J|}|dus:Jt|jjdksDJdS)N@Erg @Fr) _consolidatelenr$blocksr'r(_consolidate_inplace)rr+ consolidatedrecons return_valuerrr test_consolidate@s   z,TestDataFrameBlockInternals.test_consolidatecCs4|}ttdtdD] }t||t|<q dS)NAZ)copyrangeordchr)rr+frameletterrrr test_consolidate_inplaceQsz4TestDataFrameBlockInternals.test_consolidate_inplacecCs|r+tjtdd d|jd<Wdn1swY|jddks)JdSd|jd<|jddks;Jd|d<|d}d|jd<|jddkrSJ|dks[JdS)Nz read-onlymatchr/r0)pytestraises ValueErrorvaluesall)rr+using_copy_on_writecolrrr test_modify_valuesXs   z.TestDataFrameBlockInternals.test_modify_valuescCs<d|d<|j}d||dk<d||dk<t||jdS)Nr/r0r)rKr<r'assert_almost_equal)rr+r-rrr test_boolean_set_unconsls    z3TestDataFrameBlockInternals.test_boolean_set_unconsc Cstddgi}|d}ttdgtjdd}t||tddgi}|d}ttdgtjdd}t||tdtddddgi}|d}tttddddgtj dd}t||tdddgi}|d}tttj dgtj dd}t||tddd gi}|d}ttdd gtj dd}t||tdd d gi}|d}ttd d gtj dd}t||tdd d gi}|d}ttd d gtj dd}t||tdd dgi}|d}ttd dgtj dd}t||tdddgi}|d}ttdtj gtj dd}t||tdd dgi}|d}ttd tj gtj dd}t||tdgd i}|d}ttgd tj dd}t||tdddtddddgi}|d}ttddtddddgtj dd}t||dS)Nr:l)namelirTg?rPy?@rg@)@rTNrTi) rrr)asarrayint64r'assert_series_equaluint64robject_nanfloat_complex_)rrresultr-rrr test_constructor_with_convertush           z9TestDataFrameBlockInternals.test_constructor_with_convertcCstdddtjtdddgtdddtdddtdddgg}t|}|j}tddi}t|d<tddd |d <|djd ksCJ|d jd ksLJ|j}ttd gdtdtdtdgt dgdd}t ||dS)NrrErPirzdatetime64[ns]r)dayssecondsrM8[ns]zm8[ns]float64objectztimedelta64[ns]ABCD)foorrindex) rr)rZrdtypesrnowrr#listr'rW)rfloat_string_framedatarr]r-rrr test_construction_with_mixeds*    z8TestDataFrameBlockInternals.test_construction_with_mixedcCstjgddd}ttdd}||d<tdtjdddd itdd}t|d|tt d t d dd  d d tdd}|j ddksKJ|j dd ksTJttdd}t d|d<tjgddd|d<t||dS)NrrPrztimedelta64[s]r"rrhr:z00:00:01s)rrr)rzM8[s])dt1dt2rrrbrs 2013-01-01)rtz 2013-01-02z 2013-01-03z datetime64[D])r)arrayrr=rtimedelta_ranger'assert_numpy_array_equalto_numpyrr astyperj datetime64r()rarrrr-rrr "test_construction_with_conversionss* z>TestDataFrameBlockInternals.test_construction_with_conversionscCsdd}d}tjt|d|gdWdn1swYtjtdd |dWdn1s9wYtjtdd |dWdn1sTwYd }tjt|d |d WddS1srwYdS) NcSs0tttdddddfd}t|gd|dS)Nr_raa )r:rC)rncolumnsr#)rl itertoolsrepeatrr)r#rnrrr fszGTestDataFrameBlockInternals.test_constructor_compound_dtypes..fz@compound dtypes are not implemented in the DataFrame constructorrC))r:z datetime64[h])rstr)rr*zargument must berVrczD^Unknown datetime string format, unable to parse: aa, at position 0$rb)rHrINotImplementedError TypeErrorrJ)rrmsgrrr test_constructor_compound_dtypess   "zs     u