testing of reduce in iter_contiguous

Author: flexatone

src/__init__.pyi

@@ -49,6 +49,7 @@ class BlockIndex:
     def iter_contiguous(self,
             __key: tp.Union[slice, np.ndarray, tp.List[int]],
             ascending: bool = False,
+            reduce: bool = False,
             ) -> tp.Iterator[tp.Tuple[int, int]]: ...
 
 

test/test_block_index.py

@@ -610,7 +610,10 @@ def test_block_index_iter_contiguous_b(self) -> None:
             list(bi1.iter_contiguous(slice(0, 8, 3))),
             [(0, slice(0, 1, None)), (0, slice(3, 4, None)), (0, slice(6, 7, None))]
             )
-
+        self.assertEqual(
+            list(bi1.iter_contiguous(slice(0, 8, 3), reduce=True)),
+            [(0, 0), (0, 3), (0, 6)]
+            )
 
     def test_block_index_iter_contiguous_c(self) -> None:
         bi1 = BlockIndex()