21
21
from unittest .mock import MagicMock , patch
22
22
23
23
from torchx .cli .argparse_util import ArgOnceAction , torchxconfig
24
-
25
24
from torchx .cli .cmd_run import _parse_component_name_and_args , CmdBuiltins , CmdRun
26
25
from torchx .schedulers .local_scheduler import SignalException
27
26
@@ -216,38 +215,41 @@ def test_store_experiment_id(self, mock_runner_run: MagicMock) -> None:
216
215
self .assertEqual (call_kwargs ["parent_run_id" ], "experiment_1" )
217
216
218
217
def test_parse_component_name_and_args_no_default (self ) -> None :
218
+ # set dirs to test tmpdir so tests don't accidentally pick up user's $HOME/.torchxconfig
219
+ dirs = [str (self .tmpdir )]
220
+
219
221
sp = argparse .ArgumentParser (prog = "test" )
220
222
self .assertEqual (
221
223
("utils.echo" , []),
222
- _parse_component_name_and_args (["utils.echo" ], sp ),
224
+ _parse_component_name_and_args (["utils.echo" ], sp , dirs ),
223
225
)
224
226
self .assertEqual (
225
227
("utils.echo" , []),
226
- _parse_component_name_and_args (["--" , "utils.echo" ], sp ),
228
+ _parse_component_name_and_args (["--" , "utils.echo" ], sp , dirs ),
227
229
)
228
230
self .assertEqual (
229
231
("utils.echo" , ["--msg" , "hello" ]),
230
- _parse_component_name_and_args (["utils.echo" , "--msg" , "hello" ], sp ),
232
+ _parse_component_name_and_args (["utils.echo" , "--msg" , "hello" ], sp , dirs ),
231
233
)
232
234
233
235
self .assertEqual (
234
236
("utils.echo" , ["--msg" , "hello" , "--" , "--" ]),
235
237
_parse_component_name_and_args (
236
- ["utils.echo" , "--msg" , "hello" , "--" , "--" ], sp
238
+ ["utils.echo" , "--msg" , "hello" , "--" , "--" ], sp , dirs
237
239
),
238
240
)
239
241
240
242
self .assertEqual (
241
243
("utils.echo" , ["--msg" , "hello" , "-" , "-" ]),
242
244
_parse_component_name_and_args (
243
- ["utils.echo" , "--msg" , "hello" , "-" , "-" ], sp
245
+ ["utils.echo" , "--msg" , "hello" , "-" , "-" ], sp , dirs
244
246
),
245
247
)
246
248
247
249
self .assertEqual (
248
250
("utils.echo" , ["--msg" , "hello" , "- " , "- " ]),
249
251
_parse_component_name_and_args (
250
- ["utils.echo" , "--msg" , "hello" , "- " , "- " ], sp
252
+ ["utils.echo" , "--msg" , "hello" , "- " , "- " ], sp , dirs
251
253
),
252
254
)
253
255
@@ -274,32 +276,35 @@ def test_parse_component_name_and_args_no_default(self) -> None:
274
276
"-m" ,
275
277
],
276
278
sp ,
279
+ dirs ,
277
280
),
278
281
)
279
282
280
283
with self .assertRaises (SystemExit ):
281
- _parse_component_name_and_args (["--" ], sp )
284
+ _parse_component_name_and_args (["--" ], sp , dirs )
282
285
283
286
with self .assertRaises (SystemExit ):
284
- _parse_component_name_and_args (["--msg" , "hello" ], sp )
287
+ _parse_component_name_and_args (["--msg" , "hello" ], sp , dirs )
285
288
286
289
with self .assertRaises (SystemExit ):
287
- _parse_component_name_and_args (["-m" , "hello" ], sp )
290
+ _parse_component_name_and_args (["-m" , "hello" ], sp , dirs )
288
291
289
292
with self .assertRaises (SystemExit ):
290
- _parse_component_name_and_args (["-m" , "hello" , "-m" , "repeate" ], sp )
293
+ _parse_component_name_and_args (["-m" , "hello" , "-m" , "repeate" ], sp , dirs )
291
294
292
295
with self .assertRaises (SystemExit ):
293
- _parse_component_name_and_args (["--msg" , "hello" , "--msg" , "repeate" ], sp )
296
+ _parse_component_name_and_args (
297
+ ["--msg" , "hello" , "--msg" , "repeate" ], sp , dirs
298
+ )
294
299
295
300
with self .assertRaises (SystemExit ):
296
301
_parse_component_name_and_args (
297
- ["--msg " , "hello" , "--msg " , "repeate" ], sp
302
+ ["--msg " , "hello" , "--msg " , "repeate" ], sp , dirs
298
303
)
299
304
300
305
with self .assertRaises (SystemExit ):
301
306
_parse_component_name_and_args (
302
- ["--m" , "hello" , "--" , "--msg" , "msg" , "--msg" , "repeate" ], sp
307
+ ["--m" , "hello" , "--" , "--msg" , "msg" , "--msg" , "repeate" ], sp , dirs
303
308
)
304
309
305
310
def test_parse_component_name_and_args_with_default (self ) -> None :
0 commit comments