From 551c83739835ceb3e2044f8fbeb3b9a7bc4f9608 Mon Sep 17 00:00:00 2001
From: Simon Cruanes <simon.cruanes.2007@m4x.org>
Date: Wed, 25 Jul 2018 14:31:01 -0500
Subject: [PATCH] test: more tests for vector

---
 src/core/CCVector.ml | 73 +++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 66 insertions(+), 7 deletions(-)

diff --git a/src/core/CCVector.ml b/src/core/CCVector.ml
index 1280264c4..3eb77b59d 100644
--- a/src/core/CCVector.ml
+++ b/src/core/CCVector.ml
@@ -293,16 +293,31 @@ let equal eq v1 v2 =
     equal (=) (of_list l1) (of_list l2) = (l1=l2))
 *)
 
+(*$QR
+  Q.(pair (small_list small_int)(small_list small_int)) (fun (l1,l2) ->
+    let v1 = of_list l1 in
+    let v2 = of_list l2 in
+    equal (=) v1 v2 = (l1=l2))
+*)
+
 let compare cmp v1 v2 =
   let n = min v1.size v2.size in
   let rec check i =
     if i = n
     then compare v1.size v2.size
-    else
+    else (
       let c = cmp (get v1 i) (get v2 i) in
       if c = 0 then check (i+1) else c
+    )
   in check 0
 
+(*$QR
+  Q.(pair (small_list small_int)(small_list small_int)) (fun (l1,l2) ->
+    let v1 = of_list l1 in
+    let v2 = of_list l2 in
+    compare Pervasives.compare v1 v2 = CCList.compare Pervasives.compare l1 l2)
+*)
+
 exception Empty
 
 let pop_exn v =
@@ -349,6 +364,13 @@ let copy v = {
   OUnit.assert_bool "not_empty" (not (is_empty v));
 *)
 
+(*$QR
+  Q.(small_list small_int) (fun l ->
+    let v = of_list l in
+    let v' = copy v in
+    equal (=) v v')
+*)
+
 let shrink v n =
   if n < v.size then v.size <- n
 
@@ -532,6 +554,12 @@ let exists p v =
     else p v.vec.(i) || check (i+1)
   in check 0
 
+(*$QR
+  Q.(pair (fun1 Observable.int bool) (small_list small_int)) (fun (Q.Fun (_,f),l) ->
+    let v = of_list l in
+    exists f v = List.exists f l)
+*)
+
 let for_all p v =
   let n = v.size in
   let rec check i =
@@ -539,6 +567,12 @@ let for_all p v =
     else p v.vec.(i) && check (i+1)
   in check 0
 
+(*$QR
+  Q.(pair (fun1 Observable.int bool) (small_list small_int)) (fun (Q.Fun (_,f),l) ->
+    let v = of_list l in
+    for_all f v = List.for_all f l)
+*)
+
 let member ~eq x v =
   exists (eq x) v
 
@@ -556,6 +590,12 @@ let find p v =
   try Some (find_exn p v)
   with Not_found -> None
 
+(*$QR
+  Q.(pair (fun1 Observable.int bool) (small_list small_int)) (fun (Q.Fun (_,f),l) ->
+    let v = of_list l in
+    find f v = CCList.find_pred f l)
+*)
+
 let find_map f v =
   let n = v.size in
   let rec search i =
@@ -578,10 +618,16 @@ let filter_map f v =
   iter
     (fun x -> match f x with
        | None -> ()
-       | Some y -> push v' y
-    ) v;
+       | Some y -> push v' y)
+    v;
   v'
 
+(*$QR
+  Q.(pair (fun1 Observable.int (option bool)) (small_list small_int)) (fun (Q.Fun (_,f),l) ->
+    let v = of_list l in
+    to_list (filter_map f v) = CCList.filter_map f l)
+*)
+
 let flat_map f v =
   let v' = create () in
   iter (fun x -> iter (push v') (f x)) v;
@@ -592,8 +638,8 @@ let flat_map_seq f v =
   iter
     (fun x ->
        let seq = f x in
-       append_seq v' seq;
-    ) v;
+       append_seq v' seq)
+    v;
   v'
 
 let flat_map_list f v =
@@ -601,8 +647,8 @@ let flat_map_list f v =
   iter
     (fun x ->
        let l = f x in
-       append_list v' l;
-    ) v;
+       append_list v' l)
+    v;
   v'
 
 let (>>=) x f = flat_map f x
@@ -622,6 +668,13 @@ let rev_in_place v =
     done
   )
 
+(*$QR
+  Q.(small_list small_int) (fun l ->
+    let v = of_list l in
+    rev_in_place v;
+    to_list v = List.rev l)
+*)
+
 let rev v =
   let v' = copy v in
   rev_in_place v';
@@ -633,6 +686,12 @@ let rev v =
   rev (create ()) |> to_list = []
 *)
 
+(*$QR
+  Q.(small_list small_int) (fun l ->
+    let v = of_list l in
+    to_list (rev v) = List.rev l)
+*)
+
 let rev_iter f v =
   for i = v.size-1 downto 0 do
     f v.vec.(i)