lambdaTests.ml 25.1 KB
Newer Older
1
open OUnit2
Pietro Abate's avatar
Pietro Abate committed
2
3
open Camlp4.PreCast 

Pietro Abate's avatar
Pietro Abate committed
4
(* Typed -> Lambda *)
5
let run_test_compile msg expected totest =
Pietro Abate's avatar
Pietro Abate committed
6
7
8
9
  let aux str =
    try
      let expr = Parse.ExprParser.of_string_no_file str in
      let env, texpr = Compute.to_typed expr in
Pietro Abate's avatar
Pietro Abate committed
10
      Format.printf "Computed Typed -> %s%!@." (Printer.typed_to_string texpr);
Pietro Abate's avatar
Pietro Abate committed
11
12
13
14
15
      let lambdaexpr = Compile.compile env texpr in
      Printer.lambda_to_string lambdaexpr
    with
      | Compute.Error -> exit 3
      | Loc.Exc_located (loc, exn) ->
16
17
18
19
20
	let l = Loc.start_line loc in
	let cbegin = Loc.start_off loc - Loc.start_bol loc in
	let cend = Loc.stop_off loc - Loc.start_bol loc in
	Printf.eprintf "File %s, line %d, characters %d-%d:\n" (Loc.file_name loc) l
	  cbegin cend; raise exn
Pietro Abate's avatar
Pietro Abate committed
21
22
      | e -> Printf.eprintf "Runtime error.\n"; raise e
  in
23
  fun _ -> assert_equal ~msg:msg ~printer:(fun x -> x) expected (aux totest)
Pietro Abate's avatar
Pietro Abate committed
24
25

let tests_poly_abstr = [
26
27
28
  "Test CDuce.lambda.const_abstr failed",
  "Abstraction(Dummy,,,,Sel(,(Int -> Int),{}))",
  "fun f x : Int : Int -> 2";
Pietro Abate's avatar
Pietro Abate committed
29

30
  "Test CDuce.lambda.poly.identity failed",
31
32
33
  "Abstraction(Dummy,,,,Sel(,([ Char* ] | Int -> [ Char* ] | Int),Comp({},{ { `$A = 
Int } ,{ `$A = [ Char* ]
 } })))",
Julien Lopez's avatar
Julien Lopez committed
34
  "(fun f x : 'A[{A/Int},{A/String}] : 'A[{A/Int},{A/String}] -> x) [{A/Int},{A/String}]";
Pietro Abate's avatar
Pietro Abate committed
35
36

  "Test CDuce.runtime.poly.tail failed",
37
38
39
40
41
42
43
  "Abstraction(Dummy,,,,Sel(,([ (`$A & Int | Char | Atom | (Any,Any) |
                             <(Any) (Any)>Any | Arrow)* ] -> [ (`$A & Int |
                                                               Char | 
                                                               Atom |
                                                               (Any,Any) |
                                                               <(Any) (Any)>Any |
                                                               Arrow)* ]),{}))",
44
  "fun tail x : ['A[{}]] : ['A[{}]] -> match x : ['A[{}]] with | (el : 'A[{}]) :: (rest : ['A[{}]]) -> rest";
Pietro Abate's avatar
Pietro Abate committed
45

46
47
48
49
  "Test CDuce.runtime.poly.pair failed", "Abstraction(Dummy,,,,Sel(,((`$A & Int | Char | Atom | (Any,Any) |
                           <(Any) (Any)>Any | Arrow,`$B & Int | Char | 
                           Atom | (Any,Any) | <(Any) (Any)>Any | Arrow) -> 
`$A & Int | Char | Atom | (Any,Any) | <(Any) (Any)>Any | Arrow),{}))",
50
  "fun pair x : ('A[{}] * 'B[{}]) : 'A[{}] -> match x : ('A[{}] * 'B[{}]) with | (z : 'A[{}], y : 'B[{}]) -> z";
51

52
53
54
55
56
57
58
59
60
  "Test CDuce.runtime.poly.match_abstr failed", "Apply(Match(Abstraction(Dummy,,,,Sel(,(`$A & Int | Char | Atom | (Any,Any) |
                                       <(Any) (Any)>Any | Arrow -> `$A & Int |
                                                                   Char |
                                                                   Atom |
                                                                   (Any,Any) |
                                                                   <(Any) (Any)>Any |
                                                                   Arrow),{})), {accept_chars=false; brs_disp=<disp>; brs_rhs=[| (2, TVar(Local(0),Comp({},{ { `$A = 
