Fixed expected call tests

Author: lucianopaz

homepy/tests/models/test_base.py

@@ -334,12 +334,13 @@ def test_model_batched_sample_posterior_predictive(
         for var_name in model.get_var_names_to_resample(idata, var_names=var_names)
     ]
     vars_in_trace_ = [model.model[var_name] for var_name in idata.posterior.data_vars]
+    constant_coords = set(model.model.coords)
     mock_compile_forward_sampling_function.assert_called_once_with(
         outputs=outputs_,
         vars_in_trace=vars_in_trace_,
         basic_rvs=model.model.basic_RVs,
         constant_data={},
-        constant_coords=set(),
+        constant_coords=constant_coords,
         allow_input_downcast=True,
         accept_inplace=True,
     )
@@ -392,7 +393,10 @@ def test_new_model_batched_sample_posterior_predictive(
         vars_in_trace=vars_in_trace_,
         basic_rvs=new_model.model.basic_RVs,
         constant_data={},
-        constant_coords=set(),
+        constant_coords={
+            dim for dim in new_model.model.coords
+            if dim not in ['C_99_ids', 'observation', 'C_cage']
+        },
         allow_input_downcast=True,
         accept_inplace=True,
     )