feat: .chain(other) (#213)

iterpy/arr.py

@@ -8,6 +8,7 @@
 
     from iterpy.iter import Iter
 
+
 T = TypeVar("T")
 S = TypeVar("S")
 
@@ -130,6 +131,9 @@ def clone(self) -> Arr[T]:
     def zip(self, other: Arr[S]) -> Arr[tuple[T, S]]:
         return Arr(zip(self, other))
 
+    def chain(self, other: Arr[T]) -> Arr[T]:
+        return self.lazy().chain(other.lazy()).collect()
+
     ############################################################
     # Auto-generated overloads for flatten                     #
     # Code for generating the following is in _generate_pyi.py #

iterpy/iter.py

@@ -146,6 +146,9 @@ def clone(self) -> Iter[T]:
     def zip(self, other: Iter[S]) -> Iter[tuple[T, S]]:
         return Iter(zip(self, other))
 
+    def chain(self, other: Iter[T]) -> Iter[T]:
+        return Iter((*self, *other))
+
     ############################################################
     # Auto-generated overloads for flatten                     #
     # Code for generating the following is in _generate_pyi.py #

iterpy/test_arr.py

@@ -154,6 +154,13 @@ def test_looping():
         assert i in [1, 2, 3]
 
 
+def test_chain():
+    iterator = Arr([1, 2])
+    iterator2 = Arr([3, 4])
+    result: list[int] = iterator.chain(iterator2).to_list()
+    assert result == [1, 2, 3, 4]
+
+
 @pytest.mark.benchmark()
 def test_benchmark_large_flattening():
     test_input = Arr(range(100_000)).map(lambda x: Arr([x]))

iterpy/test_iter.py

@@ -143,6 +143,13 @@ def test_looping():
         assert i in [1, 2, 3]
 
 
+def test_chain():
+    iterator = Iter([1, 2])
+    iterator2 = Iter([3, 4])
+    result: list[int] = iterator.chain(iterator2).to_list()
+    assert result == [1, 2, 3, 4]
+
+
 @pytest.mark.benchmark()
 def test_benchmark_large_flattening():
     test_input = Iter(range(100_000)).map(lambda x: Iter([x]))