@@ -23,9 +23,12 @@ def test_set_output_directory(inputs, expected, tmp_path):
23
23
existing_dir = Path (tmp_path ).resolve () / "existing_dir"
24
24
existing_dir .mkdir (parents = True , exist_ok = True )
25
25
26
- expected_output_directory = Path (tmp_path ).resolve () / expected [0 ]
27
- actual_args = argparse .Namespace (output_directory = inputs [0 ])
26
+ actual_parser = argparse .ArgumentParser ()
27
+ actual_parser .add_argument ("--output_directory" )
28
+ actual_args = actual_parser .parse_args (["--output_directory" , inputs [0 ]])
28
29
actual_args .output_directory = set_output_directory (actual_args )
30
+
31
+ expected_output_directory = Path (tmp_path ).resolve () / expected [0 ]
29
32
assert actual_args .output_directory == expected_output_directory
30
33
assert Path (actual_args .output_directory ).exists ()
31
34
assert Path (actual_args .output_directory ).is_dir ()
@@ -38,7 +41,10 @@ def test_set_output_directory_bad(tmp_path):
38
41
existing_file = Path (tmp_path ).resolve () / "existing_file.py"
39
42
existing_file .touch ()
40
43
41
- actual_args = argparse .Namespace (output_directory = "existing_file.py" )
44
+ actual_parser = argparse .ArgumentParser ()
45
+ actual_parser .add_argument ("--output_directory" )
46
+ actual_args = actual_parser .parse_args (["--output_directory" , "existing_file.py" ])
47
+
42
48
with pytest .raises (FileExistsError ):
43
49
actual_args .output_directory = set_output_directory (actual_args )
44
50
assert Path (actual_args .output_directory ).exists ()
@@ -60,9 +66,12 @@ def test_set_input_directory(inputs, expected, tmp_path):
60
66
existing_dir = Path (tmp_path ).resolve () / "existing_dir"
61
67
existing_dir .mkdir (parents = True , exist_ok = True )
62
68
63
- expected_input_directory = Path (tmp_path ).resolve () / expected [0 ]
64
- actual_args = argparse .Namespace (input_file = inputs [0 ])
69
+ actual_parser = argparse .ArgumentParser ()
70
+ actual_parser .add_argument ("--input_file" )
71
+ actual_args = actual_parser .parse_args (["--input_file" , inputs [0 ]])
65
72
actual_args = set_input_directory (actual_args )
73
+
74
+ expected_input_directory = Path (tmp_path ).resolve () / expected [0 ]
66
75
assert actual_args .input_directory == expected_input_directory
67
76
68
77
0 commit comments