update 5 files update README.md and test_arr.py

iterpy/arr.py

@@ -130,6 +130,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).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

@@ -153,6 +153,12 @@ def test_looping():
     for i in test_iterator:
         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():

iterpy/test_iter.py

@@ -142,6 +142,12 @@ def test_looping():
     for i in test_iterator:
         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():