1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
| import pytest
|
| from pandas.core.frame import DataFrame
|
|
| @pytest.fixture
| def dataframe():
| return DataFrame({"a": [1, 2], "b": [3, 4]})
|
|
| class TestDataFrameValidate:
| """Tests for error handling related to data types of method arguments."""
|
| @pytest.mark.parametrize(
| "func",
| [
| "query",
| "eval",
| "set_index",
| "reset_index",
| "dropna",
| "drop_duplicates",
| "sort_values",
| ],
| )
| @pytest.mark.parametrize("inplace", [1, "True", [1, 2, 3], 5.0])
| def test_validate_bool_args(self, dataframe, func, inplace):
| msg = 'For argument "inplace" expected type bool'
| kwargs = {"inplace": inplace}
|
| if func == "query":
| kwargs["expr"] = "a > b"
| elif func == "eval":
| kwargs["expr"] = "a + b"
| elif func == "set_index":
| kwargs["keys"] = ["a"]
| elif func == "sort_values":
| kwargs["by"] = ["a"]
|
| with pytest.raises(ValueError, match=msg):
| getattr(dataframe, func)(**kwargs)
|
|
