From 332380479bd7ea718c04124fe35693773eca1112 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Sat, 23 Apr 2022 02:07:01 +0800
Subject: [PATCH] Avoid warning in np primitive type tests. (#7833)

---
 tests/python/test_basic_models.py |  2 +-
 tests/python/test_predict.py      | 10 +++++++---
 tests/python/test_with_pandas.py  |  2 +-
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/tests/python/test_basic_models.py b/tests/python/test_basic_models.py
index 2cfa26402..510aec506 100644
--- a/tests/python/test_basic_models.py
+++ b/tests/python/test_basic_models.py
@@ -569,7 +569,7 @@ class TestModels:
         y = rng.randn(rows)
         feature_names = ["test_feature_" + str(i) for i in range(cols)]
         X_pd = pd.DataFrame(X, columns=feature_names)
-        X_pd.iloc[:, 3] = X_pd.iloc[:, 3].astype(np.int)
+        X_pd.iloc[:, 3] = X_pd.iloc[:, 3].astype(np.int32)
 
         Xy = xgb.DMatrix(X_pd, y)
         assert Xy.feature_types[3] == "int"
diff --git a/tests/python/test_predict.py b/tests/python/test_predict.py
index b34d508cd..f4ea944e8 100644
--- a/tests/python/test_predict.py
+++ b/tests/python/test_predict.py
@@ -245,19 +245,23 @@ class TestInplacePredict:
         predt_orig = self.booster.inplace_predict(orig)
         # all primitive types in numpy
         for dtype in [
-            np.signedinteger,
+            np.int32,
+            np.int64,
             np.byte,
             np.short,
             np.intc,
             np.int_,
             np.longlong,
-            np.unsignedinteger,
+            np.uint32,
+            np.uint64,
             np.ubyte,
             np.ushort,
             np.uintc,
             np.uint,
             np.ulonglong,
-            np.floating,
+            np.float16,
+            np.float32,
+            np.float64,
             np.half,
             np.single,
             np.double,
diff --git a/tests/python/test_with_pandas.py b/tests/python/test_with_pandas.py
index c55d698bb..1401dd699 100644
--- a/tests/python/test_with_pandas.py
+++ b/tests/python/test_with_pandas.py
@@ -328,5 +328,5 @@ class TestPandas:
             return to_bytes(Xy)
 
         b0 = test_bool(pd.BooleanDtype())
-        b1 = test_bool(np.bool)
+        b1 = test_bool(bool)
         assert b0 != b1         # None is converted to False with np.bool