From a29134ef430872f6d87ff282bcf67de28f8203b8 Mon Sep 17 00:00:00 2001 From: Patrick Shriwise Date: Wed, 16 Mar 2022 12:18:47 -0500 Subject: [PATCH] Test against flat array now that the attribute is shaped --- tests/unit_tests/weightwindows/test_wwinp_reader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit_tests/weightwindows/test_wwinp_reader.py b/tests/unit_tests/weightwindows/test_wwinp_reader.py index c225f12f2c..9aeaa9f59e 100644 --- a/tests/unit_tests/weightwindows/test_wwinp_reader.py +++ b/tests/unit_tests/weightwindows/test_wwinp_reader.py @@ -119,7 +119,7 @@ def test_wwinp_reader(wwinp_data, request): # a reversed array of the flat index into the numpy array n_wws = np.prod((*mesh.dimension, e_bounds.size - 1)) exp_ww_lb = np.linspace(1, n_wws, n_wws)[::-1] - np.testing.assert_array_equal(exp_ww_lb, ww.lower_ww_bounds) + np.testing.assert_array_equal(exp_ww_lb, ww.lower_ww_bounds.flatten()) # check expected failures