Fix leaked temporary file in unit tests

ligo/tests/test_em_bright.py

@@ -9,17 +9,17 @@ from .. import em_bright
 
 
 def test_source_classification_pe():
-    f = NamedTemporaryFile()
-    filename = f.name
-    with h5py.File(f) as tmp_h5:
-        data = np.array([(1.2, 1.0, 0.0, 0.0, 100.0), (2.0, 0.5, 0.99, 0.99, 150.0)],
-                        dtype=[('mc', '<f8'), ('q', '<f8'),
-                               ('a1', '<f8'), ('a2', '<f8'), ('dist', '<f8')])
-        tmp_h5.create_dataset(
-            'lalinference/lalinference_mcmc/posterior_samples',
-            data=data
-        )
-    r = em_bright.source_classification_pe(filename)
+    with NamedTemporaryFile() as f:
+        filename = f.name
+        with h5py.File(f) as tmp_h5:
+            data = np.array([(1.2, 1.0, 0.0, 0.0, 100.0), (2.0, 0.5, 0.99, 0.99, 150.0)],
+                            dtype=[('mc', '<f8'), ('q', '<f8'),
+                                   ('a1', '<f8'), ('a2', '<f8'), ('dist', '<f8')])
+            tmp_h5.create_dataset(
+                'lalinference/lalinference_mcmc/posterior_samples',
+                data=data
+            )
+        r = em_bright.source_classification_pe(filename)
     assert r == (1.0, 1.0)