tweak test

jinlow · Jan 18, 2024 · 9373893 · 9373893
1 parent 1692cf0
commit 9373893
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/py-forust/tests/test_booster.py b/py-forust/tests/test_booster.py
@@ -56,6 +56,8 @@ def test_n_records(X_y, subsample: float, max_leaves: int, grow_policy: str):
         assert root_records == total_records
         if subsample == 1.0:
             assert root_records == X.shape[0]
+        else:
+            assert root_records < X.shape[0]
 
 
 def test_booster_no_variance(X_y):