add test on vector beta distribution

CUQI-DTU · Sep 12, 2024 · 158a26c · 158a26c
1 parent a3bec98
commit 158a26c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/zexperimental/test_mcmc.py b/tests/zexperimental/test_mcmc.py
@@ -915,13 +915,13 @@ def test_HybridGibbs_updates_state_only_after_accepting_sample():
     cuqi.experimental.mcmc.NUTS(cuqi.distribution.Beta(0.5, 0.5), \
         initial_point=np.array([0.1])), \
     cuqi.experimental.mcmc.MH(target=cuqi.distribution.Beta(np.array([0.5,0.5]), np.array([0.5,0.5])), \
-        initial_point=np.array([0.1]), scale=0.1),
+        initial_point=np.array([0.1,0.1]), scale=0.1),
     cuqi.experimental.mcmc.ULA(cuqi.distribution.Beta(np.array([0.5,0.5]), np.array([0.5,0.5])), \
-        initial_point=np.array([0.1]), scale=0.1),
+        initial_point=np.array([0.1,0.1]), scale=0.1),
     cuqi.experimental.mcmc.MALA(cuqi.distribution.Beta(np.array([0.5,0.5]), np.array([0.5,0.5])), \
-        initial_point=np.array([0.1]), scale=0.1),
+        initial_point=np.array([0.1,0.1]), scale=0.1),
     cuqi.experimental.mcmc.NUTS(cuqi.distribution.Beta(np.array([0.5,0.5]), np.array([0.5,0.5])), \
-        initial_point=np.array([0.1]))
+        initial_point=np.array([0.1,0.1]))
 ]
 
 @pytest.mark.parametrize("sampler", sampler_instances_for_bounded_distribution)