diff --git a/tests/unit_tests/test_deplete_integrator.py b/tests/unit_tests/test_deplete_integrator.py index 8cade45b14..40090264ae 100644 --- a/tests/unit_tests/test_deplete_integrator.py +++ b/tests/unit_tests/test_deplete_integrator.py @@ -64,8 +64,9 @@ def test_results_save(run_in_tmpdir): r1[:] = np.random.rand(2, 2, 2) # Create global terms - eigvl1 = np.random.rand(stages) - eigvl2 = np.random.rand(stages) + # Col 0: eig, Col 1: uncertainty + eigvl1 = np.random.rand(stages, 2) + eigvl2 = np.random.rand(stages, 2) eigvl1 = comm.bcast(eigvl1, root=0) eigvl2 = comm.bcast(eigvl2, root=0) diff --git a/tests/unit_tests/test_deplete_resultslist.py b/tests/unit_tests/test_deplete_resultslist.py index 5a066be6e4..c4b7aceba5 100644 --- a/tests/unit_tests/test_deplete_resultslist.py +++ b/tests/unit_tests/test_deplete_resultslist.py @@ -44,6 +44,8 @@ def test_get_eigenvalue(res): t_ref = [0.0, 1296000.0, 2592000.0, 3888000.0] k_ref = [1.16984322, 1.19097427, 1.03012572, 1.20045627] + u_ref = [0.0375587, 0.0347639, 0.07216021, 0.02839642] np.testing.assert_allclose(t, t_ref) - np.testing.assert_allclose(k, k_ref) + np.testing.assert_allclose(k[:, 0], k_ref) + np.testing.assert_allclose(k[:, 1], u_ref)