fix tests :)

sourmash-bio · Jun 16, 2021 · 9b50748 · 9b50748
1 parent aaa4548
commit 9b50748
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/sourmash/sig/picklist.py b/src/sourmash/sig/picklist.py
@@ -9,7 +9,7 @@
 preprocess['md5'] = lambda x: x
 
 # identifier matches/prefix foo - space delimited identifiers
-preprocess['ident.'] = lambda x: x.split(' ')[0].split('.')[0]
+preprocess['identprefix'] = lambda x: x.split(' ')[0].split('.')[0]
 preprocess['ident'] = lambda x: x.split(' ')[0]
 
 # match 8 characters
@@ -79,8 +79,10 @@ def _get_sig_attribute(self, ss):
             q = ss.name
         elif coltype == 'ident':
             q = ss.name
-        elif coltype == 'ident.':
+        elif coltype == 'identprefix':
             q = ss.name
+        else:
+            assert 0
 
         return q
 

diff --git a/tests/test_cmd_signature.py b/tests/test_cmd_signature.py
@@ -1222,7 +1222,7 @@ def test_sig_extract_8_picklist_ident_dot(runtmp):
         w.writeheader()
         w.writerow(row)
 
-    picklist_arg = f"{picklist_csv}:nodotIdent:ident."
+    picklist_arg = f"{picklist_csv}:nodotIdent:identprefix"
     runtmp.sourmash('sig', 'extract', sig47, sig63, '--picklist', picklist_arg)
 
     # stdout should be new signature