Initial commit Fixes 3 tests (#1550)

sourmash-bio · May 24, 2021 · 516dc53 · 516dc53
1 parent 83954d1
commit 516dc53
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/test_lca.py b/tests/test_lca.py
@@ -971,7 +971,8 @@ def test_single_classify_to_output():
         print(out)
         print(err)
 
-        outdata = open(os.path.join(location, 'outfile.txt'), 'rt').read()
+        with open(os.path.join(location, 'outfile.txt'), 'rt') as fp:
+            outdata = fp.read()
         assert 'TARA_ASE_MAG_00031,found,Bacteria,Proteobacteria,Gammaproteobacteria,Alteromonadales,Alteromonadaceae,Alteromonas,Alteromonas_macleodii' in outdata
         assert 'classified 1 signatures total' in err
         assert 'loaded 1 LCA databases' in err
@@ -996,8 +997,8 @@ def test_single_classify_to_output_no_name():
         print(cmd)
         print(out)
         print(err)
-
-        outdata = open(os.path.join(location, 'outfile.txt'), 'rt').read()
+        with open(os.path.join(location, 'outfile.txt'), 'rt') as fp:
+            outdata = fp.read()
         print((outdata,))
         assert 'xyz,found,Bacteria,Proteobacteria,Gammaproteobacteria,Alteromonadales,Alteromonadaceae,Alteromonas,Alteromonas_macleodii' in outdata
         assert 'classified 1 signatures total' in err
@@ -1521,8 +1522,8 @@ def test_single_summarize_to_output():
         print(out)
         print(err)
 
-        outdata = open(os.path.join(location, 'output.txt'), 'rt').read()
-
+        with open(os.path.join(location, 'output.txt'), 'rt') as fp:
+            outdata = fp.read()
         assert 'loaded 1 signatures from 1 files total.' in err
         assert '200,Bacteria,Proteobacteria,Gammaproteobacteria' in outdata