1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
open! Import
open Uniform_array
let does_raise = Exn.does_raise
let zero_obj = Caml.Obj.repr (0 : int)
(* [create_obj_array] *)
let%test_unit _ =
let t = create_obj_array ~len:0 in
assert (length t = 0)
;;
(* [create] *)
let%test_unit _ =
let str = Caml.Obj.repr "foo" in
let t = create ~len:2 str in
assert (phys_equal (get t 0) str);
assert (phys_equal (get t 1) str)
;;
let%test_unit _ =
let float = Caml.Obj.repr 3.5 in
let t = create ~len:2 float in
assert (Caml.Obj.tag (Caml.Obj.repr t) = 0);
(* not a double array *)
assert (phys_equal (get t 0) float);
assert (phys_equal (get t 1) float);
set t 1 (Caml.Obj.repr 4.);
assert (Float.( = ) (Caml.Obj.obj (get t 1)) 4.)
;;
(* [empty] *)
let%test _ = length empty = 0
let%test _ = does_raise (fun () -> get empty 0)
(* [singleton] *)
let%test _ = length (singleton zero_obj) = 1
let%test _ = phys_equal (get (singleton zero_obj) 0) zero_obj
let%test _ = does_raise (fun () -> get (singleton zero_obj) 1)
let%test_unit _ =
let f = 13. in
let t = singleton (Caml.Obj.repr f) in
invariant t;
assert (Poly.equal (Caml.Obj.repr f) (get t 0))
;;
(* [get], [unsafe_get], [set], [unsafe_set], [unsafe_set_assuming_currently_int] *)
let%test_unit _ =
let t = create_obj_array ~len:1 in
assert (length t = 1);
assert (phys_equal (get t 0) zero_obj);
assert (phys_equal (unsafe_get t 0) zero_obj);
let one_obj = Caml.Obj.repr (1 : int) in
let check_get expect =
assert (phys_equal (get t 0) expect);
assert (phys_equal (unsafe_get t 0) expect)
in
set t 0 one_obj;
check_get one_obj;
unsafe_set t 0 zero_obj;
check_get zero_obj;
unsafe_set_assuming_currently_int t 0 one_obj;
check_get one_obj
;;
let%expect_test "exists" =
let test arr f = of_list arr |> exists ~f in
let r here = require_equal here (module Bool) in
r [%here] false (test [] Fn.id);
r [%here] true (test [ true ] Fn.id);
r [%here] true (test [ false; false; false; false; true ] Fn.id);
r [%here] true (test [ 0; 1; 2; 3; 4 ] (fun i -> i % 2 = 1));
r [%here] false (test [ 0; 2; 4; 6; 8 ] (fun i -> i % 2 = 1));
[%expect {| |}]
;;
let%expect_test "iteri" =
let test arr = of_list arr |> iteri ~f:(printf "(%d %c)") in
test [];
[%expect {| |}];
test [ 'a' ];
[%expect {| (0 a) |}];
test [ 'a'; 'b'; 'c'; 'd' ];
[%expect {| (0 a)(1 b)(2 c)(3 d) |}]
;;
module Sequence = struct
type nonrec 'a t = 'a t
type 'a z = 'a
let length = length
let get = get
let set = set
let create_bool ~len = create ~len false
end
include Base_for_tests.Test_blit.Test1 (Sequence) (Uniform_array)
let%expect_test "map2_exn" =
let test a1 a2 f =
let result = map2_exn ~f (of_list a1) (of_list a2) in
print_s [%message (result : int Uniform_array.t)]
in
test [] [] (fun _ -> failwith "don't call me");
[%expect {| (result ()) |}];
test [ 1; 2; 3 ] [ 100; 200; 300 ] ( + );
[%expect {| (result (101 202 303)) |}];
require_does_raise [%here] (fun () -> test [ 1 ] [] (fun _ _ -> 0));
[%expect {| (Invalid_argument Array.map2_exn) |}]
;;
|