Int
 } }))) |]; brs_stack_pos=0}),Const(3))",
61
  "(match (fun f x : 'A[{}] : 'A[{}] -> x) : ('A[{}] -> 'A[{}]) with | y : ('A[{}] -> 'A[{}]) -> y[{A/Int}]).3";
62
63
64



Pietro Abate's avatar
Pietro Abate committed
65
66
]

Pietro Abate's avatar
Pietro Abate committed
67
68
69
70
let tests_poly_abstr = [
  "Test CDuce.lambda.const_abstr failed",
  "Abstraction(Dummy,,,,Sel((Int -> Int),{}))",
  "fun f x : 'A[{A/Int}] : 'A[{A/Int}] -> 2";
Pietro Abate's avatar
Pietro Abate committed
71
72
73
74

  "Test CDuce.lambda.const_abstr failed",
  "Abstraction(Dummy,,,,Sel(,(Int -> Int),{}))",
  "fun f x : Int : Int -> 2";
Pietro Abate's avatar
Pietro Abate committed
75
76
];;

77
78
let tests_compile = "CDuce compile tests (Typed -> Lambda )" >:::
  List.map (fun (m,e,f) -> f >:: run_test_compile m e f) tests_poly_abstr
79

80
81
82
83
84
85
86
87
88
89
90
91
92
(* Typed -> Lambda -> Value *)
let run_test_eval str =
  try
    let expr = Parse.ExprParser.of_string_no_file str in
    let env, texpr = Compute.to_typed expr in
    let evalexpr = Compile.compile_eval_expr env texpr in
    Printer.value_to_string evalexpr
  with
    | Compute.Error -> exit 3
    | Loc.Exc_located (loc, exn) ->
      let l = Loc.start_line loc in
      let cbegin = Loc.start_off loc - Loc.start_bol loc in
      let cend = Loc.stop_off loc - Loc.start_bol loc in
93
94
      Printf.eprintf "File %s, line %d, characters %d-%d:\n" (Loc.file_name loc)
	l cbegin cend; raise exn
95
    | e -> Printf.eprintf "Runtime error.\n"; raise e
96

