TST: Add more tests for `nanmedian`, `nanquantile` and `nanpercentile`

1 files changed, 17 insertions, 0 deletions

diff --git a/numpy/lib/tests/test_nanfunctions.py b/numpy/lib/tests/test_nanfunctions.py
index dc14c3ad2..006b81a78 100644
--- a/numpy/lib/tests/test_nanfunctions.py
+++ b/numpy/lib/tests/test_nanfunctions.py
@@ -250,6 +250,7 @@ class TestNanFunctions_NumberTypes:
         np.nancumsum: np.cumsum,
         np.nancumprod: np.cumprod,
         np.nanmean: np.mean,
+        np.nanmedian: np.median,
         np.nanvar: np.var,
         np.nanstd: np.std,
     }
@@ -272,6 +273,22 @@ class TestNanFunctions_NumberTypes:
 
     @pytest.mark.parametrize(
         "nanfunc,func",
+        [(np.nanquantile, np.quantile), (np.nanpercentile, np.percentile)],
+        ids=["nanquantile", "nanpercentile"],
+    )
+    def test_nanfunc_q(self, dtype, nanfunc, func):
+        mat = self.mat.astype(dtype)
+        tgt = func(mat, q=1)
+        out = nanfunc(mat, q=1)
+
+        assert_almost_equal(out, tgt)
+        if dtype == "O":
+            assert type(out) is type(tgt)
+        else:
+            assert out.dtype == tgt.dtype
+
+    @pytest.mark.parametrize(
+        "nanfunc,func",
         [(np.nanvar, np.var), (np.nanstd, np.std)],
         ids=["nanvar", "nanstd"],
     )