28 lines
859 B
Python
28 lines
859 B
Python
|
from pandas.core.series import Series
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
import pandas.util.testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def series():
|
||
|
|
return Series([1, 2, 3, 4, 5])
|
||
|
|
|
||
|
|
|
||
|
|
class TestSeriesValidate(object):
|
||
|
|
"""Tests for error handling related to data types of method arguments."""
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("func", ["reset_index", "_set_name",
|
||
|
|
"sort_values", "sort_index",
|
||
|
|
"rename", "dropna"])
|
||
|
|
@pytest.mark.parametrize("inplace", [1, "True", [1, 2, 3], 5.0])
|
||
|
|
def test_validate_bool_args(self, series, func, inplace):
|
||
|
|
msg = "For argument \"inplace\" expected type bool"
|
||
|
|
kwargs = dict(inplace=inplace)
|
||
|
|
|
||
|
|
if func == "_set_name":
|
||
|
|
kwargs["name"] = "hello"
|
||
|
|
|
||
|
|
with tm.assert_raises_regex(ValueError, msg):
|
||
|
|
getattr(series, func)(**kwargs)
|