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