fixing random seeds

tests/conftest.py

@@ -41,6 +41,7 @@ def small_X_y_count_data():
     """Small set of X, y data, with y being counts (positive int)"""
 
     def generate_count_data(n_samples=100):
+        np.random.seed(11)  # 'tuned' to be simple to test against
         X = np.random.uniform(-2, 0, n_samples)
         n = 66 * np.abs(np.cos(X))
         p = 0.5 * np.abs(np.cos(X / 3))

tests/test_model.py

@@ -29,7 +29,7 @@ def test_XGBDistribution_early_stopping_fit(small_train_test_data):
 def test_XGBDistribution_early_stopping_fit_single_param_distribution(
     small_train_test_count_data,
 ):
-    """Integration test for single param dist"""
+    """Integration test for single param dist (which operate on squeezed arrays)"""
 
     X_train, X_test, y_train, y_test = small_train_test_count_data
 
@@ -42,7 +42,7 @@ def test_XGBDistribution_early_stopping_fit_single_param_distribution(
     model.fit(X_train, y_train, eval_set=[(X_test, y_test)])
     evals_result = model.evals_result()
 
-    # assert model.best_iteration == 6
+    assert model.best_iteration == 10
     assert isinstance(evals_result, dict)