U md<@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}|djjdks6ttj|jd<|djjdksVt|jdksdt|d|ksttdS) N20130101 US/Eastern)periodstzB)rrD)r rZ_valuesfreqAssertionErrorpdZNaTZiloc)selfZdtitsdfr`/home/sam/Atlas/atlas_env/lib/python3.8/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_mgrintZ_seriestmassert_frame_equalnpint32)r float_frameZcastedexpectedrrrtest_cast_internals7s  z/TestDataFrameBlockInternals.test_cast_internalscCsd|d<|}t|jjdks$t|}||k s8tt||d|d<t|jjdks`t|}|dksttt|jjdkstdS)N@Erg @Fr)Z _consolidatelenr!blocksrr#r$Z_consolidate_inplace)rr'Z consolidatedZreconsZ return_valuerrrtest_consolidate@s   z,TestDataFrameBlockInternals.test_consolidatecCs4|}ttdtdD]}t||t|<qdS)NAZ)copyrangeordchr)rr'frameletterrrrtest_consolidate_inplaceQsz4TestDataFrameBlockInternals.test_consolidate_inplacec Cs|rBtjtddd|jd<W5QRX|jddks>tdSd|jd<|jddksbtd|d<|d}d|jd<|jddkrt|dkstdS)Nz read-onlymatchr*r+)pytestraises ValueErrorvaluesallr)rr'using_copy_on_writecolrrrtest_modify_valuesXs  z.TestDataFrameBlockInternals.test_modify_valuescCs<d|d<|j}d||dk<d||dk<t||jdS)Nr*r+r)rAr2r#Zassert_almost_equal)rr'r(rrrtest_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||tdd dddgi}|d}ttd dddgtj dd}t||tdd dtddddgi}|d}ttd dtddddgtj dd}t||dS)Nr0l)namelirTg?rFy?@rg@g@i) rrr%Zasarrayint64r#assert_series_equalZuint64rZobject_nanZfloat_Zcomplex_)rrresultr(rrrtest_constructor_with_convertusl           z9TestDataFrameBlockInternals.test_constructor_with_convertcCstdddtjtdddgtdddtdddtdddgg}t|}|j}tddi}t|d<tddd |d <|djd kst |d jd kst |j}ttd gdtdtdtdgt dddd gd}t ||dS)Nrr;rFirzdatetime64[ns]r)dayssecondsrM8[ns]zm8[ns]float64objectztimedelta64[ns]ZABCDZfooindex) rr%rKrdtypesrnowrr rlistr#rJ)rfloat_string_framedatarrLr(rrrtest_construction_with_mixeds*    z8TestDataFrameBlockInternals.test_construction_with_mixedcCstjdddgdd}ttdd}||d<tdtjddd d itdd}t|d|tt d t d dd  d dtdd}|j ddkst |j dd kst ttdd}td|d<tjdddgdd|d<t||dS)NrrFrztimedelta64[s]rrUr0z00:00:01s)rrr)rzM8[s])dt1dt2r^rQr_z 2013-01-01z 2013-01-02z 2013-01-03z datetime64[D])r%arrayrr3rZtimedelta_ranger#Zassert_numpy_array_equalZto_numpyrr astyperWrZ datetime64r$)rarrrr(rrr"test_construction_with_conversionss. z>TestDataFrameBlockInternals.test_construction_with_conversionsc Csdd}d}tjt|d|dddgW5QRXtjtdd|d W5QRXtjtdd|d W5QRXd }tjt|d|d W5QRXdS) NcSs2tttdddddfd}t|dddg|d S) NrNrZaa r0rC)r[columnsr )rY itertoolsrepeatrr)r r[rrrfszGTestDataFrameBlockInternals.test_constructor_compound_dtypes..fz@compound dtypes are not implemented in the DataFrame constructorr9)r0z datetime64[h])rstr)rfr&zargument must berIrRzD^Unknown datetime string format, unable to parse: aa, at position 0$rQ)r>r?NotImplementedError TypeErrorr@)rrjmsgrrr test_constructor_compound_dtypessz}||krbd||<tj||<|rv||j ||f<qD||||<qDq2W5QRXd}t |j t |||gf}t |j t |||gf} || krdksnt dS)NrrrUd)rr%rKr#Zassert_produces_warningr enumeraterVr3locr-rrr) rrCrZwasColidtrDZmyidfirstsecondrrr$test_strange_column_corruption_issuehs    z@TestDataFrameBlockInternals.test_strange_column_corruption_issuecCsTtdddgj}td|i}tddddgi}t||t|jjdtsPt dS)NrrFrr0r) rr`rr#r$ isinstancer!r.r r)rrbrLr(rrr test_constructor_no_pandas_arrays   zZ pandas.errorsrZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrrrrrr r Zpandas._testingZ_testingr#Zpandas.core.internalsr r Zskip_array_manager_invalid_testZ pytestmarkr rrrrrrs    u