Skip to content

Commit 1830fc7

Browse files
committed
Unify inputdata_root arg flags.
1 parent 9bbea4e commit 1830fc7

4 files changed

Lines changed: 19 additions & 2 deletions

File tree

relink.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,10 @@ def parse_arguments():
360360
# ONLY INTENDED FOR USE IN TESTING
361361
parser.add_argument(
362362
"--inputdata-root",
363-
"-inputdata", # to match rimport
363+
"-inputdata-root",
364+
"--inputdata",
365+
"-inputdata",
366+
"-i",
364367
type=validate_directory,
365368
default=DEFAULT_INPUTDATA_ROOT,
366369
help=argparse.SUPPRESS,

rimport

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,11 @@ def build_parser() -> argparse.ArgumentParser:
7070
)
7171

7272
parser.add_argument(
73+
"--inputdata-root",
74+
"-inputdata-root",
7375
"--inputdata",
7476
"-inputdata",
77+
"-i",
7578
dest="inputdata_root",
7679
metavar="inputdata_dir",
7780
default=DEFAULT_INPUTDATA_ROOT,

tests/relink/test_args.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,17 @@ def test_multiple_source_roots_with_target(self, temp_dirs):
193193
assert str(source2.resolve()) in args.items_to_process
194194
assert args.target_root == str(target.resolve())
195195

196+
@pytest.mark.parametrize(
197+
"inputdata_flag",
198+
["-inputdata", "-i", "--inputdata", "--inputdata-root", "-inputdata-root"],
199+
)
200+
def test_inputdata_arguments_accepted(self, temp_dirs, inputdata_flag):
201+
"""Test that all inputdata argument flags are accepted."""
202+
inputdata_root, _ = temp_dirs
203+
with patch("sys.argv", ["relink.py", inputdata_flag, inputdata_root]):
204+
args = relink.parse_arguments()
205+
assert args.inputdata_root == inputdata_root
206+
196207

197208
class TestValidateDirectory:
198209
"""Test suite for validate_directory function."""

tests/rimport/test_build_parser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_list_arguments_accepted(self, list_flag):
4848
assert args.filelist == "files.txt"
4949
assert args.file is None
5050

51-
@pytest.mark.parametrize("inputdata_flag", ["-inputdata", "-i", "--inputdata"])
51+
@pytest.mark.parametrize("inputdata_flag", ["-inputdata", "-i", "--inputdata", "--inputdata-root", "-inputdata-root"])
5252
def test_inputdata_arguments_accepted(self, inputdata_flag):
5353
"""Test that all inputdata argument flags are accepted."""
5454
parser = rimport.build_parser()

0 commit comments

Comments
 (0)