U fª/e‘ã@s:dZddlZddlmZmZddlmZGdd„dƒZ dS)z[ Tests for DataFrame cumulative operations See also -------- tests.series.test_cumulative éN)Ú DataFrameÚSeriesc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)ÚTestDataFrameCumulativeOpscCs0tt d¡ dd¡tdƒtdƒd}| ¡}dS)Nééé)ÚindexÚcolumns)rÚnpÚarangeZreshapeÚrangeÚcumsum)ÚselfÚdmÚresult©rúF/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/frame/test_cumulative.pyÚtest_cumsum_corners$z-TestDataFrameCumulativeOps.test_cumsum_cornercCsÐtj|jdd…df<tj|jdd…df<tj|jdd…df<| ¡}| tj¡}t ||¡|jdd}|jtjdd}t ||¡tdt  d ¡it  d ¡d }| ¡|jdd}t  |¡t  |¡ksÌt ‚dS© Nré rééé©ZaxisÚAr)r) r ÚnanÚilocr ÚapplyrÚtmÚassert_frame_equalrr ÚshapeÚAssertionError)rÚdatetime_framer ÚexpectedÚdfZ cumsum_xsrrrÚ test_cumsums     z&TestDataFrameCumulativeOps.test_cumsumcCsötj|jdd…df<tj|jdd…df<tj|jdd…df<| ¡}| tj¡}t ||¡|jdd}|jtjdd}t ||¡|jdd}t |¡t |¡ks¨t ‚|  d¡  t ¡}| d¡| d¡|  d¡  tj ¡}| d¡| d¡dS)Nrrrrrrr)r rrÚcumprodrrrrr r!ZfillnaZastypeÚintZint32)rr"r&r#Z cumprod_xsr$rrrÚ test_cumprod2s"        z'TestDataFrameCumulativeOps.test_cumprodcCsÐtj|jdd…df<tj|jdd…df<tj|jdd…df<| ¡}| tj¡}t ||¡|jdd}|jtjdd}t ||¡tdt  d ¡it  d ¡d }| ¡|jdd}t  |¡t  |¡ksÌt ‚dSr) r rrÚcumminrrrrrr r r!)rr"r)r#r$Z cummin_xsrrrÚ test_cumminOs     z&TestDataFrameCumulativeOps.test_cummincCsÐtj|jdd…df<tj|jdd…df<tj|jdd…df<| ¡}| tj¡}t ||¡|jdd}|jtjdd}t ||¡tdt  d ¡it  d ¡d }| ¡|jdd}t  |¡t  |¡ksÌt ‚dSr) r rrÚcummaxrrrrrr r r!)rr"r+r#r$Z cummax_xsrrrÚ test_cummaxfs     z&TestDataFrameCumulativeOps.test_cummaxcCsrtdddgdddgdddgdœƒ}| ¡}ttdddgtjd tdddgtjd |d  ¡dœƒ}t ||¡dS) Nrrég@TF)rÚBÚCé)Zdtyper/)rr rr Zint64Zfloat64rr)rr$rr#rrrÚ#test_cumulative_ops_preserve_dtypes}s" ýÿz>TestDataFrameCumulativeOps.test_cumulative_ops_preserve_dtypesN) Ú__name__Ú __module__Ú __qualname__rr%r(r*r,r1rrrrrs r) Ú__doc__Znumpyr ZpandasrrZpandas._testingZ_testingrrrrrrÚs