From fb00f85dc0c9be5189b826e1fae9c59d8c158f84 Mon Sep 17 00:00:00 2001 From: Jonathan Dekermanjian Date: Mon, 30 Sep 2024 15:21:24 -0600 Subject: [PATCH] shortened TestModelCopy by removing assigned variables before comparison --- tests/model/test_core.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/tests/model/test_core.py b/tests/model/test_core.py index 6678af3b17..03dd35ca9a 100644 --- a/tests/model/test_core.py +++ b/tests/model/test_core.py @@ -1782,12 +1782,10 @@ def test_copy_model(self, copy_method) -> None: samples=1, random_seed=42 ) - simple_model_prior_predictive_val = simple_model_prior_predictive["prior"]["y"].values - copy_simple_model_prior_predictive_val = copy_simple_model_prior_predictive["prior"][ - "y" - ].values - - assert simple_model_prior_predictive_val == copy_simple_model_prior_predictive_val + assert ( + simple_model_prior_predictive["prior"]["y"].values + == copy_simple_model_prior_predictive["prior"]["y"].values + ) with copy_simple_model: z = pm.Deterministic("z", copy_simple_model["alpha"] + 1)