Textbook file closing for test files

ml31415 · May 27, 2024 · 138df2d · 138df2d
1 parent 9d4dfad
commit 138df2d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/tests/integration/benchmark.py b/tests/integration/benchmark.py
@@ -11,7 +11,8 @@
 
 def test_performance(benchmark):
     path = RESOURCES_DIR / "data/27312315.bz2"
-    data = bz2.open(path).read()
+    with bz2.open(path) as f:
+        data = f.read()
     result = benchmark.pedantic(stream_decode_lines, args=(data,))
     assert len(result) == 50854
     for msg in result:

diff --git a/tests/integration/test_integration.py b/tests/integration/test_integration.py
@@ -85,10 +85,11 @@ def test_responses(path):
 def test_archive(path):
     mc = MarketSubscriptionCache()
     i = 0
-    for i, line in enumerate(bz2.open(path), start=1):
-        msg = stream_decode(line)
-        assert isinstance(msg, MCM)
-        mc.update(msg)
+    with bz2.open(path) as f:
+        for i, line in enumerate(f, start=1):
+            msg = stream_decode(line)
+            assert isinstance(msg, MCM)
+            mc.update(msg)
 
     required_count = LINE_COUNT.get(path.name)
     if required_count: