Use a.all() for equality in capi EnergyFunctionFilter tests

This commit is contained in:
Andrew Johnson 2019-07-26 10:14:20 -05:00
parent 9b7ab0bacb
commit 9d952c3c58
No known key found for this signature in database
GPG key ID: 253418E91B7F6FEB

View file

@ -188,9 +188,9 @@ def test_energy_function_filter(capi_init):
"""Test special __new__ and __init__ for EnergyFunctionFilter"""
efunc = openmc.capi.EnergyFunctionFilter([0.0, 1.0], [0.0, 2.0])
assert len(efunc.energy) == 2
assert efunc.energy == [0.0, 1.0]
assert (efunc.energy == [0.0, 1.0]).all()
assert len(efunc.y) == 2
assert efunc.y == [0.0, 2.0]
assert (efunc.y == [0.0, 2.0]).all()
def test_tally(capi_init):