def test_is_wanted_file_caps(extension):
assert files._is_wanted_file(f"test{extension.upper()}") is True
def test_is_wanted_file_not():
assert files._is_wanted_file(f"test.py") is False
def test_is_wanted_file(extension):
assert files._is_wanted_file(f"test{extension}") is True