MCPcopy Create free account
hub / github.com/apple/axlearn / test_insert_flags

Method test_insert_flags

axlearn/cli/utils_test.py:305–349  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

303 self.assertEqual(args.argv, [self.root_module, "--root_default=some_value"])
304
305 def test_insert_flags(self):
306 cases = [
307 (
308 [self.root_module],
309 [self.root_module, "--i", "100"],
310 ),
311 (
312 [self.root_module, "--a=1", "--b", "2"],
313 [self.root_module, "--a=1", "--b", "2", "--i", "100"],
314 ),
315 (
316 [self.root_module, "--a=1", "--b", "2", "--", self.root_module, "--c", "3"],
317 [
318 self.root_module,
319 "--a=1",
320 "--b",
321 "2",
322 "--i",
323 "100",
324 "--",
325 self.root_module,
326 "--c",
327 "3",
328 ],
329 ),
330 (
331 [self.root_module, "--", self.root_module, "--", self.root_module],
332 [
333 self.root_module,
334 "--i",
335 "100",
336 "--",
337 self.root_module,
338 "--",
339 self.root_module,
340 ],
341 ),
342 (
343 [self.root_module, "a", "b", " -- ", self.root_module],
344 [self.root_module, "a", "b", "--i", "100", " -- ", self.root_module],
345 ),
346 ]
347 for argv, expected in cases:
348 result = _insert_flags(argv, ["--i", "100"])
349 self.assertSequenceEqual(result, expected)
350
351 @parameterized.parameters([dict(argv=[]), dict(argv=["--flag", "value"])])
352 def test_insert_flags_argv0(self, argv: list[str]):

Callers

nothing calls this directly

Calls 1

_insert_flagsFunction · 0.90

Tested by

no test coverage detected