97
let tests_eval = "CDuce runtime tests (Typed -> Lambda -> Value)" >:::
98
99
  [
    "abstr" >:: ( fun test_ctxt ->
100
101
102
      assert_equal ~msg:"Test CDuce.runtime.abstr.let_simple failed"
	~printer:(fun x -> x) "3"
	(run_test_eval "let x : Int = 3 in x : Int");
103
104
105
106
107
108
109
110
111
112
113
114
115
116
      assert_equal ~msg:"Test CDuce.runtime.abstr.let_medium failed"
	~printer:(fun x -> x) "2"
	(run_test_eval "let f : (Int -> Int) = (fun (Int -> Int) | x : Int -> x)
                        in (f.f.2) : Int");
      assert_equal ~msg:"Test CDuce.runtime.abstr.let_nested_simple failed"
	~printer:(fun x -> x) "3"
	(run_test_eval "let f : (Int -> Int) = (fun (Int -> Int) | x : Int -> x + 1)
                        in (let x : Int = f.2
                            in x : Int) : Int");
      assert_equal ~msg:"Test CDuce.runtime.abstr.let_nested_medium failed"
	~printer:(fun x -> x) "4"
	(run_test_eval "let f : (Int -> Int) = (fun (Int -> Int) | x : Int -> x + 1)
                        in (let x : Int = f.2
                            in f.x : Int) : Int");
117
      assert_equal ~msg:"Test CDuce.runtime.abstr.simple failed"
118
	~printer:(fun x -> x) "Abstraction((Int, Int),{})"
119
	(run_test_eval "fun f x : Int : Int -> 2");
120
      assert_equal ~msg:"Test CDuce.runtime.abstr.medium failed"
121
	~printer:(fun x -> x)
122
	"Abstraction((Int, [ Char* ] -> [ Int Char* ]),{})"
123
	(run_test_eval "fun f x : Int y : String : (Int*String) -> x,y");
124
125
126
127
    );

    "apply" >:: ( fun test_ctxt ->
      assert_equal ~msg:"Test CDuce.runtime.apply.simple failed"
128
	~printer:(fun x -> x) "2"
129
	(run_test_eval "(fun f x : Int : Int -> x).2");
130
      assert_equal ~msg:"Test CDuce.runtime.apply.simple_pair failed"
131
	~printer:(fun x -> x) "(3, 2, {})"
132
	(run_test_eval "(fun f x : (Int*Int) : (Int*Int) -> x).(3,2)");
133
      assert_equal ~msg:"Test CDuce.runtime.apply.medium failed"
134
	~printer:(fun x -> x) "(2, 3, {})"
135
	(run_test_eval "((fun f x : Int y : Int : (Int*Int) -> x,y).2).3");
136
137
138
    );

    "misc" >:: ( fun test_ctxt ->
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
      assert_equal ~msg:"Test CDuce.runtime.misc.even failed"
	~printer:(fun x -> x)
	"Abstraction((Int, Bool) ,(Any \\ (Int), Any \\ (Int)),{})"
	(run_test_eval "fun ((Int -> Bool) & ((!Int) -> (!Int)))
                          | x : Int -> (match (x % 2) : Int with
                                          | 0 -> `true
                                          | 1 -> `false)
                          | x : (!Int) -> x");
      assert_equal ~msg:"Test CDuce.runtime.misc.even_applied1 failed"
	~printer:(fun x -> x)
	"Atom(false)"
	(run_test_eval "(fun ((Int -> Bool) & ((!Int) -> (!Int)))
                          | x : Int -> (match (x % 2) : Int with
                                          | 0 -> `true
                                          | 1 -> `false)
                          | x : (!Int) -> x).5");
      assert_equal ~msg:"Test CDuce.runtime.misc.even_applied2 failed"
	~printer:(fun x -> x)
	"Atom(true)"
	(run_test_eval "(fun ((Int -> Bool) & ((!Int) -> (!Int)))
                          | x : Int -> (match (x % 2) : Int with
                                          | 0 -> `true
                                          | 1 -> `false)
                          | x : (!Int) -> x).8");
      assert_equal ~msg:"Test CDuce.runtime.misc.even_applied3 failed"
	~printer:(fun x -> x)
165
	"(2, (3, Atom(nil), {}), {})"
166
167
168
169
170
	(run_test_eval "(fun ((Int -> Bool) & ((!Int) -> (!Int)))
                          | x : Int -> (match (x % 2) : Int with
                                          | 0 -> `true
                                          | 1 -> `false)
                          | x : (!Int) -> x).[2; 3]");
171
      assert_equal ~msg:"Test CDuce.runtime.misc.is_int failed"
172
	~printer:(fun x -> x)
173
174
	"Abstraction((Int, Bool) ,(Bool, Bool) ,(Any \\ (Int | Bool), Any \\ (Int | Bool)),{})"
	(run_test_eval "fun ((Int -> Bool) & (Bool -> Bool) & ((!(Int|Bool)) -> (!(Int|Bool))))
175
                          | x : Int -> `true
176
177
                          | x : Bool -> `false
                          | x : (!(Int|Bool)) -> x");
178
179
180
      assert_equal ~msg:"Test CDuce.runtime.misc.is_int_applied1 failed"
	~printer:(fun x -> x)
	"Atom(true)"
181
	(run_test_eval "(fun ((Int -> Bool) & (Bool -> Bool) & ((!(Int|Bool)) -> (!(Int|Bool))))
182
                           | x : Int -> `true
183
184
                           | x : Bool -> `false
                           | x : (!(Int|Bool)) -> x).2");
185
186
187
      assert_equal ~msg:"Test CDuce.runtime.misc.is_int_applied2 failed"
	~printer:(fun x -> x)
	"Atom(false)"
188
	(run_test_eval "(fun ((Int -> Bool) & (Bool -> Bool) & ((!(Int|Bool)) -> (!(Int|Bool))))
189
                           | x : Int -> `true
190
191
                           | x : Bool -> `false
                           | x : (!(Int|Bool)) -> x).`true");
192
193
      assert_equal ~msg:"Test CDuce.runtime.misc.is_int_applied3 failed"
	~printer:(fun x -> x)
194
	"(2, (3, Atom(nil), {}), {})"
195
	(run_test_eval "(fun ((Int -> Bool) & (Bool -> Bool) & ((!(Int|Bool)) -> (!(Int|Bool))))
196
                           | x : Int -> `true
197
198
                           | x : Bool -> `false
                           | x : (!(Int|Bool)) -> x).[2; 3]");
199
      assert_equal ~msg:"Test CDuce.runtime.misc.map failed"
200
	~printer:(fun x -> x)
201
202
203
204
205
206
207
208
	"Abstraction(((`$A & Int | Char | Atom | (Any,Any) | <(Any) (Any)>Any | Arrow) -> 
             (`$B & Int | Char | Atom | (Any,Any) | <(Any) (Any)>Any | 
             Arrow), [ (`$A & Int | Char | Atom | (Any,Any) |
                       <(Any) (Any)>Any | Arrow)* ] -> [ (`$B & Int | 
                                                         Char | Atom |
                                                         (Any,Any) |
                                                         <(Any) (Any)>Any |
                                                         Arrow)* ]),{})"
209
210
211
212
	(run_test_eval "fun map f : ('A[{}]->'B[{}]) x : ['A[{}]] : ['B[{}]] ->
                          match x : ['A[{}]] with
                            | (el : 'A[{}]) :: [] -> f.el
                            | (el : 'A[{}]) :: (rest : ['A[{}]]) -> ((f.el), ((map.f).rest))");
213
214
      assert_equal ~msg:"Test CDuce.runtime.misc.map_even_simple failed"
	~printer:(fun x -> x)
215
	"(\"hey\", (Atom(false), Atom(nil), {}), {})"
Julien Lopez's avatar
Julien Lopez committed
216
217
218
	(run_test_eval "(fun map f : ('A[{A/Int},{A/Bool}]->'B[{A/Int},{A/Bool}]) x : ['A[{A/Int},{A/Bool}]] : ['B[{A/Int},{A/Bool}]] ->
                          match x : ['A[{A/Int},{A/Bool}]] with
                            | (el : 'A[{A/Int},{A/Bool}]) :: (rest : ['A[{A/Int},{A/Bool}]]) -> ((f.el), ((map.f).rest))
219
                            | [] -> []).(fun ((Int -> Bool) & ((!Int) -> (!Int)))
220
221
222
223
224
225
                          | x : Int -> (match (x % 2) : Int with
                                          | 0 -> `true
                                          | 1 -> `false)
                          | x : (!Int) -> x).[\"hey\"; 3]");
      assert_equal ~msg:"Test CDuce.runtime.misc.map_even_medium failed"
	~printer:(fun x -> x)
226
	"(Atom(true), (\"hey\", (Atom(false), (Atom(true), Atom(nil), {}), {}), {}), {})"
Julien Lopez's avatar
Julien Lopez committed
227
228
229
	(run_test_eval "(fun map f : ('A[{A/Int},{A/Bool}]->'B[{A/Int},{A/Bool}]) x : ['A[{A/Int},{A/Bool}]] : ['B[{A/Int},{A/Bool}]] ->
                          match x : ['A[{A/Int},{A/Bool}]] with
                            | (el : 'A[{A/Int},{A/Bool}]) :: (rest : ['A[{A/Int},{A/Bool}]]) -> ((f.el), ((map.f).rest))
230
                            | [] -> []).(fun ((Int -> Bool) & ((!Int) -> (!Int)))
231
232
233
                          | x : Int -> (match (x % 2) : Int with
                                          | 0 -> `true
                                          | 1 -> `false)
234
                          | x : (!Int) -> x).[4; \"hey\"; 3; 2]");
235
236
237
238
239
240
241
242
243
      assert_equal ~msg:"Test CDuce.runtime.misc.map_even_hard failed"
	~printer:(fun x -> x)
	"(Atom(true), (\"hey\", ((3, (5, Atom(nil), {}), {}), (Atom(true), (Abstraction((
`$C & Int | Char | Atom | (Any,Any) | <(Any) (Any)>Any | Arrow, `$C & Int |
                                                                Char | 
                                                                Atom |
                                                                (Any,Any) |
                                                                <(Any) (Any)>Any |
                                                                Arrow),{}), (Atom(false), Atom(nil), {}), {}), {}), {}), {}), {})"
Julien Lopez's avatar
Julien Lopez committed
244
245
246
	(run_test_eval "(fun map f : ('A[{A/Int},{A/Bool}]->'B[{A/Int},{A/Bool}]) x : ['A[{A/Int},{A/Bool}]] : ['B[{A/Int},{A/Bool}]] ->
                          match x : ['A[{A/Int},{A/Bool}]] with
                            | (el : 'A[{A/Int},{A/Bool}]) :: (rest : ['A[{A/Int},{A/Bool}]]) -> ((f.el), ((map.f).rest))
247
248
249
250
                            | [] -> []).(fun ((Int -> Bool) & ((!Int) -> (!Int)))
                          | x : Int -> (match (x % 2) : Int with
                                          | 0 -> `true
                                          | 1 -> `false)
251
                          | x : (!Int) -> x).[4; \"hey\"; [3; 5]; 2; (fun ('C[{}] -> 'C[{}]) | x : 'C[{}] -> x); 3+4]");
252
253
      assert_equal ~msg:"Test CDuce.runtime.misc.map_is_int_simple failed"
	~printer:(fun x -> x)
254
	"(Atom(false), (Atom(true), Atom(nil), {}), {})"
Julien Lopez's avatar
Julien Lopez committed
255
256
257
	(run_test_eval "(fun map f : ('A[{A/Int},{A/Bool}]->'B[{A/Int},{A/Bool}]) x : ['A[{A/Int},{A/Bool}]] : ['B[{A/Int},{A/Bool}]] ->
                          match x : ['A[{A/Int},{A/Bool}]] with
                            | (el : 'A[{A/Int},{A/Bool}]) :: (rest : ['A[{A/Int},{A/Bool}]]) -> ((f.el), ((map.f).rest))
258
                            | [] -> []).(fun ((Int -> Bool) & (Bool -> Bool) & ((!(Int|Bool)) -> (!(Int|Bool))))
259
260
261
262
263
                          | x : Int -> `true
                          | x : Bool -> `false
                          | x : (!(Int|Bool)) -> x).[`true; 3]");
      assert_equal ~msg:"Test CDuce.runtime.misc.map_is_int_medium failed"
	~printer:(fun x -> x)
264
	"(Atom(false), (Atom(true), (Atom(false), Atom(nil), {}), {}), {})"
Julien Lopez's avatar
Julien Lopez committed
265
266
267
	(run_test_eval "(fun map f : ('A[{A/Int},{A/Bool}]->'B[{A/Int},{A/Bool}]) x : ['A[{A/Int},{A/Bool}]] : ['B[{A/Int},{A/Bool}]] ->
                          match x : ['A[{A/Int},{A/Bool}]] with
                            | (el : 'A[{A/Int},{A/Bool}]) :: (rest : ['A[{A/Int},{A/Bool}]]) -> ((f.el), ((map.f).rest))
268
                            | [] -> []).(fun ((Int -> Bool) & (Bool -> Bool) & ((!(Int|Bool)) -> (!(Int|Bool))))
269
270
271
                          | x : Int -> `true
                          | x : Bool -> `false
                          | x : (!(Int|Bool)) -> x).[`true; 3; `true]");
272
      assert_equal ~msg:"Test CDuce.runtime.misc.firsts failed"
273
	~printer:(fun x -> x)
274
	"Abstraction(((Int,Int), X1 -> X1 where X1 = (Int,Int)),{})"
275
	(run_test_eval "fun firsts x : (Int*Int) y : (Int*Int) : (Int*Int) ->
276
277
278
                     match x,y : ((Int*Int)*(Int*Int)) with
                       | (a : Int,_ : Int),(b : Int,_ : Int) -> a,b");
      assert_equal ~msg:"Test CDuce.runtime.misc.firsts_applied failed"
279
	~printer:(fun x -> x) "(5, 1, {})"
280
	(run_test_eval "((fun firsts x : (Int*Int) y : (Int*Int) : (Int*Int) ->
281
282
283
                       match x,y : ((Int*Int)*(Int*Int)) with
                         | (a : Int,_ : Int),(b : Int,_ : Int) -> a,b)
                   .(5, 3)).(1, 4)");
Julien Lopez's avatar
Julien Lopez committed
284
      assert_equal ~msg:"Test CDuce.runtime.misc.applier failed"
285
	~printer:(fun x -> x) "Abstraction((Int, Int -> Int -> Int),{})"
286
	(run_test_eval "fun applier x : Int f : (Int->Int) : Int -> f.x");
Julien Lopez's avatar
Julien Lopez committed
287
      assert_equal ~msg:"Test CDuce.runtime.misc.applier_applied failed"
288
	~printer:(fun x -> x) "2"
289
	(run_test_eval "((fun applier x : Int f : (Int->Int) : Int ->
Julien Lopez's avatar
Julien Lopez committed
290
                       f.x).2).(fun g x : Int : Int -> x)");
291
292
    );

293
294
295
296
297
298
299
300
301
    "arith" >:: ( fun test_ctxt ->
      assert_equal ~msg:"Test CDuce.runtime.arith.simple failed"
	~printer:(fun x -> x) "5"
	(run_test_eval "2+3");
      assert_equal ~msg:"Test CDuce.runtime.arith.complete failed"
	~printer:(fun x -> x) "1"
	(run_test_eval "2+5*7%2-8/4");
    );

302
303
    "match" >:: ( fun test_ctxt ->
      assert_equal ~msg:"Test CDuce.runtime.match.simple failed"
304
	~printer:(fun x -> x) "1"
305
	(run_test_eval "match 1 : Int with | 1 -> 1");
306
      assert_equal ~msg:"Test CDuce.runtime.match.simple_var failed"
307
	~printer:(fun x -> x) "2"
308
	(run_test_eval "(fun f x : Int : Int ->
309
                      match x : Int with | _ : Int -> x).2");
310
      assert_equal ~msg:"Test CDuce.runtime.match.medium failed"
311
	~printer:(fun x -> x) "2"
312
	(run_test_eval "(fun f x : Int : Int ->
313
                      match x : Int with | 1 -> 3 | _ : Int -> x).2");
314
      assert_equal ~msg:"Test CDuce.runtime.match.rec failed"
315
	~printer:(fun x -> x) "3"
316
	(run_test_eval "(fun f x : Int : Int ->
317
                      match x : Int with | 1 -> 3 | _ : Int -> f.1).2");
318
319
320
321
      assert_equal ~msg:"Test CDuce.runtime.match.desugar_if failed"
	~printer:(fun x -> x) "0"
	(run_test_eval "((fun f x : Int y : Int : Int ->
                           match (x = y) : Bool with | `true -> 0 | `false -> 1).2).2");
322
323
324
325
    );

    "string" >:: ( fun test_ctxt ->
      assert_equal ~msg:"Test CDuce.runtime.string.simple failed"
326
	~printer:(fun x -> x) "\"The cake is a lie\""
327
	(run_test_eval "\"The cake is a lie\"");
328
329
    );

Julien Lopez's avatar
Julien Lopez committed
330
331
332
    "list" >:: ( fun test_ctxt ->
      assert_equal ~msg:"Test CDuce.runtime.list.simple failed"
	~printer:(fun x -> x) "1"
333
	(run_test_eval "match [1; 2] : [Int] with
334
335
                     | (el : Int) :: (_ : [Int]) -> el
                     | [] -> 3");
Julien Lopez's avatar
Julien Lopez committed
336
337
      assert_equal ~msg:"Test CDuce.runtime.list.simple2 failed"
	~printer:(fun x -> x) "3"
338
339
340
	(run_test_eval "match [] : [Int] with
                     | (el : Int) :: (_ : [Int]) -> el
                     | [] -> 3");
Julien Lopez's avatar
Julien Lopez committed
341
      assert_equal ~msg:"Test CDuce.runtime.list.tail failed"
342
	~printer:(fun x -> x) "Abstraction(([ Int* ], [ Int* ]),{})"
343
	(run_test_eval "fun tail x : [Int] : [Int] -> match x : [Int] with
344
                     | (_ : Int) :: (rest : [Int]) -> rest");
345
      assert_equal ~msg:"Test CDuce.runtime.list.tail.eval failed"
346
	~printer:(fun x -> x) "(2, (5, Atom(nil), {}), {})"
347
	(run_test_eval "(fun tail x : [Int] : [Int] -> match x : [Int] with
348
                     | (_ : Int) :: (rest : [Int]) -> rest).[1; 2; 5]");
349
      assert_equal ~msg:"Test CDuce.runtime.list.last failed"
350
	~printer:(fun x -> x) "7"
351
	(run_test_eval "(fun f x : [Int] : [Int] -> match x : [Int] with
352
                     | (el : Int) :: [] -> el
353
                     | (_ : Int) :: (rest : [Int]) -> f.rest).[1; 2; 5; 4; 8; 7]");
354
355
356
      assert_equal ~msg:"Test CDuce.runtime.list.plusone failed"
	~printer:(fun x -> x) "(2, (3, (6, (5, (9, 8, {}), {}), {}), {}), {})"
	(run_test_eval "(fun f x : [Int] : [Int] -> match x : [Int] with
357
358
                     | (el : Int) :: [] -> el+1
                     | (el : Int) :: (rest : [Int]) -> ((el+1), (f.rest))).[1; 2; 5; 4; 8; 7]");
359
360
361
362
363
      assert_equal ~msg:"Test CDuce.runtime.list.concat failed"
	~printer:(fun x -> x) "(1, (2, (5, (4, (8, (7, (2, (3, (4, Atom(nil), {}), {}), {}), {}), {}), {}), {}), {}), {})"
	(run_test_eval "(fun concat x : [[Int]] : [Int] -> match x : [[Int]] with
                     | (el : [Int]) :: [] -> el
                     | (el : [Int]) :: (rest : [[Int]]) -> (el@(concat.rest))).[[1; 2; 5; 4; 8; 7]; [2; 3; 4]]");
364
365
366
367
368
369
370
      assert_equal ~msg:"Test CDuce.runtime.list.length_easy failed"
	~printer:(fun x -> x) "6"
	(run_test_eval "(fun length x : [Int] : Int -> match x : [Int] with
                     | [] -> 0
                     | (el : Int) :: (rest : [Int]) -> length.rest + 1).[1; 2; 5; 4; 8; 7]");
      assert_equal ~msg:"Test CDuce.runtime.list.length_hard failed"
	~printer:(fun x -> x) "((6, 0, {}), 2, {})"
Julien Lopez's avatar
Julien Lopez committed
371
372
	(run_test_eval "let length : (['A[{A/Int},{A/Bool}]] -> Int) =
                          (fun f l : ['A[{A/Int},{A/Bool}]] : Int -> match l : ['A[{A/Int},{A/Bool}]] with
373
                             | [] -> 0
Julien Lopez's avatar
Julien Lopez committed
374
                             | (el : 'A[{A/Int},{A/Bool}]) :: (rest : ['A[{A/Int},{A/Bool}]]) -> f.rest + 1)
375
376
                        in
                        (length.[1; 2; 5; 4; 8; 7], length.[], length.[`true; 2]) : (Int*Int*Int)");
377
378
      assert_equal ~msg:"Test CDuce.runtime.list.nth failed"
	~printer:(fun x -> x) "5"
Julien Lopez's avatar
Julien Lopez committed
379
380
381
382
	(run_test_eval "(fun nth l : ['A[{A/(Int|Bool)}]] n : Int : 'A[{A/(Int|Bool)}] ->
                           match l : ['A[{A/(Int|Bool)}]] with
                             | (el : 'A[{A/(Int|Bool)}]) :: [] -> el
                             | (el : 'A[{A/(Int|Bool)}]) :: (rest : ['A[{A/(Int|Bool)}]]) ->
383
                               (if n = 0 then el else nth.rest.(n-1))).[1; 2; 5; `true; 2].2");
384
385
386
387
388
389
390
      assert_equal ~msg:"Test CDuce.runtime.list.rev failed"
	~printer:(fun x -> x) "(2, (Atom(true), (5, (2, (1, Atom(nil), {}), {}), {}), {}), {})"
	(run_test_eval "(fun rev l : ['A[{A/(Int|Bool)}]] : ['A[{A/(Int|Bool)}]] ->
                           match l : ['A[{A/(Int|Bool)}]] with
                             | (el : 'A[{A/(Int|Bool)}]) :: [] -> [el]
                             | (el : 'A[{A/(Int|Bool)}]) :: (rest : ['A[{A/(Int|Bool)}]]) ->
                               (rev.rest) @ [el]).[1; 2; 5; `true; 2]");
Julien Lopez's avatar
Julien Lopez committed
391
392
    );

393
394
    "union" >:: ( fun test_ctxt ->
      assert_equal ~msg:"Test CDuce.runtime.union.identity failed"
395
	~printer:(fun x -> x) "Abstraction(([ Char* ] | Int, [ Char* ] | Int),{})"
396
	(run_test_eval "fun f x : (Int | String) : (Int | String) -> x");
397
      assert_equal ~msg:"Test CDuce.runtime.union.match failed"
398
	~printer:(fun x -> x) "Abstraction(([ Char* ] | Int, [ Char* ] | Int),{})"
399
	(run_test_eval "fun f x : (Int | String) : (Int | String) ->
400
                     match x : (Int | String) with
401
402
                       | _ : Int -> 2
                       | _ : String -> \"Piece of cake\"");
403
404
      assert_equal ~msg:"Test CDuce.runtime.union.match_applied failed"
	~printer:(fun x -> x) "2"
405
	(run_test_eval "(fun f x : (Int | String) : (Int | String) ->
406
                      match x : (Int | String) with
407
408
                        | _ : Int -> 2
                        | _ : String -> \"Piece of cake\").5");
409
410
      assert_equal ~msg:"Test CDuce.runtime.union.match_applied2 failed"
	~printer:(fun x -> x) "\"Piece of cake\""
411
	(run_test_eval "(fun f x : (Int | String) : (Int | String) ->
412
                      match x : (Int | String) with
413
414
                        | _ : Int -> 2
                        | _ : String -> \"Piece of cake\").\"test\"");
415
416
    );

417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
    "union_precise" >:: ( fun test_ctxt ->
      assert_equal ~msg:"Test CDuce.runtime.union_precise.identity failed"
	~printer:(fun x -> x) "Abstraction((Int, Int) ,([ Char* ], [ Char* ]),{})"
	(run_test_eval "fun ((Int -> Int) & (String -> String))
                          | x : (Int | String) -> x");
      assert_equal ~msg:"Test CDuce.runtime.union_precise.match failed"
	~printer:(fun x -> x) "Abstraction((Int, Int) ,([ Char* ], [ Char* ]),{})"
	(run_test_eval "fun ((Int -> Int) & (String -> String))
                          | x : Int -> 2
                          | x : String -> \"Piece of cake\"");
      assert_equal ~msg:"Test CDuce.runtime.union_precise.match_applied failed"
	~printer:(fun x -> x) "2"
	(run_test_eval "(fun ((Int -> Int) & (String -> String))
                           | x : Int -> 2
                           | x : String -> \"Piece of cake\").5");
      assert_equal ~msg:"Test CDuce.runtime.union_precise.match_applied2 failed"
	~printer:(fun x -> x) "\"Piece of cake\""
	(run_test_eval "(fun ((Int -> Int) & (String -> String))
                           | x : Int -> 2
                           | x : String -> \"Piece of cake\").\"test\"");
    );

439
    "poly" >:: ( fun test_ctxt ->
Julien Lopez's avatar
Julien Lopez committed
440
441
      assert_equal ~msg:"Test CDuce.runtime.poly.identity_pure failed"
	~printer:(fun x -> x)
442
443
	"Abstraction((`$A & Int | Char | Atom | (Any,Any) | <(Any) (Any)>Any | Arrow, 
`$A & Int | Char | Atom | (Any,Any) | <(Any) (Any)>Any | Arrow),{})"
444
	(run_test_eval "fun f x : 'A[{}] : 'A[{}] -> x");
445
      assert_equal ~msg:"Test CDuce.runtime.poly.identity failed"
446
	~printer:(fun x -> x)
447
	"Abstraction(([ Char* ] | Int, [ Char* ] | Int),{})"
Julien Lopez's avatar
Julien Lopez committed
448
449
	(run_test_eval "(fun f x : 'A[{A/Int},{A/String}] : 'A[{A/Int},{A/String}] -> x)
                   [{A/Int},{A/String}]");
450
451
      assert_equal ~msg:"Test CDuce.runtime.poly.identity_applied failed"
	~printer:(fun x -> x) "2"
Julien Lopez's avatar
Julien Lopez committed
452
453
	(run_test_eval "((fun f x : 'A[{A/Int},{A/String}] : 'A[{A/Int},{A/String}] -> x)
                   [{A/Int},{A/String}]).2");
454
455
      assert_equal ~msg:"Test CDuce.runtime.poly.identity_applied2 failed"
	~printer:(fun x -> x) "2"
Julien Lopez's avatar
Julien Lopez committed
456
	(run_test_eval "((fun f x : 'A[{A/String}] : 'A[{A/String}] -> x)[{A/String}]).2");
Julien Lopez's avatar
Julien Lopez committed
457
458
      assert_equal ~msg:"Test CDuce.runtime.poly.tail failed"
	~printer:(fun x -> x)
459
460
461
	"Abstraction(([ (`$A & Int | Char | Atom | (Any,Any) | <(Any) (Any)>Any |
               Arrow)* ], [ (`$A & Int | Char | Atom | (Any,Any) |
                            <(Any) (Any)>Any | Arrow)* ]),{})"
462
463
	(run_test_eval "fun tail x : ['A[{}]] : ['A[{}]] -> match x : ['A[{}]] with
                     | (_ : 'A[{}]) :: (rest : ['A[{}]]) -> rest");
Julien Lopez's avatar
Julien Lopez committed
464
      assert_equal ~msg:"Test CDuce.runtime.poly.tail_applied failed"
465
	~printer:(fun x -> x) "(7, (8, (5, Atom(nil), {}), {}), {})"
466
467
	(run_test_eval "(fun tail x : ['A[{}]] : ['A[{}]] -> match x : ['A[{}]] with
                     | (_ : 'A[{}]) :: (rest : ['A[{}]]) -> rest).[3; 7; 8; 5]");
468

469
470
    );

471
472
  ]

Pietro Abate's avatar
Pietro Abate committed
473
474
475
476
let _ =
  run_test_tt_main (
    test_list
      [ tests_compile;
Pietro Abate's avatar
Pietro Abate committed
477
        (* tests_eval *)
Pietro Abate's avatar
Pietro Abate committed
478
479
480
      ]
  )
;;