fix sort bug in 'sourmash search'

sourmash-bio · Jun 15, 2016 · 0cb1872 · 0cb1872
1 parent e9aec89
commit 0cb1872
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/sourmash b/sourmash
@@ -83,7 +83,7 @@ Commands can be:
 
         # any matches? sort, show.
         if distances:
-            distances.sort(reverse=True)
+            distances.sort(reverse=True, key = lambda x: x[0])
             print('%d matches:' % len(distances))
             for distance, match, filename in distances[:3]:
                 print('\t', match.name(), '\t', "%.3f" % distance,

diff --git a/sourmash_lib/test_sourmash.py b/sourmash_lib/test_sourmash.py
@@ -94,6 +94,27 @@ def test_sourmash_search():
         assert '0.958' in out
 
 
+def test_sourmash_search_2():
+    with utils.TempDirectory() as location:
+        testdata1 = utils.get_test_data('short.fa')
+        testdata2 = utils.get_test_data('short2.fa')
+        testdata3 = utils.get_test_data('short3.fa')
+        status, out, err = utils.runscript('sourmash',
+                                           ['compute', testdata1, testdata2,
+                                            testdata3],
+                                           in_directory=location)
+
+
+
+        status, out, err = utils.runscript('sourmash',
+                                           ['search', 'short.fa.sig',
+                                            'short2.fa.sig', 'short3.fa.sig'],
+                                           in_directory=location)
+        print(status, out, err)
+        assert '2 matches' in out
+        assert '0.958' in out
+
+
 def test_mash_csv_to_sig():
     with utils.TempDirectory() as location:
         testdata1 = utils.get_test_data('short.fa.msh.dump')