add test for fixed random seed

sherbold · Jul 25, 2023 · c674603 · c674603
1 parent b91fad3
commit c674603
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_autorank.py b/tests/test_autorank.py
@@ -797,3 +797,18 @@ def test_cd_order(self):
         plot_stats(res)
         plot_stats(res_asc)
         plt.draw()
+
+    def test_fix_random_state(self):
+        std = 0.15
+        means = [0.3, 0.3]
+        data = pd.DataFrame()
+        for i, mean in enumerate(means):
+            data['pop_%i' % i] = np.random.normal(mean, std, self.sample_size).clip(0, 1)
+
+        res_42_1 = autorank(data, alpha=0.05, nsamples=100, verbose=self.verbose, approach='bayesian', random_state=42)
+        res_42_2 = autorank(data, alpha=0.05, nsamples=100, verbose=self.verbose, approach='bayesian', random_state=42)
+        res_43_1 = autorank(data, alpha=0.05, nsamples=100, verbose=self.verbose, approach='bayesian', random_state=43)
+
+        self.assertTrue(res_42_1.posterior_matrix.equals(res_42_2.posterior_matrix))
+        self.assertFalse(res_42_1.posterior_matrix.equals(res_43_1.posterior_matrix))
+