1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/builtin/map_test.v

1038 lines
16 KiB
V

import rand
const strings = unique_strings(7000, 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_values_many() {
mut m := map[string]int{}
for i, s in strings {
m[s] = i
}
values := m.values()
assert values.len == strings.len
assert values.len == m.len
}
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
assert m.values().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'
// Test `.values()`
values := m.values()
assert values.len == 1
assert 80 !in values
assert 101 in values
assert m.values().len == 1
assert m.values()[0] == 101
// //
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]u8{}
m10['test'] = u8(0)
assert m10['test'] == u8(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'] = unsafe { nil }
assert m14['test'] == unsafe { nil }
mut m15 := map[string]&u8{}
m15['test'] = &u8(0)
assert m15['test'] == &u8(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_clear() {
mut m := map[string]int{}
m['one'] = 1
m['two'] = 2
m.clear()
assert m.len == 0
}
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 {
assert true
} 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.values() == [9, 16, 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_voidptr_values() {
mut m := map[string]voidptr{}
v := 5
m['var'] = &v
m['map'] = &m
assert m['var'] == &v
assert m['map'] == &m
assert m.values().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[u8]u8{}
byte_max := u8(255)
for i in u8(0) .. byte_max {
m[i] = i
assert m[i] == i
}
for k, v in m {
assert k == v
}
for i in u8(0) .. 100 {
m[i]++
assert m[i] == i + 1
}
assert m.len == byte_max
keys := m.keys()
for i in u8(0) .. byte_max {
assert keys[i] == i
}
for i in u8(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 u64(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]}"
}
type Map = map[int]int
fn test_alias_of_map_delete() {
mut m := Map(map[int]int{})
m[11] = 111
m[22] = 222
println(m)
m.delete(11)
println(m)
assert m.len == 1
assert m[22] == 222
}
struct State {
mut:
state map[string]rune
}
struct Foo {
bar int
baz int
}
fn (mut st State) add(s string, r rune) {
st.state[s] = r
}
fn (mut st State) add2(foos []Foo) {
for f in foos {
st.state['${f.bar},${f.baz}'] = `z`
}
}
fn (mut st State) reset() {
st.state.clear()
}
fn test_map_clear() {
mut s := State{}
item1 := [Foo{
bar: 1
baz: 2
}, Foo{
bar: 3
baz: 4
}]
s.add2(item1)
assert s.state.len == 2
s.reset()
s.add2(item1)
assert s.state.len == 2
}