Adding more tests for assocN

fsprojects · May 28, 2012 · 67ff653 · 67ff653
1 parent 791af50
commit 67ff653
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 1 deletion.
diff --git a/src/FSharpx.Core/DataStructures/Vector.fs b/src/FSharpx.Core/DataStructures/Vector.fs
@@ -1,4 +1,5 @@
-module FSharpx.DataStructures.Vector
+/// vector implementation ported from https://github.com/clojure/clojure/blob/master/src/jvm/clojure/lang/PersistentVector.java
+module FSharpx.DataStructures.Vector
 
 open FSharpx
 

diff --git a/tests/FSharpx.DataStructures.Tests/PersistentVectorTest.fs b/tests/FSharpx.DataStructures.Tests/PersistentVectorTest.fs
@@ -55,3 +55,14 @@ let ``vector with 30000 elements should be convertable to a seq``() =
 let ``vector can be created from a seq``() =
     let xs = [7;88;1;4;25;30] 
     ofSeq xs |> toSeq |> Seq.toList |> should equal xs
+
+[<Test>]
+let ``vector with 30000 elements should allow assocN``() =
+    let vector = ref empty
+    for i in 1..30000 do
+        vector := cons i (!vector)
+
+    for i in 1..30000 do
+        vector := assocN (i-1) (i*2) (!vector)
+
+    !vector |> toSeq |> Seq.toList |> should equal [for i in 1..30000 -> i*2]
diff --git a/tests/FSharpx.DataStructures.Tests/TransientVectorTest.fs b/tests/FSharpx.DataStructures.Tests/TransientVectorTest.fs
@@ -50,3 +50,15 @@ let ``vector with 30000 elements should be convertable to a seq``() =
         vector := conj i (!vector)
 
     !vector |> Seq.toList |> should equal [1..30000]
+
+
+[<Test>]
+let ``vector with 30000 elements should allow assocN``() =
+    let vector = ref empty
+    for i in 1..30000 do
+        vector := conj i (!vector)
+
+    for i in 1..30000 do
+        vector := assocN (i-1) (i*2) (!vector)
+
+    !vector |> Seq.toList |> should equal [for i in 1..30000 -> i*2]