Update tests.

trivialfis · Dec 23, 2021 · f4b8f31 · f4b8f31
1 parent 3b3b48c
commit f4b8f31
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/tests/python/test_updaters.py b/tests/python/test_updaters.py
@@ -45,14 +45,20 @@ def test_exact(self, param, num_rounds, dataset):
         result = train_result(param, dataset.get_dmat(), num_rounds)
         assert tm.non_increasing(result['train'][dataset.metric])
 
-    @given(exact_parameter_strategy, strategies.integers(1, 20),
-           tm.dataset_strategy)
+    @given(
+        exact_parameter_strategy,
+        hist_parameter_strategy,
+        strategies.integers(1, 20),
+        tm.dataset_strategy,
+    )
     @settings(deadline=None)
-    def test_approx(self, param, num_rounds, dataset):
-        param['tree_method'] = 'approx'
+    def test_approx(self, param, hist_param, num_rounds, dataset):
+        param["tree_method"] = "approx"
         param = dataset.set_params(param)
+        param.update(hist_param)
         result = train_result(param, dataset.get_dmat(), num_rounds)
-        assert tm.non_increasing(result['train'][dataset.metric], 1e-3)
+        note(result)
+        assert tm.non_increasing(result["train"][dataset.metric])
 
     @pytest.mark.skipif(**tm.no_sklearn())
     def test_pruner(self):