Fix chunks(countOf: 1) (#293)

apple · Sep 19, 2023 · 281e27c · 281e27c
1 parent 4a1fb99
commit 281e27c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/Sources/AsyncAlgorithms/AsyncChunksOfCountSequence.swift b/Sources/AsyncAlgorithms/AsyncChunksOfCountSequence.swift
@@ -49,6 +49,10 @@ public struct AsyncChunksOfCountSequence<Base: AsyncSequence, Collected: RangeRe
         return nil
       }
 
+      if count == 1 {
+        return Collected(CollectionOfOne(first))
+      }
+
       var result: Collected = .init()
       result.append(first)
 

diff --git a/Tests/AsyncAlgorithmsTests/TestChunk.swift b/Tests/AsyncAlgorithmsTests/TestChunk.swift
@@ -24,6 +24,14 @@ func concatCharacters(_ array: [String]) -> String {
 }
 
 final class TestChunk: XCTestCase {
+  func test_count_one() {
+    validate {
+      "ABCDE|"
+      $0.inputs[0].chunks(ofCount: 1).map(concatCharacters)
+      "ABCDE|"
+    }
+  }
+
   func test_signal_equalChunks() {
     validate {
       "ABC-    DEF-    GHI-     |"