len_slice: Add some more unit tests.

Adds some more tests to get full coverage of the `len_slice` function.
jakirkham · Dec 2, 2016 · 0add7ef · 0add7ef
1 parent 824f8a5
commit 0add7ef
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/tests/test_kenjutsu.py b/tests/test_kenjutsu.py
@@ -293,6 +293,18 @@ def test_len_slice(self):
         l = kenjutsu.len_slice(slice(None), 10)
         self.assertEqual(l, len(range(10)[:]))
 
+        l = kenjutsu.len_slice(slice(None, None, 20), 10)
+        self.assertEqual(l, 1)
+
+        l = kenjutsu.len_slice(slice(None, None, 20), 10)
+        self.assertEqual(l, len(range(10)[::20]))
+
+        l = kenjutsu.len_slice(slice(-15, None, -20), 10)
+        self.assertEqual(l, 0)
+
+        l = kenjutsu.len_slice(slice(-15, None, -20), 10)
+        self.assertEqual(l, len(range(10)[-15::-20]))
+
         l = kenjutsu.len_slice(slice(2, None), 10)
         self.assertEqual(l, 8)
 
@@ -323,6 +335,18 @@ def test_len_slice(self):
         l = kenjutsu.len_slice(slice(2, 6, 3))
         self.assertEqual(l, len(range(10)[2:6:3]))
 
+        l = kenjutsu.len_slice(slice(None, 15, -2), 10)
+        self.assertEqual(l, 0)
+
+        l = kenjutsu.len_slice(slice(None, 15, -2), 10)
+        self.assertEqual(l, len(range(10)[:15:-2]))
+
+        l = kenjutsu.len_slice(slice(None, -3, 2), 10)
+        self.assertEqual(l, 4)
+
+        l = kenjutsu.len_slice(slice(None, -3, 2), 10)
+        self.assertEqual(l, len(range(10)[:-3:2]))
+
 
     def test_len_slices(self):
         with self.assertRaises(kenjutsu.UnknownSliceLengthException):