clean up fasta headers in unit test to compare to expected

broadinstitute · Mar 28, 2024 · 895a557 · 895a557
1 parent 2edbb6e
commit 895a557
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/unit/test_assembly.py b/test/unit/test_assembly.py
@@ -705,7 +705,7 @@ def test_output_sorted_by_product(self):
                 clusters = inf.readlines()
             self.assertEqual(len(clusters), 1)
             with open(out_skani_dist, 'r') as inf:
-                hits = list(row['Ref_name'].split()[0] for row in csv.DictReader(inf, delimiter='\t'))
+                hits = list(row['Ref_name'].split()[0].split('_')[1] for row in csv.DictReader(inf, delimiter='\t'))
             self.assertEqual(len(hits), 10)
             expected = ['L24917.1','DQ473501.1','DQ473496.1','FJ445140.1','FJ445177.1','FJ445116.1','DQ473499.1','DQ473507.1','DQ473498.1','GQ223229.1']
             self.assertEqual(hits, expected)