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

110 lines
1.3 KiB
V
Raw Normal View History

module maps
fn test_filter() {
m1 := {
0: 'ab'
1: 'bc'
2: 'cd'
3: 'de'
4: 'ef'
5: 'fg'
}
assert filter(m1, fn (k int, v string) bool {
return k % 2 == 0
}) == {
0: 'ab'
2: 'cd'
4: 'ef'
}
assert filter(m1, fn (k int, v string) bool {
return v.contains('b') || v.contains('c')
}) == {
0: 'ab'
1: 'bc'
2: 'cd'
}
}
fn test_to_array() {
m1 := {
`a`: 'bc'
`d`: 'ef'
`g`: 'hi'
}
assert to_array(m1, fn (k rune, v string) string {
return '${k}${v}'
}) == ['abc', 'def', 'ghi']
}
fn test_flat_map() {
m1 := {
1: [2, 3]
4: [5, 6]
7: [8, 9]
}
assert flat_map<int, []int, int>(m1, fn (k int, v []int) []int {
mut a := [k]
a << v
return a
}) == [1, 2, 3, 4, 5, 6, 7, 8, 9]
}
fn test_to_map() {
m1 := {
0: '0'
1: '1'
2: '2'
3: '3'
4: '4'
5: '5'
}
assert to_map<int, string, string, int>(m1, fn (k int, v string) (string, int) {
return v, k
}) == {
'0': 0
'1': 1
'2': 2
'3': 3
'4': 4
'5': 5
}
}
fn test_invert() {
m1 := {
0: '0'
1: '1'
2: '2'
3: '3'
4: '4'
5: '5'
}
assert invert(m1) == {
'0': 0
'1': 1
'2': 2
'3': 3
'4': 4
'5': 5
}
}
fn test_from_array() {
a1 := [
'a',
'b',
'c',
'd',
'e',
'f',
]
assert from_array(a1) == {
0: 'a'
1: 'b'
2: 'c'
3: 'd'
4: 'e'
5: 'f'
}
}