Add one unit test about digitalize

Author: sdpython

_unittests/ut_mltree/test_tree_digitize.py

@@ -95,6 +95,17 @@ def test_tree_digitize5_false(self):
         pred = tree.predict(bins.reshape((-1, 1)))
         self.assertEqualArray(expected, pred)
 
+    def test_tree_digitize_bigger(self):
+        x = numpy.array([0, 1, 2, 3, 4, 5, 6, -1], dtype=numpy.float32)
+        bins = numpy.array([0, 1, 2, 3, 4], dtype=numpy.float32)
+        expected = numpy.digitize(x, bins, right=True)
+        tree = digitize2tree(bins, right=True)
+        pred = tree.predict(x.reshape((-1, 1)))
+        self.assertEqualArray(expected, pred)
+        expected = numpy.digitize(bins, bins, right=True)
+        pred = tree.predict(bins.reshape((-1, 1)))
+        self.assertEqualArray(expected, pred)
+
 
 if __name__ == "__main__":
     unittest.main()