import rand const ( strings = unique_strings(20000, 10) ) fn unique_strings(arr_len int, str_len int) []string { mut arr := []string{cap: arr_len} for arr.len < arr_len { str := rand.string(str_len) if str !in arr { arr << str } } return arr } fn test_get_and_set_many() { mut m := map[string]int{} for i, s in strings { m[s] = i assert m[s] == i assert m.len == i + 1 } for i, s in strings { assert m[s] == i } assert m.len == strings.len } fn test_for_in_many() { mut m := map[string]int{} for i, s in strings { m[s] = i } for k, v in m { assert m[k] == v } } fn test_keys_many() { mut m := map[string]int{} for i, s in strings { m[s] = i } keys := m.keys() assert keys.len == strings.len assert keys.len == m.len assert keys == strings } fn test_deletes_many() { mut m := map[string]int{} for i, s in strings { m[s] = i } for i, s in strings { m.delete(s) assert m[s] == 0 assert m.len == strings.len - (i + 1) } assert m.len == 0 assert m.keys().len == 0 } struct User { mut: name string } struct Aaa { mut: m map[string]int users map[string]User } fn (mut a Aaa) set(key string, val int) { a.m[key] = val } fn test_map() { mut m := map[string]int{} assert m.len == 0 m['hi'] = 80 m['hello'] = 101 assert m['hi'] == 80 assert m['hello'] == 101 assert m.len == 2 assert 'hi' in m mut sum := 0 // Test `for in` for _, val in m { sum += val } assert sum == 80 + 101 // Test `.keys()` keys := m.keys() assert keys.len == 2 assert 'hi' in keys assert 'hello' in keys m.delete('hi') assert m.len == 1 m.delete('aloha') assert m.len == 1 assert m['hi'] == 0 assert m.keys().len == 1 assert m.keys()[0] == 'hello' // // mut users := map[string]User{} users['1'] = User{'Peter'} peter := users['1'] assert peter.name == 'Peter' mut a := Aaa{ m: map[string]int{} users: map[string]User{} } a.users['Bob'] = User{'Bob'} q := a.users['Bob'] assert q.name == 'Bob' // test struct field change a.users['Bob'].name = 'bob' q2 := a.users['Bob'] assert q2.name == 'bob' a.m['one'] = 1 a.set('two', 2) assert a.m['one'] == 1 assert a.m['two'] == 2 } fn test_map_init() { one := 'one' three := 'three' m := { one: 1 'two': 2 three: 1 + 2 } assert m['one'] == 1 assert m['two'] == 2 assert m['three'] == 3 assert m['unknown'] == 0 } fn test_string_map() { // m := map[string]Fn } fn test_large_map() { // ticks := time.ticks() mut nums := map[string]int{} n := 30 * 1000 for i in 0 .. n { key := i.str() nums[key] = i } assert nums['1'] == 1 assert nums['999'] == 999 assert nums['1000000'] == 0 // println(time.ticks() - ticks) } fn test_various_map_value() { mut m1 := map[string]int{} m1['test'] = 1 assert m1['test'] == 1 mut m2 := map[string]string{} m2['test'] = 'test' assert m2['test'] == 'test' mut m3 := map[string]i8{} m3['test'] = i8(0) assert m3['test'] == i8(0) mut m4 := map[string]i16{} m4['test'] = i16(0) assert m4['test'] == i16(0) mut m7 := map[string]u16{} m7['test'] = u16(0) assert m7['test'] == u16(0) mut m8 := map[string]u32{} m8['test'] = u32(0) assert m8['test'] == u32(0) mut m9 := map[string]bool{} m9['test'] = true assert m9['test'] == true mut m10 := map[string]byte{} m10['test'] = byte(0) assert m10['test'] == byte(0) mut m11 := map[string]f32{} m11['test'] = f32(0.0) assert m11['test'] == f32(0.0) mut m12 := map[string]f64{} m12['test'] = f64(0.0) assert m12['test'] == f64(0.0) // mut m13 := map[string]rune // m13['test'] = rune(0) // assert m13['test'] == rune(0) mut m14 := map[string]voidptr{} m14['test'] = voidptr(0) assert m14['test'] == voidptr(0) mut m15 := map[string]&byte{} m15['test'] = &byte(0) assert m15['test'] == &byte(0) mut m16 := map[string]i64{} m16['test'] = i64(0) assert m16['test'] == i64(0) mut m17 := map[string]u64{} m17['test'] = u64(0) assert m17['test'] == u64(0) mut m18 := map[string]&int{} m18['test'] = &int(0) assert m18['test'] == &int(0) } fn test_string_arr() { mut m := map[string][]string{} m['a'] = ['one', 'two'] assert m['a'].len == 2 assert m['a'][0] == 'one' assert m['a'][1] == 'two' } fn mut_map(mut m map[string]int) { m['a'] = 10 } fn test_mut_arg() { mut m := map[string]int{} mut_map(mut m) a := m['a'] assert a == 10 } fn test_delete() { mut m := map[string]int{} m['one'] = 1 m['two'] = 2 println(m['two']) // => "2" m.delete('two') println(m['two'].str()) // => 0 assert ('two' in m) == false println('two' in m) // => true, on Linux and Windows <-- wrong ! } fn test_delete_size() { arr := ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'] mut m := map[string]int{} for _ in 0 .. 10 { for i in 0 .. 10 { m[arr[i]] = i } assert m.len == 10 println(m.len) for i in 0 .. 10 { m.delete(arr[i]) } } } fn test_nested_for_in() { mut m := map[string]int{} for i in 0 .. 1000 { m[i.str()] = i } mut i := 0 for key1, _ in m { assert key1 == i.str() i++ mut j := 0 for key2, _ in m { assert key2 == j.str() j++ } } } fn test_delete_in_for_in() { mut m := map[string]string{} for i in 0 .. 1000 { m[i.str()] = i.str() } mut i := 0 for key, _ in m { assert key == i.str() m.delete(key) i++ } assert m.str() == '{}' assert m.len == 0 } fn test_set_in_for_in() { mut m := map[string]string{} for i in 0 .. 10 { m[i.str()] = i.str() } mut last_key := '' mut i := 0 for key, _ in m { m['10'] = '10' assert key == i.str() last_key = key i++ } assert last_key == '10' } fn test_delete_and_set_in_for_in() { mut m := map[string]string{} for i in 0 .. 1000 { m[i.str()] = i.str() } mut i := 0 for key, _ in m { assert key == i.str() m.delete(key) m[key] = i.str() if i == 999 { break } i++ } assert m.len == 1000 i = 0 for key, _ in m { assert m[key] == i.str() i++ } assert i == 1000 } struct Mstruct1 { pub mut: mymap map[string]int } struct Mstruct2 { pub mut: mymap map[string]f64 } struct Mstruct3 { pub mut: mymap map[string]u16 } fn test_map_assign() { mut a := map[string]f64{} mut b := map[string]int{} mut c := map[string]u16{} a = { 'x': 12.4 'y': 3 } b = { 'u': -13 'v': 12 } c = { 's': u16(5) 't': 3 } _ := Mstruct1{{ 'p': 12 }} _ := Mstruct2{{ 'q': 1.7 }} _ := Mstruct3{{ 'r': u16(6) 's': 5 }} } fn test_postfix_op_directly() { mut a := map[string]int{} a['aaa']++ assert a['aaa'] == 1 a['aaa']++ assert a['aaa'] == 2 a['bbb']-- assert a['bbb'] == -1 a['bbb']-- assert a['bbb'] == -2 } fn test_map_push_directly() { mut a := map[string][]string{} a['aaa'] << ['a', 'b', 'c'] assert a['aaa'].len == 3 assert a['aaa'] == ['a', 'b', 'c'] } fn test_assign_directly() { mut a := map[string]int{} a['aaa'] += 4 assert a['aaa'] == 4 a['aaa'] -= 2 assert a['aaa'] == 2 } fn test_map_in_directly() { for k, v in { 'aa': 1 } { assert k == 'aa' assert v == 1 } } fn test_plus_assign_string() { mut m := { 'one': '' } m['one'] += '1' assert m.len == 1 assert m['one'] == '1' } fn test_map_keys_to_array() { m := { 'a': 'b' 'c': 'd' } mut arr := []string{} for k, _ in m { arr << k } sarr := arr.str() println(sarr) assert sarr == "['a', 'c']" } fn map_in_mut(mut m map[string]int) { if 'one' in m { m['one'] = 2 } } fn test_map_in_mut() { mut m := { 'one': 1 } map_in_mut(mut m) assert m['one'] == 2 } fn test_map_in() { m := { 'Foo': 'bar' } if 'foo'.capitalize() in m { println('ok') } else { assert false } } fn mut_map_with_relation_op_in_fn(mut m map[string]int) { if m['one'] == 1 { m['three'] = 3 } if m['two'] != 1 { m['four'] = 4 } if m['one'] > 0 { m['five'] = 5 } if m['one'] < 2 { m['six'] = 6 } if m['two'] >= 2 { m['seven'] = 7 } if m['two'] <= 2 { m['eight'] = 8 } } fn test_mut_map_with_relation_op_in_fn() { mut m := { 'one': 1 'two': 2 } mut_map_with_relation_op_in_fn(mut m) assert 'three' in m assert 'four' in m assert 'five' in m assert 'six' in m assert 'seven' in m assert 'eight' in m } fn test_map_str_after_delete() { mut m := { 'first': 1 'second': 2 'third': 3 } osm := '$m' m.delete('second') nsm := '$m' println('m: $m') assert osm == "{'first': 1, 'second': 2, 'third': 3}" assert nsm == "{'first': 1, 'third': 3}" } fn test_modify_map_value() { mut m1 := { 'foo': 3 'bar': -7 } m1['foo'] += 5 m1['bar'] *= -2 assert m1['foo'] == 8 assert m1['bar'] == 14 } fn test_map_clone() { mut nums := { 'foo': 1 'bar': 2 } mut nums2 := nums.clone() nums2['foo']++ nums2['bar'] *= 4 assert nums['foo'] == 1 assert nums['bar'] == 2 assert nums2['foo'] == 2 assert nums2['bar'] == 8 } struct MValue { name string misc map[string]string } fn test_map_default_zero() { m := map[string]MValue{} v := m['unknown'] x := v.misc['x'] println(x) assert x == '' } fn test_map_or() { m := { 'first': 1 'second': 2 'third': 3 } _ = m // num := m['first'] or { return } } fn test_int_keys() { mut m := map[int]int{} m[3] = 9 m[4] = 16 assert m.len == 2 assert m[3] == 9 assert m[4] == 16 m[5] += 24 m[5]++ assert m[5] == 25 mut m2 := { 3: 9 4: 16 5: 25 } four := 4 m2.delete(3) m2.delete(four) m2.delete(5) assert m2.len == 0 assert m2[3] == 0 assert m2[4] == 0 assert m2[5] == 0 assert m2.keys() == [] m2 = { 3: 9 4: 16 5: 25 } assert m2.len == 3 // clone mc := m.clone() same := mc == m assert same assert mc.len == 3 assert mc.keys() == [3, 4, 5] mut all := []int{} for k, v in mc { assert m[k] == v all << k all << v } assert all == [3, 9, 4, 16, 5, 25] mut m3 := { 1: 'one' 2: 'two' } assert m3[1] == 'one' m3.delete(1) } enum Color { red green blue } type ColorAlias = Color fn test_alias_enum() { mut m := map[ColorAlias]string{} m[Color.red] = 'hi' assert m[Color.red] == 'hi' } fn test_enum_in_map() { mut m := map[Color]string{} m[Color.red] = 'hi' assert Color.red in m assert Color.green !in m assert Color.blue !in m } fn test_voidptr_keys() { mut m := map[voidptr]string{} v := 5 m[&v] = 'var' m[&m] = 'map' assert m[&v] == 'var' assert m[&m] == 'map' assert m.len == 2 } fn test_rune_keys() { mut m := { `!`: 2 `%`: 3 } assert typeof(m).name == 'map[rune]int' assert m[`!`] == 2 m[`@`] = 7 assert m.len == 3 println(m) assert '$m' == '{`!`: 2, `%`: 3, `@`: 7}' mut a := []rune{} for k, v in m { a << k a << rune(v) + `0` } assert a == [`!`, `2`, `%`, `3`, `@`, `7`] } fn test_eq() { a := { 'a': 1 'b': 2 } assert a == { 'a': 1 'b': 2 } b := { 'a': [[1]] 'b': [[2]] } assert b == { 'a': [[1]] 'b': [[2]] } c := { 'a': { '11': 1 } 'b': { '22': 2 } } assert c == { 'a': { '11': 1 } 'b': { '22': 2 } } d := { 'a': MValue{ name: 'aa' misc: { '11': '1' } } 'b': MValue{ name: 'bb' misc: { '22': '2' } } } assert d == { 'a': MValue{ name: 'aa' misc: { '11': '1' } } 'b': MValue{ name: 'bb' misc: { '22': '2' } } } } fn test_non_string_key_map_str() { assert { 23: 4 }.str() == '{23: 4}' assert { `a`: 12 `b`: 13 }.str() == '{`a`: 12, `b`: 13}' assert { 23: 'foo' 25: 'bar' }.str() == "{23: 'foo', 25: 'bar'}" } fn test_map_assign_empty_map_init() { mut a := { 'one': 1 } a = {} println(a) assert a == map[string]int{} assert '$a' == '{}' } fn test_in_map_literal() { assert 1 in { 1: 'one' } } fn test_byte_keys() { mut m := map[byte]byte{} byte_max := byte(255) for i in byte(0) .. byte_max { m[i] = i assert m[i] == i } for k, v in m { assert k == v } for i in byte(0) .. 100 { m[i]++ assert m[i] == i + 1 } assert m.len == byte_max keys := m.keys() for i in byte(0) .. byte_max { assert keys[i] == i } for i in byte(0) .. byte_max { m.delete(i) assert m[i] == 0 } assert m.len == 0 } fn test_i16_keys() { mut m := map[i16]i16{} end := i16(1000) for i in i16(0) .. end { m[i] = i assert m[i] == i } for k, v in m { assert k == v } for i in i16(0) .. 500 { m[i]++ assert m[i] == i + 1 } assert m.len == end keys := m.keys() for i in i16(0) .. end { assert keys[i] == i } for i in i16(0) .. end { m.delete(i) assert m[i] == 0 } assert m.len == 0 } fn test_u16_keys() { mut m := map[u16]u16{} end := u16(1000) for i in u16(0) .. end { m[i] = i assert m[i] == i } for k, v in m { assert k == v } for i in u16(0) .. 500 { m[i]++ assert m[i] == i + 1 } assert m.len == end keys := m.keys() for i in u16(0) .. end { assert keys[i] == i } for i in u16(0) .. end { m.delete(i) assert m[i] == 0 } assert m.len == 0 } fn test_u32_keys() { mut m := map[u32]u32{} end := u32(1000) for i in u32(0) .. end { m[i] = i assert m[i] == i } for k, v in m { assert k == v } for i in u32(0) .. 500 { m[i]++ assert m[i] == i + 1 } assert m.len == end keys := m.keys() for i in u32(0) .. end { assert keys[i] == i } for i in u32(0) .. end { m.delete(i) assert m[i] == 0 } assert m.len == 0 } fn test_int_keys2() { mut m := map[int]int{} end := 1000 for i in int(0) .. end { m[i] = i assert m[i] == i } for k, v in m { assert k == v } for i in int(0) .. 500 { m[i]++ assert m[i] == i + 1 } assert m.len == end keys := m.keys() for i in int(0) .. end { assert keys[i] == i } for i in int(0) .. end { m.delete(i) assert m[i] == 0 } assert m.len == 0 } fn test_i64_keys() { mut m := map[i64]i64{} end := i64(1000) for i in i64(0) .. end { m[i] = i assert m[i] == i } for k, v in m { assert k == v } for i in i64(0) .. 500 { m[i]++ assert m[i] == i + 1 } assert m.len == end keys := m.keys() for i in i64(0) .. end { assert keys[i] == i } for i in i64(0) .. end { m.delete(i) assert m[i] == 0 } assert m.len == 0 } fn test_u64_keys() { mut m := map[u64]u64{} end := u64(1000) for i in u64(0) .. end { m[i] = i assert m[i] == i } for k, v in m { assert k == v } for i in u64(0) .. 500 { m[i]++ assert m[i] == i + 1 } assert m.len == end keys := m.keys() for i in u64(0) .. end { assert keys[i] == i } for i in u64(0) .. end { m.delete(i) assert m[i] == 0 } assert m.len == 0 } fn test_map_set_fixed_array_variable() { mut m := map[string][2]f64{} m['A'] = [1.1, 2.2]! println(m) assert '$m' == "{'A': [1.1, 2.2]}" mut m2 := map[string][2]f64{} arr := [1.1, 2.2]! m2['A'] = arr println(m2) assert '$m2' == "{'A': [1.1, 2.2]}" }