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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
from osrf_pycommon.cli_utils import common
def test_extract_jobs_flags():
extract_jobs_flags = common.extract_jobs_flags
valid_mflags = [
'-j8 -l8', 'j8 ', '-j', 'j', '-l8', 'l8',
'-l', 'l', '-j18', ' -j8 l9', '-j1 -l1',
'--jobs=8', '--jobs 8', '--jobs', '--load-average',
'--load-average=8', '--load-average 8', '--jobs=8 -l9'
]
results = [
'-j8 -l8', 'j8', '-j', 'j', '-l8', 'l8',
'-l', 'l', '-j18', '-j8 l9', '-j1 -l1',
'--jobs=8', '--jobs 8', '--jobs', '--load-average',
'--load-average=8', '--load-average 8', '--jobs=8 -l9'
]
leftover_results = [
'', ' ', '', '', '', '',
'', '', '', '', '',
'', '', '', '',
'', '', ''
]
for mflag, result in zip(valid_mflags, zip(results, leftover_results)):
result, leftover_result = result
matches, leftovers = extract_jobs_flags(mflag)
print('--')
print("input: '{0}'".format(mflag))
print("matched: '{0}'".format(matches))
print("expected: '{0}'".format(result))
assert matches == result, "should match '{0}'".format(mflag)
assert leftovers == leftover_result, "expected leftovers '{0}': '{1}'"\
.format(leftovers)
invalid_mflags = ['', '--jobs= 8', '--jobs8']
for mflag in invalid_mflags:
matches, leftovers = extract_jobs_flags(mflag)
assert matches == '', "should not match '{0}'".format(mflag)
assert leftovers == mflag, "'{0}' should be in leftovers: '{1}'"\
.format(mflag, leftovers)
mixed_flags = [
'target -j8 -l8 --other-option'
]
results = [
('-j8 -l8', 'target --other-option')
]
for args, result in zip(mixed_flags, results):
expected_matches, expected_leftovers = result
matches, leftovers = extract_jobs_flags(args)
assert matches == expected_matches, "should have matched '{0}'"\
.format(expected_matches)
assert leftovers == expected_leftovers, "should have left '{0}' '{1}'"\
.format(expected_leftovers, leftovers)
def test_extract_argument_group():
extract_argument_group = common.extract_argument_group
# Example 1 from docstring
args = ['foo', '--args', 'bar', '--baz']
expected = (['foo'], ['bar', '--baz'])
results = extract_argument_group(args, '--args')
assert expected == results, (args, expected, results)
# Example 2 from docstring
args = ['foo', '--args', 'bar', '--baz', '---', '--', '--foo-option']
expected = (['foo', '--foo-option'], ['bar', '--baz', '--'])
results = extract_argument_group(args, '--args')
assert expected == results, (args, expected, results)
# Example 3 from docstring
args = ['foo',
'--args', 'ping', '--',
'bar',
'--args', 'pong', '--',
'baz',
'--args', '--']
expected = (['foo', 'bar', 'baz'], ['ping', 'pong'])
results = extract_argument_group(args, '--args')
assert expected == results, (args, expected, results)
# Example with delimiter but no arguments
args = ['foo', '--args']
expected = (['foo'], [])
results = extract_argument_group(args, '--args')
assert expected == results, (args, expected, results)
# Example with no delimiter
args = ['foo', 'bar']
expected = (['foo', 'bar'], [])
results = extract_argument_group(args, '--args')
assert expected == results, (args, expected, results)
|
