open Printf
open OUnit

module OrderedInt = struct
  type t  = int
  let compare = compare
end

module G = Graph.Make (OrderedInt)
module DSL = Graph.MakeDSL (G)

open DSL

let test_graph _ =
  let g = (G.empty false) +> (1, 2) +> (1, 5) +> (1, 6)
    +> (2, 5) +> (2, 3) +> (3, 4)
    +> (5, 4) in
  let parent = G.bfs 1 g in
  assert_equal ~msg:"parent(2)" (Hashtbl.find parent 2) 1 ;
  assert_equal ~msg:"parent(3)" (Hashtbl.find parent 3) 2 ;
  assert_equal ~msg:"parent(4)" (Hashtbl.find parent 4) 5 ;
  assert_equal ~msg:"parent(5)" (Hashtbl.find parent 5) 1 ;
  assert_equal ~msg:"parent(6)" (Hashtbl.find parent 6) 1

let suite = "Graph" >::: [
  "graph" >:: test_graph
]

let _ = run_test_tt_main suite