python-by-example-150-chall.../venv/lib/python3.6/site-packages/pandas/tests/series/test_validate.py

21 lines
668 B
Python
Raw Permalink Normal View History

2019-08-04 20:26:35 +08:00
import pytest
class TestSeriesValidate:
"""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, string_series, func, inplace):
msg = 'For argument "inplace" expected type bool'
kwargs = dict(inplace=inplace)
if func == "_set_name":
kwargs["name"] = "hello"
with pytest.raises(ValueError, match=msg):
getattr(string_series, func)(**kwargs)