Merge pull request #16 from eulerfx/master

fix AsyncSeq.scanAsync #11

Author: dsyme

src/FSharp.Control.AsyncSeq/AsyncSeq.fs

@@ -255,14 +255,16 @@ module AsyncSeq =
     | Nil -> return def
     | Cons(h, _) -> return h }
 
-  let rec scanAsync f (state:'State) (input : AsyncSeq<'T>) = asyncSeq {
-    let! v = input
-    match v with
-    | Nil -> ()
-    | Cons(h, t) ->
+  let scanAsync f (state:'TState) (input : AsyncSeq<'T>) =    
+    let rec go f state s = asyncSeq {
+      let! v = s
+      match v with
+      | Nil -> ()
+      | Cons(h, t) ->
         let! v = f state h
         yield v
-        yield! t |> scanAsync f v }
+        yield! t |> go f v }
+    asyncSeq { yield state ; yield! go f state input }
 
   let iterAsync f (input : AsyncSeq<'T>) = async {
     for itm in input do 

tests/FSharp.Control.AsyncSeq.Tests/AsyncSeqTests.fs

@@ -153,7 +153,16 @@ let ``AsyncSeq.scanAsync``() =
   let f i a = i + a
   let z = 0
   let actual = ls |> AsyncSeq.ofSeq |> AsyncSeq.scanAsync (fun i a -> f i a |> async.Return) z
-  let expected = ls |> Seq.scan f z |> Seq.skip 1 |> AsyncSeq.ofSeq
+  let expected = ls |> List.scan f z |> AsyncSeq.ofSeq
+  Assert.True(EQ expected actual)
+
+[<Test>]
+let ``AsyncSeq.scanAsync on empty should emit initial state``() =
+  let ls = List.empty
+  let f i a = i + a
+  let z = 0
+  let actual = ls |> AsyncSeq.ofSeq |> AsyncSeq.scanAsync (fun i a -> f i a |> async.Return) z
+  let expected = ls |> List.scan f z |> AsyncSeq.ofSeq
   Assert.True(EQ expected actual)