def test_compute_decreased_value_for_ee(): up = np.array([[[False, True], [True, False]], [[True, False], [False, True]], [[False, True], [False, False]], [[True, False], [False, False]], [[False, False], [True, False]], [[False, False], [True, False]]], dtype=bool) lo = np.array([[[False, False], [False, False]], [[False, False], [False, False]], [[False, False], [True, False]], [[False, False], [False, True]], [[False, True], [False, False]], [[False, True], [False, False]]], dtype=bool) model_outputs = np.array([0.97, 0.71, 2.39, 0.97, 2.3, 2.39, 1.87, 2.40, 0.87, 2.15, 1.71, 1.54, 2.15, 2.17, 1.54, 2.2, 1.87, 1.0], dtype=np.float) op_vec = model_outputs.reshape(6, 3) actual = get_decreased_values(op_vec, up, lo) desired = np.array([[0.71, 0.97, 0.87, 2.15, 2.17, 1.87], [0.97, 2.30, 1.87, 1.54, 2.17, 1.87]], dtype=np.float) assert_allclose(actual, desired, atol=1e-1)