test_smart_open: close gzip obj after writing (#822)

fixes tests under python 3.12
piskvorky · May 8, 2024 · 4acd4fc · 4acd4fc
1 parent b3d8621
commit 4acd4fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/smart_open/tests/test_smart_open.py b/smart_open/tests/test_smart_open.py
@@ -1441,7 +1441,8 @@ def gzip_compress(data, filename=None):
     buf = io.BytesIO()
     buf.name = filename
     with mock.patch('time.time', _MOCK_TIME):
-        gzip.GzipFile(fileobj=buf, mode='w').write(data)
+        with gzip.GzipFile(fileobj=buf, mode='w') as gz:
+            gz.write(data)
     return buf.getvalue()