TST: Update test cases

metagentools · Jun 7, 2024 · 991125a · 991125a
1 parent e2c4056
commit 991125a
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -68,4 +68,34 @@ def test_visualise_run(runner, tmp_dir):
     paths = DATADIR / "Sim-5G+metaSPAdes" / "contigs.paths"
     args = f"--assembler spades --initial {initial} --final {final} --graph {graph} --paths {paths} --output {outpath}".split()
     r = runner.invoke(visualise, args, catch_exceptions=False)
+    assert r.exit_code == 0, r.output
+
+def test_visualise_prefix(runner, tmp_dir):
+    outpath = tmp_dir
+    initial = DATADIR / "Sim-5G+metaSPAdes" / "metacoag_res.csv"
+    final = DATADIR / "Sim-5G+metaSPAdes" / "graphbin_res.csv"
+    graph = DATADIR / "Sim-5G+metaSPAdes" / "assembly_graph_with_scaffolds.gfa"
+    paths = DATADIR / "Sim-5G+metaSPAdes" / "contigs.paths"
+    args = f"--assembler spades --initial {initial} --final {final} --graph {graph} --paths {paths} --prefix test --output {outpath}".split()
+    r = runner.invoke(visualise, args, catch_exceptions=False)
+    assert r.exit_code == 0, r.output
+
+def test_visualise_type(runner, tmp_dir):
+    outpath = tmp_dir
+    initial = DATADIR / "Sim-5G+metaSPAdes" / "metacoag_res.csv"
+    final = DATADIR / "Sim-5G+metaSPAdes" / "graphbin_res.csv"
+    graph = DATADIR / "Sim-5G+metaSPAdes" / "assembly_graph_with_scaffolds.gfa"
+    paths = DATADIR / "Sim-5G+metaSPAdes" / "contigs.paths"
+    args = f"--assembler spades --initial {initial} --final {final} --graph {graph} --paths {paths} --type svg --output {outpath}".split()
+    r = runner.invoke(visualise, args, catch_exceptions=False)
+    assert r.exit_code == 0, r.output
+
+def test_visualise_outdir(runner, tmp_dir):
+    outpath = tmp_dir / "testing"
+    initial = DATADIR / "Sim-5G+metaSPAdes" / "metacoag_res.csv"
+    final = DATADIR / "Sim-5G+metaSPAdes" / "graphbin_res.csv"
+    graph = DATADIR / "Sim-5G+metaSPAdes" / "assembly_graph_with_scaffolds.gfa"
+    paths = DATADIR / "Sim-5G+metaSPAdes" / "contigs.paths"
+    args = f"--assembler spades --initial {initial} --final {final} --graph {graph} --paths {paths} --output {outpath}".split()
+    r = runner.invoke(visualise, args, catch_exceptions=False)
     assert r.exit_code == 0, r.output