Add tests for depth()

jcvi/graphics/landscape.py

@@ -597,6 +597,7 @@ def depth(args):
 
     image_name = opts.outfile
     savefig(image_name, dpi=iopts.dpi, iopts=iopts)
+    return image_name
 
 
 def add_window_options(p):

tests/graphics/data/chrinfo.txt

@@ -0,0 +1,12 @@
+chr01A, #c51b7d, 1A
+chr01B, #4d9221, 1B
+chr02A, #c51b7d, 2A
+chr02B, #4d9221, 2B
+chr03A, #c51b7d, 3A
+chr03B, #4d9221, 3B
+chr04A, #c51b7d, 4A
+chr04B, #4d9221, 4B
+chr05A, #c51b7d, 5A
+chr05B, #4d9221, 5B
+chr06A, #c51b7d, 6A
+chr06B, #4d9221, 6B

tests/graphics/data/titleinfo.txt

@@ -0,0 +1,3 @@
+VAR0_srtd.wgs.regions.bed.gz, *S. species*, ‘Variety 1’
+VAR1_srtd.wgs.regions.bed.gz, *S. species*, ‘Variety 2’
+VAR2_srtd.wgs.regions.bed.gz, *S. species*, ‘Variety 3’

tests/graphics/test_landscape.py

@@ -2,7 +2,25 @@
 import os.path as op
 
 from jcvi.apps.base import cleanup
-from jcvi.graphics.landscape import stack
+from jcvi.graphics.landscape import depth, stack
+
+
+def test_depth():
+    cwd = os.getcwd()
+    os.chdir(op.join(op.dirname(__file__), "data"))
+    cleanup("depth.pdf")
+    image_name = depth(
+        [
+            "VAR0_srtd.wgs.regions.bed.gz",
+            "VAR1_srtd.wgs.regions.bed.gz",
+            "VAR2_srtd.wgs.regions.bed.gz",
+            "--chrinfo=chrinfo.txt",
+            "--titleinfo=titleinfo.txt",
+            "--figsize=11x7",
+        ]
+    )
+    assert op.exists(image_name)
+    os.chdir(cwd)
 
 
 def test_stack():