Python cli_main Exemples

Exemple #1

Fichier : test_examples.py Projet : 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()

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