test_validate_args.py 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. import pytest
  2. from pandas.util._validators import validate_args
  3. _fname = "func"
  4. def test_bad_min_fname_arg_count():
  5. msg = "'max_fname_arg_count' must be non-negative"
  6. with pytest.raises(ValueError, match=msg):
  7. validate_args(_fname, (None,), -1, "foo")
  8. def test_bad_arg_length_max_value_single():
  9. args = (None, None)
  10. compat_args = ("foo",)
  11. min_fname_arg_count = 0
  12. max_length = len(compat_args) + min_fname_arg_count
  13. actual_length = len(args) + min_fname_arg_count
  14. msg = (
  15. rf"{_fname}\(\) takes at most {max_length} "
  16. rf"argument \({actual_length} given\)"
  17. )
  18. with pytest.raises(TypeError, match=msg):
  19. validate_args(_fname, args, min_fname_arg_count, compat_args)
  20. def test_bad_arg_length_max_value_multiple():
  21. args = (None, None)
  22. compat_args = {"foo": None}
  23. min_fname_arg_count = 2
  24. max_length = len(compat_args) + min_fname_arg_count
  25. actual_length = len(args) + min_fname_arg_count
  26. msg = (
  27. rf"{_fname}\(\) takes at most {max_length} "
  28. rf"arguments \({actual_length} given\)"
  29. )
  30. with pytest.raises(TypeError, match=msg):
  31. validate_args(_fname, args, min_fname_arg_count, compat_args)
  32. @pytest.mark.parametrize("i", range(1, 3))
  33. def test_not_all_defaults(i):
  34. bad_arg = "foo"
  35. msg = (
  36. f"the '{bad_arg}' parameter is not supported "
  37. rf"in the pandas implementation of {_fname}\(\)"
  38. )
  39. compat_args = {"foo": 2, "bar": -1, "baz": 3}
  40. arg_vals = (1, -1, 3)
  41. with pytest.raises(ValueError, match=msg):
  42. validate_args(_fname, arg_vals[:i], 2, compat_args)
  43. def test_validation():
  44. # No exceptions should be raised.
  45. validate_args(_fname, (None,), 2, {"out": None})
  46. compat_args = {"axis": 1, "out": None}
  47. validate_args(_fname, (1, None), 2, compat_args)