From f4d3ee7de576662e555da75f5f697ccfc8237d11 Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Fri, 7 Apr 2017 09:52:29 -0500
Subject: [PATCH] Make StatePoint a context manager

---
 openmc/statepoint.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/openmc/statepoint.py b/openmc/statepoint.py
index 3eec9a20e..28f35ba7d 100644
--- a/openmc/statepoint.py
+++ b/openmc/statepoint.py
@@ -138,6 +138,12 @@ class StatePoint(object):
                     vol = openmc.VolumeCalculation.from_hdf5(path_i)
                     self.add_volume_information(vol)
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *exc):
+        self._f.close()
+
     @property
     def cmfd_on(self):
         return self._f.attrs['cmfd_on'] > 0