added test cases

Author: raguiar2

pandas/tests/frame/test_rank.py

@@ -191,6 +191,20 @@ def test_rank_na_option(self):
         tm.assert_numpy_array_equal(ranks0.values, exp0)
         tm.assert_numpy_array_equal(ranks1.values, exp1)
 
+        # bad values throw error
+        msg = "na_option must be one of 'keep', 'top', or 'bottom'"
+
+        with pytest.raises(ValueError) as excinfo:
+            self.frame.rank(na_option='bad', ascending=False)
+
+        assert msg == str(excinfo.value)
+
+        # invalid type
+        with pytest.raises(ValueError) as excinfo:
+            self.frame.rank(na_option=True, ascending=False)
+
+        assert msg == str(excinfo.value)
+
     def test_rank_axis(self):
         # check if using axes' names gives the same result
         df = DataFrame([[2, 1], [4, 3]])

pandas/tests/groupby/test_rank.py

@@ -248,6 +248,17 @@ def test_rank_avg_even_vals():
 def test_rank_object_raises(ties_method, ascending, na_option,
                             pct, vals):
     df = DataFrame({'key': ['foo'] * 5, 'val': vals})
+
+    with tm.assert_raises_regex(TypeError, "not callable"):
+        df.groupby('key').rank(method=ties_method,
+                               ascending=ascending,
+                               na_option='bad', pct=pct)
+
+    with tm.assert_raises_regex(TypeError, "not callable"):
+        df.groupby('key').rank(method=ties_method,
+                               ascending=ascending,
+                               na_option=True, pct=pct)
+
     with tm.assert_raises_regex(TypeError, "not callable"):
         df.groupby('key').rank(method=ties_method,
                                ascending=ascending,

pandas/tests/series/test_rank.py

@@ -183,6 +183,19 @@ def test_rank_categorical(self):
             exp_keep
         )
 
+        # Test invalid values for na_option
+        msg = "na_option must be one of 'keep', 'top', or 'bottom'"
+        with pytest.raises(ValueError) as excinfo:
+            na_ser.rank(na_option='bad', ascending=False)
+
+        assert msg == str(excinfo.value)
+
+        # invalid type
+        with pytest.raises(ValueError):
+            na_ser.rank(na_option=True, ascending=False)
+
+        assert msg == str(excinfo.value)
+
         # Test with pct=True
         na_ser = Series(['first', 'second', 'third', 'fourth', np.NaN]).astype(
             CategoricalDtype(['first', 'second', 'third', 'fourth'], True))