Fix flaky TestMixture.test_component_choice_random (#6222)

* TST: Fix seed for draws

* TST: Use binomial test

* TST: Revert to previous assertion

* TST: Increase draws to 200

* Implement suggestion to lower draws to 20

Co-authored-by: Ricardo Vieira <[email protected]>

* Correct comment to match the number of draws

Co-authored-by: Ricardo Vieira <[email protected]>

Author: bherwerth

pymc/tests/distributions/test_mixture.py

@@ -333,8 +333,8 @@ def test_component_choice_random(self):
             weights = [0.5, 0.5]
             components = [Normal.dist(-10, 0.01), Normal.dist(10, 0.01)]
             mix = Mixture.dist(weights, components)
-        draws = draw(mix, draws=10)
-        # Probability of coming from same component 10 times is 0.5**10
+        draws = draw(mix, draws=20, random_seed=self.get_random_state())
+        # Probability of coming from same component 20 times is 0.5**20
         assert np.unique(draws > 0).size == 2
 
     @pytest.mark.parametrize(