diff --git a/tests/test_engine.py b/tests/test_engine.py index 416fe713750291f5c8332d1dc6a01b19e8497b8b..5e88a5c055afab374552f6e0f3ff258427055581 100644 --- a/tests/test_engine.py +++ b/tests/test_engine.py @@ -123,14 +123,15 @@ def test_valid_error() -> None: inp.marginals[0].dist_type = 'normal' inp.marginals[0].parameters = [0, 1] mm = PCE(inp) - mm.fit([[0.0, 0.0], [1.0, 1.0]], {'Z': [[0.5], [0.5]]}) # Shape (2, 2) + mm.fit([[0.0], [1.0]], {'Z': [[0.5], [0.5]]}) # Shape (2, 2) expdes = ExpDesigns(inp) mod = PL() + # Update valid_samples to match input size during fitting - expdes.valid_samples = np.array([[0.5, 0.5]]) # Shape (1, 2) - expdes.valid_model_runs = {'Z': np.array([[0.5], [0.5]])} + expdes.valid_samples = np.array([[0.5]]) # Shape (1, 2) + expdes.valid_model_runs = {'Z': np.array([[0.5]])} mod.mc_reference['mean'] = [0.5] mod.mc_reference['std'] = [0.0] mod.output.names = ['Z']