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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
from typing import List, Literal, Optional, Union
import pytest
from argh.assembling import TypingHintArgSpecGuesser
@pytest.mark.parametrize("arg_type", TypingHintArgSpecGuesser.BASIC_TYPES)
def test_simple_types(arg_type):
guess = TypingHintArgSpecGuesser.typing_hint_to_arg_spec_params
# just the basic type
assert guess(arg_type) == {"type": arg_type}
# basic type or None
assert guess(Optional[arg_type]) == {
"type": arg_type,
"required": False,
}
assert guess(Union[None, arg_type]) == {"required": False}
# multiple basic types: the first one is used and None is looked up
assert guess(Union[arg_type, str, None]) == {
"type": arg_type,
"required": False,
}
assert guess(Union[str, arg_type, None]) == {
"type": str,
"required": False,
}
def test_list():
guess = TypingHintArgSpecGuesser.typing_hint_to_arg_spec_params
assert guess(list) == {"nargs": "*"}
assert guess(List) == {"nargs": "*"}
assert guess(Optional[list]) == {"nargs": "*", "required": False}
assert guess(Optional[List]) == {"nargs": "*", "required": False}
assert guess(List[str]) == {"nargs": "*", "type": str}
assert guess(List[int]) == {"nargs": "*", "type": int}
assert guess(Optional[List[str]]) == {"nargs": "*", "type": str, "required": False}
assert guess(Optional[List[tuple]]) == {"nargs": "*", "required": False}
assert guess(List[list]) == {"nargs": "*"}
assert guess(List[tuple]) == {"nargs": "*"}
def test_literal():
guess = TypingHintArgSpecGuesser.typing_hint_to_arg_spec_params
assert guess(Literal["a"]) == {"choices": ("a",), "type": str}
assert guess(Literal["a", "b"]) == {"choices": ("a", "b"), "type": str}
assert guess(Literal[1]) == {"choices": (1,), "type": int}
@pytest.mark.parametrize("arg_type", (dict, tuple))
def test_unusable_types(arg_type):
guess = TypingHintArgSpecGuesser.typing_hint_to_arg_spec_params
assert guess(arg_type) == {}
|
