crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 05:35:49 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Bug fix for comparison of ParticleFilter objects

Browse source
This commit is contained in:
Paul Romano 2020-03-23 20:50:04 -05:00
parent 91c1749b98
commit 9fffaa0c26
1 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
openmc/filter.py
View file

@ -671,6 +671,16 @@ class ParticleFilter(Filter):
The number of filter bins
"""
def __eq__(self, other):
if type(self) is not type(other):
return False
elif len(self.bins) != len(other.bins):
return False
else:
return np.all(self.bins == other.bins)
__hash__ = Filter.__hash__
@Filter.bins.setter
def bins(self, bins):
bins = np.atleast_1d(bins)
@ -1686,10 +1696,8 @@ class EnergyFunctionFilter(Filter):
return False
elif not all(self.energy == other.energy):
return False
elif not all(self.y == other.y):
return False
else:
return True
return all(self.y == other.y)
def __gt__(self, other):
if type(self) is not type(other):