B x\@s,ddlZddlZddlmZmZmZmZddlm m Z ej ddZ ej ddZej ddZej d d Zej d d Zej d dZej ddZej ddZej ddZej ddZej ddZej ddZej ddZej ddZej dd Zej d!d"Zej d#d$ZdS)%N) DataFrameNaTcompat date_rangecCs ttS)zm Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']. )rtm getSeriesDatarr:lib/python3.7/site-packages/pandas/tests/frame/conftest.py float_framesr cCs8tt}tj|jdd<tj|jddddf<|S)z Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']; some entries are missing  N)rrrnpnanloc)dfrrr float_frame_with_nas rcCsttddddgdS)zl Fixture for DataFrame of floats with index of unique strings Columns are ['D', 'C', 'B', 'A'] DCBA)columns)rrrrrrr float_frame2 srcCsFttdk}|t}tj|jdd<tj|jddddf<|S)z Fixture for DataFrame of booleans with index of unique strings Columns are ['A', 'B', 'C', 'D']; some entries are missing rr r r rrN)rrrastypeobjectrrr)rrrr bool_frame_with_na*s  rcCs:tddttD}tddt|DtjdS)zj Fixture for DataFrame of ints with index of unique strings Columns are ['A', 'B', 'C', 'D'] cSsi|]\}}|t|qSr)rint).0kvrrr @szint_frame..cSsi|]\}}||qSrr)rcsrrr r"Cs)dtype)rr iteritemsrrrint64)rrrr int_frame9sr(cCs ttS)zb Fixture for DataFrame of floats with DatetimeIndex Columns are ['A', 'B', 'C', 'D'] )rrZgetTimeSeriesDatarrrr datetime_frameFsr)cCstt}d|d<|S)z Fixture for DataFrame of floats and strings with index of unique strings Columns are ['A', 'B', 'C', 'D', 'foo']. barfoo)rrr)rrrr float_string_framePs r,cCsHtt}|jd|_|jd|_|jd|_|jd|_|S)z| Fixture for DataFrame of different float types with index of unique strings Columns are ['A', 'B', 'C', 'D']. float32float16float64)rrrrrrrr)rrrr mixed_float_frame\s  r0cCsHtt}|jd|_|jd|_|jd|_|jd|_|S)z| Fixture for DataFrame of different float types with index of unique strings Columns are ['A', 'B', 'C', 'D']. r-r.r/)rrrrrrr)rrrr mixed_float_frame2ks  r1cCs`tddttD}|jd|_tjt |j dd|_ |j d|_ |j d|_ |S)zz Fixture for DataFrame of different int types with index of unique strings Columns are ['A', 'B', 'C', 'D']. cSsi|]\}}|t|qSr)rr)rr r!rrr r"sz#mixed_int_frame..int32Zuint64)r%Zuint8r') rrr&rrrrrZoneslenrrr)rrrr mixed_int_framezsr4c Cs>tdddtjdgdddtjdgdddd tdd S) z Fixture for DataFrame of float/int/string columns with RangeIndex Columns are ['a', 'b', 'c', 'float32', 'int32']. g?r+r r-)r%r2)abr#r-r2)index)rrarrayZarangerrrr mixed_type_framesr;cCsDttdddtddddtddddd}t|jd<t|jd <|S) z Fixture for DataFrame of date_range Series with different time zones Columns are ['A', 'B', 'C']; some entries are missing Z20130101)periodsz US/Eastern)r=ZtzZCET)rrr)r6r6)r6r5)rrrZiloc)rrrr timezone_frames   r>cCstiS)z% Fixture for empty DataFrame )rrrrr empty_framesr?cCs tjddS)z9 Fixture for Series of floats with DatetimeIndex )nper)rmakeTimeSeriesrrrr datetime_seriessrCcCstjddddS)z9 Fixture for Series of floats with DatetimeIndex r@)rAr N)rrBrrrr datetime_series_shortsrDcCs<tdddgdddgddd gg}t|d d d gd ddgdS)zn Fixture for simple 3x3 DataFrame Columns are ['one', 'two', 'three'], index is ['a', 'b', 'c']. g?g@g@g@g@g@g@g @g"@onetwothreer7r8r#)rr9)rr:r)Zarrrrr simple_frames   rHc CsPtdddddgdddddgdddd d gtjd tjd tjd d }|S) z Fixture for DataFrame of columns that can be used for indexing Columns are ['A', 'B', 'C', 'D', 'E', ('tuple', 'as', 'label')]; 'A' & 'B' contain duplicates (but are jointly unique), the rest are unique. r+r*rErFrGr7r8r#der )rrrrE)tupleasZlabel)rrZrandomZrandn)rrrr frame_of_index_colss    rN)ZnumpyrZpytestZpandasrrrrZpandas.util.testingutilZtestingrZfixturer rrrr(r)r,r0r1r4r;r>r?rCrDrHrNrrrr s(