reformat_slices: Add tests for Ellipsis.

jakirkham · Dec 3, 2016 · 6d299f7 · 6d299f7
1 parent 718d6b2
commit 6d299f7
Showing 1 changed file with 78 additions and 0 deletions.
diff --git a/tests/test_kenjutsu.py b/tests/test_kenjutsu.py
@@ -103,6 +103,18 @@ def test_reformat_slices(self):
             (slice(0, None, 1),)
         )
 
+        rf_slice = kenjutsu.reformat_slices(Ellipsis)
+        self.assertEqual(
+            rf_slice,
+            (Ellipsis,)
+        )
+
+        rf_slice = kenjutsu.reformat_slices(Ellipsis, 10)
+        self.assertEqual(
+            rf_slice,
+            (slice(0, 10, 1),)
+        )
+
         rf_slice = kenjutsu.reformat_slices(slice(None), 10)
         self.assertEqual(
             rf_slice,
@@ -151,6 +163,72 @@ def test_reformat_slices(self):
             )
         )
 
+        rf_slice = kenjutsu.reformat_slices(
+            Ellipsis,
+            (2, 3, 4, 5)
+        )
+        self.assertEqual(
+            rf_slice,
+            (
+                slice(0, 2, 1),
+                slice(0, 3, 1),
+                slice(0, 4, 1),
+                slice(0, 5, 1)
+            )
+        )
+
+        rf_slice = kenjutsu.reformat_slices(
+            (
+                Ellipsis,
+                slice(0, 1)
+            ),
+            (2, 3, 4, 5)
+        )
+        self.assertEqual(
+            rf_slice,
+            (
+                slice(0, 2, 1),
+                slice(0, 3, 1),
+                slice(0, 4, 1),
+                slice(0, 1, 1)
+            )
+        )
+
+        rf_slice = kenjutsu.reformat_slices(
+            (
+                slice(0, 1),
+                Ellipsis
+            ),
+            (2, 3, 4, 5)
+        )
+        self.assertEqual(
+            rf_slice,
+            (
+                slice(0, 1, 1),
+                slice(0, 3, 1),
+                slice(0, 4, 1),
+                slice(0, 5, 1)
+            )
+        )
+
+        rf_slice = kenjutsu.reformat_slices(
+            (
+                slice(0, 1),
+                Ellipsis,
+                slice(0, 1)
+            ),
+            (2, 3, 4, 5)
+        )
+        self.assertEqual(
+            rf_slice,
+            (
+                slice(0, 1, 1),
+                slice(0, 3, 1),
+                slice(0, 4, 1),
+                slice(0, 1, 1)
+            )
+        )
+
 
     def test_len_slice(self):
         with self.assertRaises(kenjutsu.UnknownSliceLengthException):