123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- import pytest
- from pandas.util._validators import validate_args_and_kwargs
- _fname = "func"
- def test_invalid_total_length_max_length_one():
- compat_args = ("foo",)
- kwargs = {"foo": "FOO"}
- args = ("FoO", "BaZ")
- min_fname_arg_count = 0
- max_length = len(compat_args) + min_fname_arg_count
- actual_length = len(kwargs) + len(args) + min_fname_arg_count
- msg = (
- rf"{_fname}\(\) takes at most {max_length} "
- rf"argument \({actual_length} given\)"
- )
- with pytest.raises(TypeError, match=msg):
- validate_args_and_kwargs(_fname, args, kwargs, min_fname_arg_count, compat_args)
- def test_invalid_total_length_max_length_multiple():
- compat_args = ("foo", "bar", "baz")
- kwargs = {"foo": "FOO", "bar": "BAR"}
- args = ("FoO", "BaZ")
- min_fname_arg_count = 2
- max_length = len(compat_args) + min_fname_arg_count
- actual_length = len(kwargs) + len(args) + min_fname_arg_count
- msg = (
- rf"{_fname}\(\) takes at most {max_length} "
- rf"arguments \({actual_length} given\)"
- )
- with pytest.raises(TypeError, match=msg):
- validate_args_and_kwargs(_fname, args, kwargs, min_fname_arg_count, compat_args)
- @pytest.mark.parametrize("args,kwargs", [((), {"foo": -5, "bar": 2}), ((-5, 2), {})])
- def test_missing_args_or_kwargs(args, kwargs):
- bad_arg = "bar"
- min_fname_arg_count = 2
- compat_args = {"foo": -5, bad_arg: 1}
- msg = (
- rf"the '{bad_arg}' parameter is not supported "
- rf"in the pandas implementation of {_fname}\(\)"
- )
- with pytest.raises(ValueError, match=msg):
- validate_args_and_kwargs(_fname, args, kwargs, min_fname_arg_count, compat_args)
- def test_duplicate_argument():
- min_fname_arg_count = 2
- compat_args = {"foo": None, "bar": None, "baz": None}
- kwargs = {"foo": None, "bar": None}
- args = (None,) # duplicate value for "foo"
- msg = rf"{_fname}\(\) got multiple values for keyword argument 'foo'"
- with pytest.raises(TypeError, match=msg):
- validate_args_and_kwargs(_fname, args, kwargs, min_fname_arg_count, compat_args)
- def test_validation():
- # No exceptions should be raised.
- compat_args = {"foo": 1, "bar": None, "baz": -2}
- kwargs = {"baz": -2}
- args = (1, None)
- min_fname_arg_count = 2
- validate_args_and_kwargs(_fname, args, kwargs, min_fname_arg_count, compat_args)
|