reformat_slice: Add some unittests.

Adds the existing doctests as unittests.
jakirkham · Dec 1, 2016 · 6c3e58e · 6c3e58e
1 parent 67df035
commit 6c3e58e
Showing 1 changed file with 51 additions and 0 deletions.
diff --git a/tests/test_kenjutsu.py b/tests/test_kenjutsu.py
@@ -221,6 +221,57 @@ def test_reformat_slice(self):
         )
 
 
+    def test_reformat_slices(self):
+        rf_slice = kenjutsu.reformat_slices(slice(None))
+        self.assertEqual(
+            rf_slice,
+            (slice(0, None, 1),)
+        )
+
+
+        rf_slice = kenjutsu.reformat_slices((slice(None),))
+        self.assertEqual(
+            rf_slice,
+            (slice(0, None, 1),)
+        )
+
+
+        rf_slice = kenjutsu.reformat_slices((
+            slice(None),
+            slice(3, None),
+            slice(None, 5),
+            slice(None, None, 2)
+        ))
+        self.assertEqual(
+            rf_slice,
+            (
+                slice(0, None, 1),
+                slice(3, None, 1),
+                slice(0, 5, 1),
+                slice(0, None, 2)
+            )
+        )
+
+        rf_slice = kenjutsu.reformat_slices(
+            (
+                slice(None),
+                slice(3, None),
+                slice(None, 5),
+                slice(None, None, 2)
+            ),
+            (10, 13, 15, 20)
+        )
+        self.assertEqual(
+            rf_slice,
+            (
+                slice(0, 10, 1),
+                slice(3, 13, 1),
+                slice(0, 5, 1),
+                slice(0, 20, 2)
+            )
+        )
+
+
     def test_split_blocks(self):
         blocks = kenjutsu.split_blocks((2,), (1,))
         self.assertEqual(