From b5467521e880ed9092af697d3d13b0b6d999d00e Mon Sep 17 00:00:00 2001 From: Tom Augspurger Date: Tue, 16 May 2017 10:27:52 -0500 Subject: [PATCH] TST: Add test for clip-na Additional test cases for https://github.com/pandas-dev/pandas/pull/16364 when upper and / or lower is nan. --- pandas/tests/frame/test_analytics.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/pandas/tests/frame/test_analytics.py b/pandas/tests/frame/test_analytics.py index be89b27912d1c..1b6471fcef565 100644 --- a/pandas/tests/frame/test_analytics.py +++ b/pandas/tests/frame/test_analytics.py @@ -1863,6 +1863,23 @@ def test_clip_against_frame(self): tm.assert_frame_equal(clipped_df[ub_mask], ub[ub_mask]) tm.assert_frame_equal(clipped_df[mask], df[mask]) + def test_clip_na(self): + msg = "Cannot use an NA" + with tm.assert_raises_regex(ValueError, msg): + self.frame.clip(lower=np.nan) + + with tm.assert_raises_regex(ValueError, msg): + self.frame.clip(lower=[np.nan]) + + with tm.assert_raises_regex(ValueError, msg): + self.frame.clip(upper=np.nan) + + with tm.assert_raises_regex(ValueError, msg): + self.frame.clip(upper=[np.nan]) + + with tm.assert_raises_regex(ValueError, msg): + self.frame.clip(lower=np.nan, upper=np.nan) + # Matrix-like def test_dot(self):