From e121329ef0d635308256ca7bd2b6c1a584d90864 Mon Sep 17 00:00:00 2001 From: Daniel Park Date: Mon, 18 Mar 2024 21:12:44 -0400 Subject: [PATCH] fix unit test code --- test/unit/test_assembly.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/unit/test_assembly.py b/test/unit/test_assembly.py index ea1bcee8..cccaef5c 100644 --- a/test/unit/test_assembly.py +++ b/test/unit/test_assembly.py @@ -632,9 +632,9 @@ def test_skani_contigs_to_refs(self): with open(out_clusters_filtered, 'r') as inf: clusters = inf.readlines() self.assertEqual(len(clusters), 1) - clusters = set([os.path.basename(f) for f in clusters.strip().split('\t')]) - expected_clusters = set(['ref.lasv.{}.fasta'.format(strain) for strain in ('pinneo', 'KGH_G502')]) - self.assertEqual(clusters, expected_clusters) + actual_cluster = set([os.path.basename(f) for f in clusters[0].strip().split('\t')]) + expected_cluster = set(['ref.lasv.{}.fasta'.format(strain) for strain in ('pinneo', 'KGH_G502')]) + self.assertEqual(actual_cluster, expected_cluster) class TestMutableSequence(unittest.TestCase):