Make StatePoint a context manager

This commit is contained in:
Paul Romano 2017-04-07 09:52:29 -05:00
parent fc0c907030
commit f4d3ee7de5

View file

@ -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