a < b!_@spddlZddlZddlmmZddlZddlm Z m Z m Z m Z ddl mZeddZGdddZdS)N) DataFrameIndex MultiIndexSeriescs$GfdddtdgdiS)NcseZdZefddZdS)z2gpd_style_subclass_df..SubclassedDataFramecsSNselfSubclassedDataFramerp/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/frame/test_subclass.py _constructorsz?gpd_style_subclass_df..SubclassedDataFrame._constructorN)__name__ __module__ __qualname__propertyr rr rr r sr a)rrrr r gpd_style_subclass_dfsrc@s eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zejd#d$d%Zd&d'Zd(d)Zd*d+Zejd,d-d.gd/d0Zd1d2Zd3d4Zej d5d6Z!d7d8Z"d9d:Z#d;d<Z$d=d>Z%d?d@Z&dAS)BTestDataFrameSubclassingcsGfdddtGfdddttdtdd}|}t|sRJ|j}t|sfJ|dksvJ|dd }t|sJ|dksJtd d g}d dgd dgg|d}t|dsJtddg}d dgd dgg|d}t|dsJdS)Ncs$eZdZefddZddZdS)zQTestDataFrameSubclassing.test_frame_subclassing_and_slicing..CustomSeriescsSrrr CustomSeriesrr r sz^TestDataFrameSubclassing.test_frame_subclassing_and_slicing..CustomSeries._constructorcSsdSNOKrrrrr custom_series_function$szhTestDataFrameSubclassing.test_frame_subclassing_and_slicing..CustomSeries.custom_series_functionN)rrrrr rrrrr rsrcs<eZdZdZfddZefddZZddZZ S)zTTestDataFrameSubclassing.test_frame_subclassing_and_slicing..CustomDataFramez Subclasses pandas DF, fills DF with simulation results, adds some custom plotting functions. cstj|i|dSr)super__init__)r argskw __class__rr r-sz]TestDataFrameSubclassing.test_frame_subclassing_and_slicing..CustomDataFrame.__init__csSrrr)CustomDataFramerr r 0szaTestDataFrameSubclassing.test_frame_subclassing_and_slicing..CustomDataFrame._constructorcSsdSrrrrrr custom_frame_function6szjTestDataFrameSubclassing.test_frame_subclassing_and_slicing..CustomDataFrame.custom_frame_function) rrr__doc__rrr Z_constructor_slicedr% __classcell__rr$rr"r r$'s  r$ )col1Zcol2rr)Ar,)r,Brrrcolumnsr,)r,)r-r0) rrrange isinstancer*rr%r from_tuples)r dataZcdfZ cdf_seriesZcdf_rowsZmcolZ cdf_multiZ cdf_multi2rr(r "test_frame_subclassing_and_slicings" z;TestDataFrameSubclassing.test_frame_subclassing_and_slicingcCstjgdgddgdd}d|_|jdks4J|dgjdksHJ|jddgddfjdkshJ|jd d gddfjdksJ|jd d ddfjdksJt|}t|||j|jksJ|j|jksJdS) Nr)XYrbcindexZXXXr6rr9rr)tmr ZtestattrlocilocZround_trip_pickleassert_frame_equal _metadata)r dfZ unpickledrrr test_dataframe_metadataRs     z0TestDataFrameSubclassing.test_dataframe_metadatacCstjgdgdgddgdd}|jdddf}tjgdtddd }t||t|tjslJ|jddd f}tjgdtdd d }t||t|tjsJ|jddd f}tjgdtdd d }t||t|tjsJ|jd ddf}tjgdtdd d }t||t|tjs@J|jd ddf}tjgdtddd }t||t|tjsJ|jdddf}tjgdtddd }t||t|tjsJdS)Nrr+ r6r7Zr8r;r6abcr<namerr7rLr)rrErHZXYZ)rr+rIr9r:)rrFrJ)r=r r>SubclassedSerieslistassert_series_equalr2r?r rBresexprrr test_indexing_slicedfs6      z-TestDataFrameSubclassing.test_indexing_slicedcCsJGdddt}tjtdd|jWdn1s<0YdS)Nc@seZdZeddZdS)zFTestDataFrameSubclassing.test_subclass_attr_err_propagation..AcSs|jSr)Z i_dont_existrrrr barszJTestDataFrameSubclassing.test_subclass_attr_err_propagation..A.barN)rrrrrWrrrr r,sr,z.*i_dont_exist.*)match)rpytestZraisesAttributeErrorrW)r r,rrr "test_subclass_attr_err_propagationsz;TestDataFrameSubclassing.test_subclass_attr_err_propagationcCsHtjgdgddtdd}tjgdgddtdd}|j|dd \}}tjd tjd tjd gd tjd tjd gdtd d}tjd dtjdtjgd dtjdtjgdtd d}t|tjsJt||t|tjsJt|||j|j \}}t|tj sJt ||jt|tj s6Jt ||j dS)Nrrr+rr9ACEr;rrrE)r:dABDraxisrrr+ABCDErrE) r=r rQalignnpnanr2r@rr:rPrR)r df1df2res1res2exp1exp2rrr test_subclass_aligns0$$  z,TestDataFrameSubclassing.test_subclass_aligncCs$tjgdgddtdd}tjgdtddd}|j|d d \}}tjd tjd tjd gd tjd tjd gdtdd}tjd dtjdtjgtddd}t|tjsJt||t|tjsJt ||||\}}t|tjsJt ||t|tjsJt||dS)Nr\r]r^r;r_raxrNrrbrrr+rdrrE) r=r rQrPrerfrgr2r@rR)r rBsrjrkrlrmrrr test_subclass_align_combinationss& $   z9TestDataFrameSubclassing.test_subclass_align_combinationscCsHtddgi}|D]*\}}t|tjs0Jt||j|qdS)Nrr)r=r Ziterrowsr2rPrRr>)r rBirowrrr test_subclass_iterrowssz/TestDataFrameSubclassing.test_subclass_iterrowscCs`tjgdgdgdggdgdd}|}tjgdtdtd gd }t||dS) NrrDrGr8rKr<r/) rrrrEr+rFrHrIrJZ aaabbbcccZ XYZXYZXYZr;)r=r stackrPrQrRrSrrr test_subclass_stacksz,TestDataFrameSubclassing.test_subclass_stackc Cstjgdgdgdgdgtjtttdtdddgd tjtttd td d d gd d}tjddgddgddgddgddgddgddgddggtjtttdtd td!gd"d td#d$gd d%d}|}t|||d }t||tjddgddgddgddgddgddgddgddggtjtttdtd td&gd'd td(d)gd d%d}|d }t||dS)*Nr)  !()*+AABBcdcdaaacccnamesWWXXyzyzwwwyyyrur)rzryr{r}rr~rrrrrrrrrAAAABBBBccddccddyzyzyzyzrrrWr6rOWXWXWXWXrrryz r=r rr3rQziprrvr@r rBrUrTrrr test_subclass_stack_multis^       z2TestDataFrameSubclassing.test_subclass_stack_multic Cstjgdgdgdgdgtjtttdtdddgd tjtttd td d d gd d}tjddgddgddgddgddgddgddgddggtjtttdtd td!gd"d td#d$gd d%d}|}t|||d }t||tjd&d'gddgd(d)gddgd*d+gddgd,d-gddggtjtttdtd td.gd/d td0d1gd d%d}|d }t||dS)2Nr)ry(@*@r}r~6@7@rr@@@@rrE@E@rrrrrrrrrrur)rryrr}rr~rrrrrrrrrrrrrrr6rg$@g&@g4@5@g>@g?@gD@gD@rrrrrrrrr test_subclass_stack_multi_mixedsf      z8TestDataFrameSubclassing.test_subclass_stack_multi_mixedcCs`tjgdgdgdggdgdd}|}tjgdtdtd gd }t||dS) NrrDrGr8rKru) rrErHrr+rIrrFrJZ XXXYYYZZZZ abcabcabcr;)r=r unstackrPrQrRrSrrr test_subclass_unstackYsz.TestDataFrameSubclassing.test_subclass_unstackc CsHtjgdgdgdgdgtjtttdtdddgd tjtttd td d d gd d}tjgdgdgtddgddtjtttdtdtdgdd d}|}t|||d}t||tjgdgdgtddgddtjtttdtdtdgdd d}|d}t||dS)Nrxr|rrrrrrrrrrrru)r)r}ryr~rzrr{r)rrrrrrrrr,r-rWWWWXXXXyyzzyyzzcdcdcdcdrrr)r)rryrrzrr{r)r}rr~rrrrrr:r`ABABABABrrr r=r rr3rQrrrr@rrrr test_subclass_unstack_multihs>      z4TestDataFrameSubclassing.test_subclass_unstack_multic CsHtjgdgdgdgdgtjtttdtdddgd tjtttd td d d gd d}tjgdgdgtddgddtjtttdtdtdgdd d}|}t|||d}t||tjgdgdgtddgddtjtttdtdtdgdd d}|d}t||dS)Nrrrrrrrrrrrrrru)r)r}ryr~rrrr)rrrrrrrrr,r-rrrrr)r)rryrrrrr)r}rr~rrrrrr:r`rrrrrrr !test_subclass_unstack_multi_mixedsN     z:TestDataFrameSubclassing.test_subclass_unstack_multi_mixedcCsntgdgdgdd}|jdddd}tdd d d dd d d d }d \|j_|j_t||dS)N)r,r-Crr-r,)OnerrTworr)?@@rrr)r<r/valuesr<r/rrrrr,r-r)rrru)r=r Zpivotr<rOr/r@)r rBZpivotedexpectedrrr test_subclass_pivots  z,TestDataFrameSubclassing.test_subclass_pivotcCsptddgddgddgddgd }tj|d d gd }tjgd gdgdgdggdd}t||dS)NJohnMaryDoeBo@@)firstlastheightweightrr)Zid_varsrrrrrrrrrrrrrrrrrrvariablevaluer.)r=r pdZmeltr@)r ZcheeseZmeltedrrrr test_subclassed_melts"  z-TestDataFrameSubclassing.test_subclassed_meltc Cstjdtjd}tddddddd dd d d dd dddtttd|d}|j |d<| | gdgdgdgdd}t|}| ddggd}t j |ddgddd}t||dS)N{rrr9r:)rrrr`ef@333333?ffffff?皙 @?皙?)ZA1970ZA1980ZB1970ZB1980r6id)rr9r:r`rr)rrrrrr)rrrr)rrrrrr)r6r,r-yearrr)r6r,r-r,r-)rrj)rfrandomseedrandnr=r dictrr1r<tolistZ set_indexrZ wide_to_longr@)r rorBZexp_datarZ long_framerrr test_subclassed_wide_to_longs*        z5TestDataFrameSubclassing.test_subclassed_wide_to_longcsddddtjgdgdgdgdggd d }|fd d |jfd d ddtjgdgdgdgdggd d }|jfdd dd}t|tjsJt||tgdgdgdgdg}|jdd dd}t|tjsJt|||jdd ddd}t|tjs6Jt||tgdgdgdgdg}|jdd dd}t|tjrJt||dS)NcSst|tjsJdSr)r2r=rPrsrrr check_row_subclassszJTestDataFrameSubclassing.test_subclassed_apply..check_row_subclasscSs |ddkr|dd7<|S)Nrrrg?rrrrr stretchs z?TestDataFrameSubclassing.test_subclassed_apply..stretchrrrrrr.cs|Srrrorrr #z@TestDataFrameSubclassing.test_subclassed_apply..cs|Srrrrrr r$rrrb)rrrr)rrrg@cs|Srrr)rrr r0rrcSstgdSNr)r=rPrrrr r6rcSsgdSrrrrrr r:rexpand)rcZ result_typecSsgdSrrrrrr r@r)r=r applyr2r@rPrR)r rBrresultr)rrr test_subclassed_applysD   "  "z.TestDataFrameSubclassing.test_subclassed_applyz*ignore:.*None will no longer:FutureWarningcCs>tgdgdgdd}t||}t|tjs:JdSNrrDrGrr=r getattrr2rP)r Zall_reductionsrBrrrr test_subclassed_reductionsDs z3TestDataFrameSubclassing.test_subclassed_reductionsc CsLtgddtjdddggdd}|}t|tjs>Jtgdgd gd d }|}t|tjstJtjgd gd gdgdgtjt t t dt dddgdtjt t t dt dddgdd}t t |jdd}Wdn1s 0Yt|tjs&Jt}|}t|tjsHJdS)N)rMylaZLewisrrg8@rr)FTTTF)ZPersonZAgeSingle)rrr)rr+rF)rHrIrrrxr|rrrrrrrrrrrrur)level) r=r rfrgcountr2rPrr3rQrZassert_produces_warning FutureWarningr rBrrrr test_subclassed_countLs4 ,z.TestDataFrameSubclassing.test_subclassed_countcCsBtjddgddgdddgd}|ddg}t|tjs>JdS)NrrEr)Znum_legsZ num_wingsZfalconZdogr;)r=r isinr2rrrr test_isinms z"TestDataFrameSubclassing.test_isincCsZtgdgdgdd}|}t|tjs6Jt}|}t|tjsVJdSr)r=r Z duplicatedr2rPrrrr test_duplicatedus z(TestDataFrameSubclassing.test_duplicated idx_methodidxmaxidxmincCs>tgdgdgdd}t||}t|tjs:JdSrr)r r rBrrrr test_idxs z!TestDataFrameSubclassing.test_idxcCstgdgdg}tgd}||}t|tjs>Jtgdgdg}tgd}||}t|tjs|JdS)N)rr)rrrr)rrrr)r=r rPdotr2)r rBrprrrr test_dots  z!TestDataFrameSubclassing.test_dotcCsVtgdgdgdd}|}t|tjs6J|jdd}t|tjsRJdS)NrrDrGrFr;)r=r Z memory_usager2rPrrrr test_memory_usages  z*TestDataFrameSubclassing.test_memory_usagecCsrgd}gd}tjtjdd||d}tjtjdd|dd|d}|j|dddd }t|tjsnJdS) N)rr9r:r`r)ZoneZtwoZthreeZfourr+rErurTZkendall)rcZdropmethod)r=r rfrrZcorrwithr2rP)r r<r/rhriZcorrelsrrr test_corrwithsz&TestDataFrameSubclassing.test_corrwithcCsd}tjd|dd}tjtjtjtjgtjtjtjgtjtjtjgd|d}||dd}t|tjspJ||d}t|tjsJ|d}t|tjsJdS) Nrz1/1/1990Z53s)Zperiodsfreqrr;rz 1989-12-31) rZ date_ranger=r rfrgZasofr2rP)r NrngrBrrrr test_asofs  z"TestDataFrameSubclassing.test_asofcCs:tgdgdgdd}|}t|tjs6JdSr)r=r r r2rPrrrr test_idxmin_preserves_subclasssz7TestDataFrameSubclassing.test_idxmin_preserves_subclasscCs:tgdgdgdd}|}t|tjs6JdSr)r=r r r2rPrrrr test_idxmax_preserves_subclasssz7TestDataFrameSubclassing.test_idxmax_preserves_subclasscCsTtgdgdgdd}|}t|tjs6J|}t|t|sPJdSr)r=r Zconvert_dtypesr2type)r rrBrrrr &test_convert_dtypes_preserves_subclasss z?TestDataFrameSubclassing.test_convert_dtypes_preserves_subclasscCsBtdgdi}tdgdi}||s0J||s>JdS)Nrr)rr=r equals)r rhrirrr test_equals_subclasssz-TestDataFrameSubclassing.test_equals_subclassN)'rrrr5rCrVr[rnrqrtrwrrrrrrrrrrYmarkfilterwarningsrrrr Z parametrizer rrtdZskip_if_no_scipyrrrrrrrrrr rsB7# 9>'26  !      r)ZnumpyrfrYZpandas.util._test_decoratorsutilZ_test_decoratorsr!ZpandasrrrrrZpandas._testingZ_testingr=Zfixturerrrrrr s