@@ -120,8 +120,15 @@ def test_get_bounds_with_upper_bounds(pytree_params):
120120 assert_array_equal (got_upper , expected_upper )
121121
122122
123- def test_get_bounds_numpy (array_params ):
124- got_lower , got_upper = get_internal_bounds (array_params )
123+ def test_get_bounds_numpy_propagate_none_true (array_params ):
124+ got_lower , got_upper = get_internal_bounds (array_params , propagate_none = True )
125+
126+ assert got_lower is None
127+ assert got_upper is None
128+
129+
130+ def test_get_bounds_numpy_propagate_none_false (array_params ):
131+ got_lower , got_upper = get_internal_bounds (array_params , propagate_none = False )
125132
126133 expected = np .array ([np .inf , np .inf ])
127134
@@ -139,7 +146,7 @@ def test_get_bounds_numpy_error(array_params):
139146 )
140147
141148
142- def test_get_fast_path_bounds_none_propagate_true ():
149+ def test_get_fast_path_bounds_propagate_none_true ():
143150 got_lower , got_upper = _get_fast_path_bounds (
144151 params = np .array ([1 , 2 , 3 ]),
145152 bounds = Bounds (lower = None , upper = None ),
@@ -149,7 +156,7 @@ def test_get_fast_path_bounds_none_propagate_true():
149156 assert got_upper is None
150157
151158
152- def test_get_fast_path_bounds_none_propagate_false ():
159+ def test_get_fast_path_bounds_propagate_none_false ():
153160 got_lower , got_upper = _get_fast_path_bounds (
154161 params = np .array ([1 , 2 , 3 ]),
155162 bounds = Bounds (lower = None , upper = None ),
0 commit comments