From b81f8229442ef58ceece4dad4617d461de2b5a48 Mon Sep 17 00:00:00 2001 From: Alina Lacheim <a.lacheim@outlook.de> Date: Thu, 27 Feb 2025 15:39:31 +0100 Subject: [PATCH] changed valid_error test --- tests/test_engine.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_engine.py b/tests/test_engine.py index 416fe713..5e88a5c0 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'] -- GitLab