From cb113e51744b644a9c39d8662994fc161fd4b3e6 Mon Sep 17 00:00:00 2001
From: Simon Cruanes <simon.cruanes.2007@m4x.org>
Date: Fri, 8 Mar 2013 16:59:02 +0100
Subject: [PATCH] a few more tests

---
 tests/test_sequence.ml | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/tests/test_sequence.ml b/tests/test_sequence.ml
index 457a642..e0b3c5c 100644
--- a/tests/test_sequence.ml
+++ b/tests/test_sequence.ml
@@ -139,6 +139,29 @@ let test_drop () =
 let test_rev () =
   1 -- 5 |> S.rev |> S.to_list |> OUnit.assert_equal [5;4;3;2;1]
 
+let test_unfoldr () =
+  let f x = if x < 5 then Some (string_of_int x,x+1) else None in
+  S.unfoldr f 0
+    |> S.to_list
+    |> OUnit.assert_equal ["0"; "1"; "2"; "3"; "4"]
+
+let test_hashtbl () =
+  let h = 1 -- 5
+    |> S.zip_i
+    |> S.to_hashtbl2 in
+  0 -- 4
+    |> S.iter (fun i -> OUnit.assert_equal (i+1) (Hashtbl.find h i));
+  OUnit.assert_equal [0;1;2;3;4] (S.hashtbl_keys h |> S.sort |> S.to_list);
+  ()
+
+let test_buff () =
+  let b = Buffer.create 4 in
+  "hello world"
+    |> S.of_str |> S.rev |> S.map Char.uppercase
+    |> (fun seq -> S.to_buffer seq b);
+  OUnit.assert_equal "DLROW OLLEH" (Buffer.contents b);
+  ()
+
 let suite =
   "test_sequence" >:::
     [ "test_empty" >:: test_empty;
@@ -161,4 +184,6 @@ let suite =
       "test_scan" >:: test_scan;
       "test_drop" >:: test_drop;
       "test_rev" >:: test_rev;
+      "test_unfoldr" >:: test_unfoldr;
+      "test_hashtbl" >:: test_hashtbl;
     ]