Find bacteria line

sunbeam-labs · Feb 16, 2024 · e98b4a7 · e98b4a7
1 parent 6d5ed60
commit e98b4a7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/.tests/e2e/test_full_run.py b/.tests/e2e/test_full_run.py
@@ -93,7 +93,13 @@ def test_full_run(run_sunbeam):
         header_line = f.readline()
         assert "TEST-taxa" in header_line
         assert "EMPTY-taxa" in header_line
+        test_index = header_line.split("\t").index("TEST-taxa")
+        empty_index = header_line.split("\t").index("EMPTY-taxa")
 
         lines = f.readlines()
         print(lines)
-        assert "2\t394\t0\n" in lines
+        for line in lines:
+            if line[0] == "2":
+                fields = line.split("\t")
+                assert int(fields[empty_index]) == 0
+                assert int(fields[test_index]) > 0