diff --git a/tests/test_decision_making/test_utility_functions/test_utility_functions.py b/tests/test_decision_making/test_utility_functions/test_utility_functions.py index f8e888e82..658d5b88a 100644 --- a/tests/test_decision_making/test_utility_functions/test_utility_functions.py +++ b/tests/test_decision_making/test_utility_functions/test_utility_functions.py @@ -148,11 +148,11 @@ def test_utility_functions_have_correct_shapes( posterior = generate_dummy_conjugate_posterior(dataset) posteriors = {OBJECTIVE: posterior} datasets = {OBJECTIVE: dataset} - ts_utility_builder = utility_function_builder(**utility_function_kwargs) - ts_utility_function = ts_utility_builder.build_utility_function( + utility_builder = utility_function_builder(**utility_function_kwargs) + utility_function = utility_builder.build_utility_function( posteriors=posteriors, datasets=datasets, key=key ) test_key, _ = jr.split(key) test_X = test_target_function.generate_test_points(num_test_points, test_key) - ts_utility_function_values = ts_utility_function(test_X) + ts_utility_function_values = utility_function(test_X) assert ts_utility_function_values.shape == (num_test_points, 1) diff --git a/tests/test_decision_making/test_utils.py b/tests/test_decision_making/test_utils.py index d097ee23a..f77bc6cab 100644 --- a/tests/test_decision_making/test_utils.py +++ b/tests/test_decision_making/test_utils.py @@ -21,7 +21,6 @@ from gpjax.decision_making.utils import ( OBJECTIVE, build_function_evaluator, - gaussian_cdf, ) from gpjax.typing import ( Array, @@ -45,9 +44,3 @@ def _cube(x: Float[Array, "N 1"]) -> Float[Array, "N 1"]: assert jnp.equal(datasets[OBJECTIVE].y, _square(x)).all() assert jnp.equal(datasets["CONSTRAINT"].X, x).all() assert jnp.equal(datasets["CONSTRAINT"].y, _cube(x)).all() - - -def test_gaussian_cdf(): - x = jnp.array([0.0, 1.0, 2.0]) - cdf = jnp.array([0.5, 0.84134475, 0.97724987]) - assert jnp.allclose(cdf, gaussian_cdf(x))