Python cli_main示例

示例#1

文件： test_examples.py 项目： zarzarj/pytorch-lightning

def test_examples_mnist_dali(tmpdir, cli_args):
    from pl_examples.basic_examples.dali_image_classifier import cli_main

    # update the temp dir
    cli_args = cli_args % {'tmpdir': tmpdir}
    with mock.patch("argparse._sys.argv", ["any.py"] + cli_args.strip().split()):
        cli_main()

示例#2

def test_examples_mnist_dali(cli_args):
    from pl_examples.basic_examples.dali_image_classifier import cli_main

    with mock.patch("argparse._sys.argv",
                    ["any.py"] + cli_args.strip().split()):
        cli_main()