Assert numer of fasta files in a smarter way

ssi-dk · Nov 22, 2023 · 3e7fa76 · 3e7fa76
1 parent 06dc56a
commit 3e7fa76
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -73,10 +73,11 @@ def test_pipeline(self):
         os.makedirs(self.test_dir)
         input_dir = pathlib.Path(self.bifrost_install_dir, 'bifrost', 'test_data', 'samples')
         assert(input_dir.exists())
-        fasta_filenames = input_dir.glob("*.fasta")
         file_number = 0
-        for filename in fasta_filenames:
-            file_number += 1
+        child: pathlib.Path
+        for child in input_dir.iterdir():
+            if child.is_file() and child.name.endswith('.fasta'):
+                file_number += 1
         assert(file_number == 1)
         test_args = ["--sample_name", "SRR2094561", "--outdir", self.test_dir]
         launcher.main(args=test_args)