File: assert.sml

package info (click to toggle)
mlton 20130715-3
  • links: PTS
  • area: main
  • in suites: stretch
  • size: 60,900 kB
  • ctags: 69,386
  • sloc: xml: 34,418; ansic: 17,399; lisp: 2,879; makefile: 1,605; sh: 1,254; pascal: 256; python: 143; asm: 97
file content (43 lines) | stat: -rw-r--r-- 1,580 bytes parent folder | download | duplicates (9)
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
(* assert.sml
 * 2004 Matthew Fluet (mfluet@acm.org)
 *  Ported to MLton threads.
 *)

structure Assert: ASSERT =
   struct
      structure C = Critical
      val assertFlg = false

      fun fail msg =
         (C.atomicBegin ();
          TextIO.print (concat ["ASSERT: ", msg, "\n"]);
          OS.Process.exit OS.Process.failure)

      fun assert (msgs: (unit -> string) list, 
                  msg: unit -> string, 
                  f: unit -> bool): unit =
         if assertFlg andalso not (f () handle _ => false)
            then let
                    val msgs = List.map (fn f => f ()) msgs
                    val msg = concat [String.concatWith " " msgs, " :: ", msg ()]
                 in
                    fail msg
                 end
            else ()
      fun assert' (msg: string, f: unit -> bool): unit =
         assert ([], fn () => msg, f)

      datatype z = datatype MLton.Thread.AtomicState.t
      fun assertAtomic (msg: unit -> string, n: int option): unit =
         assert ([C.atomicMsg], msg, fn () =>
                 case MLton.Thread.atomicState () of
                    Atomic m => (case n of NONE => true | SOME n => n = m)
                  | NonAtomic => false)
      fun assertNonAtomic (msg: unit -> string): unit =
         assert ([C.atomicMsg], msg, fn () =>
                 case MLton.Thread.atomicState () of
                    Atomic _ => false
                  | NonAtomic => true)
      fun assertAtomic' (msg, n) = assertAtomic (fn () => msg, n)
      fun assertNonAtomic' msg = assertNonAtomic (fn () => msg)
   end