def test_empty_kwargs_one_arg_case_insensitive(self):
def one_arg_func(B):
pass
filtered_kwargs = torchconfig.filter_args({},
one_arg_func,
ignore_cases=True)
assert len(filtered_kwargs) == 0
def test_empty_kwargs_one_arg(self):
def one_arg_func(A):
pass
filtered_kwargs = torchconfig.filter_args({},
one_arg_func,
ignore_cases=False)
assert len(filtered_kwargs) == 0
def test_empty_kwargs_no_args(self):
def no_args_func():
pass
filtered_kwargs = torchconfig.filter_args({},
no_args_func,
ignore_cases=False)
assert len(filtered_kwargs) == 0
def test_func_has_identical_args_case_insensitive(self):
def two_args_func(J, j):
pass
with pytest.raises(Exception):
filtered_kwargs = torchconfig.filter_args({},
two_args_func,
ignore_cases=True)
def test_kwargs_has_identical_keys_case_insensitive(self):
def no_args_func():
pass
with pytest.raises(Exception):
filtered_kwargs = torchconfig.filter_args({
"I": -5,
"i": -6
},
no_args_func,
ignore_cases=True)
def test_nonempty_kwargs_with_args(self):
def two_args_func(C, D):
pass
filtered_kwargs = torchconfig.filter_args({
"D": -1,
"E": -2
},
two_args_func,
ignore_cases=False)
assert len(filtered_kwargs) == 1
assert filtered_kwargs["D"] == -1
assert "C" not in filtered_kwargs
assert "E" not in filtered_kwargs
def test_nonempty_kwargs_with_args(self):
def two_args_func(F, g):
pass
filtered_kwargs = torchconfig.filter_args({
"G": -3,
"H": -4
},
two_args_func,
ignore_cases=True)
assert len(filtered_kwargs) == 1
assert filtered_kwargs["g"] == -3
assert "F" not in filtered_kwargs
assert "H" not in filtered_kwargs