Correcting return type in Geometry object and making get_all_cells/get_all_materials consistent there.

This commit is contained in:
Patrick Shriwise 2019-11-07 14:58:31 -06:00
parent c5cef78ade
commit 8478a577e3

View file

@ -279,7 +279,7 @@ class Geometry(object):
if self.root_universe is not None:
return self.root_universe.get_all_cells(memo)
else:
return []
return OrderedDict()
def get_all_universes(self):
"""Return all universes in the geometry.
@ -308,7 +308,10 @@ class Geometry(object):
"""
memo = set()
return self.root_universe.get_all_materials(memo)
if self.root_universe is not None:
return self.root_universe.get_all_materials(memo)
else:
return OrderedDict()
def get_all_material_cells(self):
"""Return all cells filled by a material