Use yield in fixture

lszeremeta · Oct 20, 2020 · 6c5f8b4 · 6c5f8b4
1 parent 7992483
commit 6c5f8b4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_outputs.py b/tests/test_outputs.py
@@ -16,8 +16,9 @@
 
 @pytest.fixture
 def csv_reader():
-    """Open test file and return csv.DictReader."""
-    return csv.DictReader(open(os.path.join(HERE, "drugbank_vocabulary_CC0_first_20_sample.csv"), 'r'))
+    """Open test file and prepare csv.DictReader."""
+    with open(os.path.join(HERE, "drugbank_vocabulary_CC0_first_20_sample.csv"), 'r') as csvfile:
+        yield csv.DictReader(csvfile)
 
 
 #