ligo/src/test/contracts/map.ligo

// Test map type and related built-in functions in PascaLIGO

type foobar is map(int, int)

const fb : foobar = map
  23 -> 0 ;
  42 -> 0 ;
end

function set_ (var n : int ; var m : foobar) : foobar is block {
  m[23] := n ;
} with m


function rm (var m : foobar) : foobar is block {
  remove 42 from map m
} with m

function size_ (const m : foobar) : nat is
  block {skip} with (size(m))

function gf (const m : foobar) : int is begin skip end with get_force(23, m)

function get (const m : foobar) : option(int) is
  begin
    skip
  end with m[42]

const bm : foobar = map
  144 -> 23 ;
  51 -> 23 ;
  42 -> 23 ;
  120 -> 23 ;
  421 -> 23 ;
end

function iter_op (const m : foobar) : int is
  var r : int := 0 ;
  function aggregate (const i : int ; const j : int) : unit is block { r := r + i + j } with unit ;
  block {
    map_iter(m , aggregate) ;
  } with r ;

function map_op (const m : foobar) : foobar is
  function increment (const i : int ; const j : int) : int is block { skip } with j + 1 ;
  block { skip } with map_map(m , increment) ;

function fold_op (const m : foobar) : int is
  function aggregate (const i : int ; const j : (int * int)) : int is block { skip } with i + j.0 + j.1 ;
  block { skip } with map_fold(m , 10 , aggregate)