test_ancient.ml

   1 (* Very basic tests of Ancient module.
   2  * $Id: test_ancient.ml,v 1.1 2006-09-27 12:07:07 rich Exp $
   3  *)
   4
   5 open Printf
   6
   7 type item = {
   8   name : string;
   9   dob : string;
  10   address : string;
  11   phone : string option;
  12   marital_status : marital_status;
  13   id : int;
  14 }
  15 and marital_status = Single | Married | Divorced
  16
  17 let gc_compact () =
  18   eprintf "compacting ... %!";
  19   Gc.compact ();
  20   let stat = Gc.stat () in
  21   let live_words = stat.Gc.live_words in
  22   eprintf "live words = %d (%d MB)\n%!"
  23     live_words (live_words * 8 / 1024 / 1024)
  24
  25 let random_string () =
  26   let n = 1 + Random.int 20 in
  27   let str = String.create n in
  28   for i = 0 to n-1 do
  29     let c = 97 + Random.int 26 in
  30     let c = Char.chr c in
  31     str.[i] <- c
  32   done;
  33   str
  34
  35 let random_string_option () =
  36   if Random.int 3 = 1 then None else Some (random_string ())
  37
  38 let random_marital_status () =
  39   match Random.int 3 with
  40   | 0 -> Single
  41   | 1 -> Married
  42   | _ -> Divorced
  43
  44 let rec output_data chan data =
  45   let n = Array.length data in
  46   for i = 0 to n-1; do
  47     output_item chan data.(i)
  48   done
  49
  50 and output_item chan item =
  51   fprintf chan "id = %d\n%!" item.id;
  52   fprintf chan "\tname = %s\n%!" item.name;
  53   fprintf chan "\tdob = %s\n%!" item.dob;
  54   fprintf chan "\taddress = %s\n%!" item.address;
  55   fprintf chan "\tphone = %s\n%!"
  56     (match item.phone with
  57      | None -> "None"
  58      | Some str -> "Some " ^ str);
  59   fprintf chan "\tmarital_status = %s\n%!"
  60     (string_of_marital_status item.marital_status)
  61
  62 and string_of_marital_status status =
  63   match status with
  64   | Single -> "Single"
  65   | Married -> "Married"
  66   | Divorced -> "Divorced"
  67
  68 let () =
  69   eprintf "Before allocating data on OCaml heap ...\n";
  70   gc_compact ();
  71   let data =
  72     Array.init 100000 (
  73       fun id ->
  74         { id = id;
  75           name = random_string ();
  76           dob = random_string ();
  77           address = random_string ();
  78           phone = random_string_option ();
  79           marital_status = random_marital_status () }
  80     ) in
  81   eprintf "After allocating data on OCaml heap ...\n";
  82   gc_compact ();
  83
  84   let chan = open_out "test_ancient.out1" in
  85   output_data chan data;
  86   close_out chan;
  87
  88   let data = Ancient.mark data in
  89   eprintf "After marking data as ancient ...\n";
  90   gc_compact ();
  91
  92   let data = Ancient.follow data in
  93   eprintf "Number of elements in array = %d\n" (Array.length data);
  94
  95   let chan = open_out "test_ancient.out2" in
  96   output_data chan data;
  97   close_out chan;
  98
  99   eprintf "After writing out ancient data ...\n";
 100   gc_compact ()