def test_cli_run_self_supervised_swav(cli_args):
"""Test running CLI for an example with default params."""
from pl_bolts.models.self_supervised.swav.swav_module import cli_main
cli_args = cli_args.split(' ') if cli_args else []
with mock.patch("argparse._sys.argv", ["any.py"] + cli_args):
cli_main()
def test_cli_run_ssl_cpc(cli_args, dataset_name):
from pl_bolts.models.self_supervised.cpc.cpc_module import cli_main
cli_args = cli_args % {"dataset_name": dataset_name}
cli_args = cli_args.strip().split(" ") if cli_args else []
with mock.patch("argparse._sys.argv", ["any.py"] + cli_args):
cli_main()
