Python cli_main Examples

Example #1

File: test_examples.py Project: 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()

Example #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()