Add more tests for nth_combination

tarjintor · Jul 15, 2018 · d8d96ca · d8d96ca
1 parent 7041f0a
commit d8d96ca
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/more_itertools/tests/test_recipes.py b/more_itertools/tests/test_recipes.py
@@ -590,6 +590,15 @@ def test_long(self):
         expected = (2, 12, 35, 126)
         self.assertEqual(actual, expected)
 
+    def test_invalid_r(self):
+        for r in (-1, 3):
+            with self.assertRaises(ValueError):
+                mi.nth_combination([], r, 0)
+
+    def test_invalid_index(self):
+        with self.assertRaises(IndexError):
+            mi.nth_combination('abcdefg', 3, -36)
+
 
 class PrependTests(TestCase):
     def test_basic(self):