def test_cli_sample(): with tests.chdir(tests.data_path("pico")): cli.main("sample model.json sampler.json".split()) cli.main("--corpus-dir model.json".split()) cli.main("--model-dir model.json".split()) cli.main("--sampler-dir model.json sampler.json".split()) cli.main("ls files model.json sampler.json".split())
def main(): parser = cli.ArgumentParser(description=__description__) parser.add_argument("dataset", help="path to output dataset") parser.add_argument("inputs", nargs='*', help="path to input datasets") args = parser.parse_args() cli.main(merge, args.dataset, args.inputs)
def test_cli_train(): with tests.chdir(tests.data_path("pico")): cli.main("train model.json".split()) cli.main("--corpus-dir model.json".split()) cli.main("--model-dir model.json".split())
def test_cli(): fs.rm("kernels.db") cli.main("db init kernels.db".split()) assert fs.exists("kernels.db") corpus_path = tests.archive("tiny", "corpus") cli.main("db explore kernels.db".split()) cli.main(f"fetch fs kernels.db {corpus_path}".split()) cli.main("preprocess kernels.db".split()) cli.main("db explore kernels.db".split()) fs.rm("kernels_out") cli.main("db dump kernels.db -d kernels_out".split()) assert fs.isdir("kernels_out") assert len(fs.ls("kernels_out")) >= 1 fs.rm("kernels.cl") cli.main("db dump kernels.db kernels.cl --file-sep --eof --reverse".split()) assert fs.isfile("kernels.cl") fs.rm("kernels_out") cli.main("db dump kernels.db --input-samples -d kernels_out".split()) assert fs.isdir("kernels_out") assert len(fs.ls("kernels_out")) == 250 fs.rm("kernels.db") fs.rm("kernels_out")
def test_cli_test_coveragerc_path(): with pytest.raises(SystemExit): cli.main("test --coveragerc-path".split())
def test_cli_test_cache_path(): with pytest.raises(SystemExit): cli.main("test --cache-path".split())
def test_cli_version(): with pytest.raises(SystemExit): cli.main("--version")
def test_cli_ls(): cli.main("ls models".split()) cli.main("ls samplers".split())