Fix #182 (#188)

brentp · Jan 26, 2021 · 72ed483 · 72ed483
1 parent f3738aa
commit 72ed483
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cyvcf2/tests/test_reader.py b/cyvcf2/tests/test_reader.py
@@ -351,15 +351,20 @@ def test_add_flag():
     w.write_record(rec)
     w.close()
 
-    v = next(VCF(f))
+    fh = VCF(f)
+    v = next(fh)
+    fh.close()
     assert v.INFO["myflag"] is True, dict(v.INFO)
 
     f = tempfile.mktemp(suffix=".vcf")
     atexit.register(os.unlink, f)
     w = Writer(f, vcf)
     rec.INFO["myflag"] = False
     w.write_record(rec)
-    v = next(VCF(f))
+    w.close()
+    fh = VCF(f)
+    v = next(fh)
+    fh.close()
     assert_raises(KeyError, v.INFO.__getitem__, "myflag")