From 9fa68075c6077f55572b65f1dd35de0d8e46f7d0 Mon Sep 17 00:00:00 2001 From: YoungHui Park Date: Wed, 11 Nov 2020 17:50:31 -0600 Subject: [PATCH] Convert dtye from mixed to a float for comparison assertion --- tests/regression_tests/surface_source/test.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/regression_tests/surface_source/test.py b/tests/regression_tests/surface_source/test.py index 57e1938013..01eb9a6392 100644 --- a/tests/regression_tests/surface_source/test.py +++ b/tests/regression_tests/surface_source/test.py @@ -75,10 +75,14 @@ class SurfaceSourceTestHarness(PyAPITestHarness): if self._model.settings.surf_src_write: with h5py.File("surface_source_true.h5", 'r') as f: src_true = f['source_bank'][()] + # Convert dtye from mixed to a float for comparison assertion + src_true.dtype = 'float64' with h5py.File("surface_source.h5", 'r') as f: src_test = f['source_bank'][()] - for (true_p, test_p) in zip(np.sort(src_true), np.sort(src_test)): - np.testing.assert_array_equal(true_p, test_p) + # Convert dtye from mixed to a float for comparison assertion + src_test.dtype = 'float64' + np.testing.assert_allclose(np.sort(src_true), np.sort(src_test), + atol=1e-07) def execute_test(self): """Build input XMLs, run OpenMC, check output and results."""