U
|
ý°d¦ ã @ sB d Z ddlZddlZddlmZmZ ddlmZ G dd dZ
|
dS )z[
|
Tests for DataFrame cumulative operations
|
|
See also
|
--------
|
tests.series.test_cumulative
|
é N)Ú DataFrameÚSeriesc @ sB e Zd Zdd Zdd Zej ddddd g¡d
|
d Zd d Z dS )ÚTestDataFrameCumulativeOpsc C sd t dt d¡it d¡d}| ¡ | ¡ | ¡ t t d¡ dd¡tdtdd}| ¡ d S )NÚAé )Úindexé é )r Úcolumns)r ÚnpZarangeÚcummaxÚcumminÚcumsumZreshapeÚrange)ÚselfÚdfÚdm© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_cumulative.pyÚtest_cumulative_ops_smoke s $z4TestDataFrameCumulativeOps.test_cumulative_ops_smokec C s t j|jdd
df< t j|jdd
df< t j|jdd
df< | d¡ t¡}| d¡ | d¡ | d¡ t j¡}| d¡ | d¡ d S )Nr é
|
r é é é )r ÚnanÚilocZfillnaZastypeÚintÚcumprodÚint32)r Údatetime_framer r r r Útest_cumprod_smoke" s
|
|
|
z-TestDataFrameCumulativeOps.test_cumprod_smokeÚmethodr r r r c C s° t j|jdd
df< t j|jdd
df< t j|jdd
df< t|| }| tt|¡}t ||¡ t||dd}|jtt|dd}t ||¡ t |¡t |¡ks¬t d S )Nr r r r r r )Zaxis)
|
r r r ÚgetattrÚapplyr ÚtmÚassert_frame_equalÚshapeÚAssertionError)r r r! ÚresultÚexpectedr r r Ú&test_cumulative_ops_match_series_apply1 s zATestDataFrameCumulativeOps.test_cumulative_ops_match_series_applyc C sr t dddgdddgdddgd}| ¡ }t tdddgtjd tdddgtjd |d
|
¡ d}t ||¡ d S ) Nr r é g @TF)r ÚBÚCé )Zdtyper- )r r r r Úint64Úfloat64r$ r% )r r r( r) r r r Útest_cumsum_preserve_dtypesD s "
|
ýÿz6TestDataFrameCumulativeOps.test_cumsum_preserve_dtypesN)
|
Ú__name__Ú
|
__module__Ú __qualname__r r ÚpytestÚmarkZ parametrizer* r1 r r r r r s
|
|
r ) Ú__doc__Únumpyr r5 Zpandasr r Zpandas._testingZ_testingr$ r r r r r Ú<module> s
|
|
