change indices to be 1-based

sourmash-bio · Oct 15, 2023 · b90f761 · b90f761
1 parent 8e7c7d4
commit b90f761
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/src/sourmash/commands.py b/src/sourmash/commands.py
@@ -281,7 +281,7 @@ def plot(args):
             sys.exit(-1)
     elif args.indices:
         # construct integer labels
-        labeltext = [str(i) for i in range(D.shape[0])]
+        labeltext = [str(i + 1) for i in range(D.shape[0])]
     else:
         assert not display_labels
         labeltext = [""] * D.shape[0]

diff --git a/tests/test_sourmash.py b/tests/test_sourmash.py
@@ -979,10 +979,10 @@ def test_plot_6_indices(runtmp):
     print(runtmp.last_result.out)
 
     expected = """\
-0\t0
-1\t1
-2\t2
-3\t3"""
+0\t1
+1\t2
+2\t3
+3\t4"""
     assert expected in runtmp.last_result.out
 
 
@@ -1000,10 +1000,10 @@ def test_plot_6_no_labels(runtmp):
     print(runtmp.last_result.out)
 
     expected = """\
-0\t0
-1\t1
-2\t2
-3\t3"""
+0\t1
+1\t2
+2\t3
+3\t4"""
     assert expected in runtmp.last_result.out
 
 
@@ -1021,10 +1021,10 @@ def test_plot_6_no_indices(runtmp):
     print(runtmp.last_result.out)
 
     expected = """\
-0\t0
-1\t1
-2\t2
-3\t3"""
+0\t1
+1\t2
+2\t3
+3\t4"""
     assert expected in runtmp.last_result.out