Fix filter impl in Stream combinators scalawithcats#252

AleAndForCode · Sep 16, 2024 · 69d1611 · 69d1611
1 parent 8f42d48
commit 69d1611
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/pages/codata/structural.md b/src/pages/codata/structural.md
@@ -318,7 +318,7 @@ trait Stream[A] {
 
       def tail: Stream[A] = {
         def loop(stream: Stream[A]): Stream[A] =
-          if pred(stream.head) then stream.tail
+          if pred(stream.head) then stream.tail.filter(pred)
           else loop(stream.tail)
 
         loop(self)
@@ -369,7 +369,7 @@ trait Stream[A] {
 
       def tail: Stream[A] = {
         def loop(stream: Stream[A]): Stream[A] =
-          if pred(stream.head) then stream.tail
+          if pred(stream.head) then stream.tail.filter(pred)
           else loop(stream.tail)
 
         loop(self)
@@ -456,7 +456,7 @@ trait Stream[A] {
 
       def tail: Stream[A] = {
         def loop(stream: Stream[A]): Stream[A] =
-          if pred(stream.head) then stream.tail
+          if pred(stream.head) then stream.tail.filter(pred)
           else loop(stream.tail)
 
         loop(self)
@@ -546,7 +546,7 @@ def filter(pred: A => Boolean): Stream[A] = {
 
     def tail: Stream[A] = {
       def loop(stream: Stream[A]): Stream[A] =
-        if pred(stream.head) then stream.tail
+        if pred(stream.head) then stream.tail.filter(pred)
         else loop(stream.tail)
 
       loop(self)