91 lines
1.5 KiB
Python
91 lines
1.5 KiB
Python
from textwrap import dedent
|
|
|
|
from pandas.util._decorators import doc
|
|
|
|
|
|
@doc(method="cumsum", operation="sum")
|
|
def cumsum(whatever):
|
|
"""
|
|
This is the {method} method.
|
|
|
|
It computes the cumulative {operation}.
|
|
"""
|
|
|
|
|
|
@doc(
|
|
cumsum,
|
|
dedent(
|
|
"""
|
|
Examples
|
|
--------
|
|
|
|
>>> cumavg([1, 2, 3])
|
|
2
|
|
"""
|
|
),
|
|
method="cumavg",
|
|
operation="average",
|
|
)
|
|
def cumavg(whatever):
|
|
pass
|
|
|
|
|
|
@doc(cumsum, method="cummax", operation="maximum")
|
|
def cummax(whatever):
|
|
pass
|
|
|
|
|
|
@doc(cummax, method="cummin", operation="minimum")
|
|
def cummin(whatever):
|
|
pass
|
|
|
|
|
|
def test_docstring_formatting():
|
|
docstr = dedent(
|
|
"""
|
|
This is the cumsum method.
|
|
|
|
It computes the cumulative sum.
|
|
"""
|
|
)
|
|
assert cumsum.__doc__ == docstr
|
|
|
|
|
|
def test_docstring_appending():
|
|
docstr = dedent(
|
|
"""
|
|
This is the cumavg method.
|
|
|
|
It computes the cumulative average.
|
|
|
|
Examples
|
|
--------
|
|
|
|
>>> cumavg([1, 2, 3])
|
|
2
|
|
"""
|
|
)
|
|
assert cumavg.__doc__ == docstr
|
|
|
|
|
|
def test_doc_template_from_func():
|
|
docstr = dedent(
|
|
"""
|
|
This is the cummax method.
|
|
|
|
It computes the cumulative maximum.
|
|
"""
|
|
)
|
|
assert cummax.__doc__ == docstr
|
|
|
|
|
|
def test_inherit_doc_template():
|
|
docstr = dedent(
|
|
"""
|
|
This is the cummin method.
|
|
|
|
It computes the cumulative minimum.
|
|
"""
|
|
)
|
|
assert cummin.__doc__ == docstr
|