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
|
from typing import List, Optional, Set
import pytest
@pytest.mark.parametrize(
"cmd_str,expected",
[
("", None),
("--empty-my-list", []),
("--empty-my-list=True", []),
("--empty-my-list=False", None),
],
)
def test_optional_list_empty_flag_default(app, cmd_str, expected, assert_parse_args):
@app.default
def foo(my_list: Optional[List[int]] = None):
pass
if expected is None:
assert_parse_args(foo, cmd_str)
else:
assert_parse_args(foo, cmd_str, expected)
@pytest.mark.parametrize(
"cmd_str,expected",
[
("", None),
("--empty-my-set", set()),
("--empty-my-set=True", set()),
("--empty-my-set=False", None),
],
)
def test_optional_set_empty_flag_default(app, cmd_str, expected, assert_parse_args):
@app.default
def foo(my_set: Optional[Set[int]] = None):
pass
if expected is None:
assert_parse_args(foo, cmd_str)
else:
assert_parse_args(foo, cmd_str, expected)
|
