v/vlib/builtin/array_test.v

1604 lines
29 KiB
V
Raw Normal View History

2019-12-18 01:26:04 +01:00
fn test_pointer() {
2020-04-26 16:25:54 +02:00
mut arr := []&int{}
2019-12-18 01:26:04 +01:00
a := 1
b := 2
c := 3
arr << &a
arr << &b
arr << &c
assert *arr[0] == 1
arr[1] = &c
assert *arr[1] == 3
mut d_arr := [arr] // [][]&int
d_arr << arr
assert *d_arr[0][1] == 3
println(*d_arr[0][1])
assert *d_arr[1][0] == 1
2019-12-18 01:26:04 +01:00
}
2020-04-14 20:32:06 +02:00
fn test_assign() {
mut arr := [2, 4, 8, 16, 32, 64, 128]
2020-04-14 20:32:06 +02:00
arr[0] = 2
arr[1] &= 255
arr[2] |= 255
arr[3] <<= 4
arr[4] >>= 4
arr[5] %= 5
arr[6] ^= 3
assert arr[0] == 2
assert arr[1] == 4 & 255
assert arr[2] == 8 | 255
assert arr[3] == 16 << 4
assert arr[4] == 32 >> 4
assert arr[5] == 64 % 5
assert arr[6] == 128 ^ 3
}
2019-06-25 15:31:56 +02:00
fn test_ints() {
mut a := [1, 5, 2, 3]
assert a.len == 4
assert a[0] == 1
assert a[2] == 2
assert a.last() == 3
a << 4
assert a.len == 5
assert a[4] == 4
assert a.last() == 4
2019-07-25 14:06:26 +02:00
s := a.str()
2019-06-25 15:31:56 +02:00
assert s == '[1, 5, 2, 3, 4]'
assert a[1] == 5
assert a.last() == 4
}
fn test_deleting() {
mut a := [1, 5, 2, 3, 4]
assert a.len == 5
assert a.str() == '[1, 5, 2, 3, 4]'
2019-06-25 15:31:56 +02:00
a.delete(0)
assert a.str() == '[5, 2, 3, 4]'
assert a.len == 4
2019-06-25 15:31:56 +02:00
a.delete(1)
assert a.str() == '[5, 3, 4]'
assert a.len == 3
2020-01-21 18:49:30 +01:00
a.delete(a.len - 1)
assert a.str() == '[5, 3]'
assert a.len == 2
2019-06-25 15:31:56 +02:00
}
fn test_slice_delete() {
mut a := [1.5, 2.5, 3.25, 4.5, 5.75]
b := a[2..4]
a.delete(0)
assert a == [2.5, 3.25, 4.5, 5.75]
assert b == [3.25, 4.5]
a = [3.75, 4.25, -1.5, 2.25, 6.0]
c := a[..3]
a.delete(2)
assert a == [3.75, 4.25, 2.25, 6.0]
assert c == [3.75, 4.25, -1.5]
}
fn test_delete_many() {
mut a := [1, 2, 3, 4, 5, 6, 7, 8, 9]
b := a[2..6]
a.delete_many(4, 3)
assert a == [1, 2, 3, 4, 8, 9]
assert b == [3, 4, 5, 6]
c := a[..a.len]
a.delete_many(2, 0) // this should just clone
a[1] = 17
assert a == [1, 17, 3, 4, 8, 9]
assert c == [1, 2, 3, 4, 8, 9]
a.delete_many(0, a.len)
assert a == []int{}
}
2019-06-25 15:31:56 +02:00
fn test_short() {
a := [1, 2, 3]
assert a.len == 3
assert a.cap == 3
assert a[0] == 1
assert a[1] == 2
assert a[2] == 3
}
fn test_large() {
mut a := [0].repeat(0)
2020-12-20 10:34:13 +01:00
for i in 0 .. 10000 {
2019-06-25 15:31:56 +02:00
a << i
}
assert a.len == 10000
assert a[234] == 234
}
struct Chunk {
val string
}
2020-05-27 18:12:34 +02:00
struct Kkk {
2019-06-25 15:31:56 +02:00
q []Chunk
}
fn test_empty() {
mut chunks := []Chunk{}
2019-06-25 15:31:56 +02:00
a := Chunk{}
assert chunks.len == 0
chunks << a
assert chunks.len == 1
2019-11-14 07:53:05 +01:00
chunks = []
2019-06-25 15:31:56 +02:00
assert chunks.len == 0
chunks << a
assert chunks.len == 1
}
2019-06-25 15:31:56 +02:00
fn test_push() {
mut a := []int{}
2019-06-25 15:31:56 +02:00
a << 1
a << 3
assert a[1] == 3
assert a.str() == '[1, 3]'
}
2019-11-11 15:18:32 +01:00
fn test_insert() {
2020-01-21 18:49:30 +01:00
mut a := [1, 2]
a.insert(0, 3)
2020-06-18 00:22:34 +02:00
assert a[0] == 3
assert a[2] == 2
assert a.len == 3
a.insert(1, 4)
assert a[1] == 4
assert a[2] == 1
assert a.len == 4
a.insert(4, 5)
assert a[4] == 5
assert a[3] == 2
assert a.len == 5
mut b := []f64{}
assert b.len == 0
b.insert(0, f64(1.1))
assert b.len == 1
assert b[0] == f64(1.1)
2020-01-21 18:49:30 +01:00
}
2020-06-18 00:22:34 +02:00
fn test_insert_many() {
mut a := [3, 4]
a.insert(0, [1, 2])
2020-12-20 10:34:13 +01:00
assert a == [1, 2, 3, 4]
b := [5, 6]
a.insert(1, b)
2020-12-20 10:34:13 +01:00
assert a == [1, 5, 6, 2, 3, 4]
}
2020-06-18 00:22:34 +02:00
fn test_prepend() {
mut a := []int{}
assert a.len == 0
a.prepend(1)
assert a.len == 1
assert a[0] == 1
mut b := []f64{}
assert b.len == 0
b.prepend(f64(1.1))
assert b.len == 1
assert b[0] == f64(1.1)
}
fn test_prepend_many() {
2020-12-20 10:34:13 +01:00
mut a := [3, 4]
a.prepend([1, 2])
assert a == [1, 2, 3, 4]
b := [5, 6]
a.prepend(b)
2020-12-20 10:34:13 +01:00
assert a == [5, 6, 1, 2, 3, 4]
}
2019-06-25 15:31:56 +02:00
fn test_strings() {
a := ['a', 'b', 'c']
2020-04-27 14:48:28 +02:00
assert a.str() == "['a', 'b', 'c']"
2019-06-25 15:31:56 +02:00
}
2020-01-21 18:49:30 +01:00
/*
fn test_compare_ints() {
assert compare_ints(1, 2) == -1
assert compare_ints(2, 1) == 1
assert compare_ints(0, 0) == 0
a := 1
b := 2
assert compare_ints(a, b) == -1
assert compare_ints(b, a) == 1
assert compare_ints(a, a) == 0
}
2020-01-21 18:49:30 +01:00
*/
fn test_repeat_int() {
a := [1234].repeat(5)
dump(a)
assert a.len == 5
for x in a {
assert x == 1234
}
2019-06-25 15:31:56 +02:00
}
fn test_repeat_f64() {
a := [1.1].repeat(10)
dump(a)
assert a.len == 10
assert a[0] == 1.1
assert a[5] == 1.1
assert a[9] == 1.1
}
fn test_repeat_f32() {
a := [f32(1.1)].repeat(10)
dump(a)
assert a.len == 10
assert a[0] == f32(1.1)
assert a[5] == f32(1.1)
assert a[9] == f32(1.1)
}
fn test_repeat_i64() {
a := [i64(-123)].repeat(10)
dump(a)
assert a.len == 10
assert a[0] == -123
assert a[5] == -123
assert a[9] == -123
}
fn test_repeat_u64() {
a := [u64(123)].repeat(10)
assert a[0] == 123
assert a[5] == 123
assert a[9] == 123
}
fn test_repeat_several_ints() {
a := [1, 2].repeat(2)
dump(a)
assert a.len == 4
assert a[0] == 1
assert a[1] == 2
assert a[2] == 1
assert a[3] == 2
}
fn test_repeat_several_strings_2() {
a := ['1', 'abc'].repeat(2)
dump(a)
assert a.len == 4
assert a[0] == '1'
assert a[1] == 'abc'
assert a[2] == '1'
assert a[3] == 'abc'
}
fn test_repeat_several_strings_0() {
mut a := ['1', 'abc'].repeat(0)
dump(a)
assert a.len == 0
a << 'abc'
assert a[0] == 'abc'
}
fn test_deep_repeat() {
mut a3 := [[[1, 1], [2, 2], [3, 3]], [[4, 4], [5, 5], [6, 6]]]
r := a3.repeat(3)
dump(r)
a3[1][1][0] = 17
assert r == [
[[1, 1], [2, 2], [3, 3]],
[[4, 4], [5, 5], [6, 6]],
[[1, 1], [2, 2], [3, 3]],
[[4, 4], [5, 5], [6, 6]],
[[1, 1], [2, 2], [3, 3]],
[[4, 4], [5, 5], [6, 6]],
]
assert a3 == [[[1, 1], [2, 2], [3, 3]], [[4, 4], [17, 5], [6, 6]]]
}
2019-06-25 15:31:56 +02:00
fn test_right() {
a := [1, 2, 3, 4]
2019-10-27 00:37:29 +02:00
c := a[1..a.len]
d := a[1..]
2019-10-27 00:37:29 +02:00
assert c[0] == 2
assert c[1] == 3
assert d[0] == 2
assert d[1] == 3
2019-06-25 15:31:56 +02:00
}
fn test_left() {
a := [1, 2, 3]
2019-10-27 00:37:29 +02:00
c := a[0..2]
d := a[..2]
2019-10-27 00:37:29 +02:00
assert c[0] == 1
assert c[1] == 2
assert d[0] == 1
assert d[1] == 2
2019-06-25 15:31:56 +02:00
}
fn test_slice() {
a := [1, 2, 3, 4]
b := a[2..4]
assert b.len == 2
assert a[1..2].len == 1
assert a.len == 4
2019-06-25 15:31:56 +02:00
}
fn test_push_many() {
mut a := [1, 2, 3]
b := [4, 5, 6]
2019-09-14 22:48:30 +02:00
a << b
2019-06-25 15:31:56 +02:00
assert a.len == 6
assert a[0] == 1
assert a[3] == 4
assert a[5] == 6
}
2019-07-17 18:17:07 +02:00
fn test_reverse() {
a := [1, 2, 3, 4]
b := ['test', 'array', 'reverse']
2019-07-17 18:35:40 +02:00
c := a.reverse()
2020-03-18 18:54:08 +01:00
println(c)
2019-07-17 18:35:40 +02:00
d := b.reverse()
2020-01-21 18:49:30 +01:00
for i, _ in c {
assert c[i] == a[a.len - i - 1]
2019-07-17 18:35:40 +02:00
}
for i, _ in d {
2020-01-21 18:49:30 +01:00
assert d[i] == b[b.len - i - 1]
2019-07-17 18:35:40 +02:00
}
e := []int{}
2020-02-29 20:44:02 +01:00
f := e.reverse()
assert f.len == 0
2019-07-17 18:17:07 +02:00
}
const (
2020-05-22 17:36:09 +02:00
c_n = 5
2019-09-14 22:48:30 +02:00
)
2019-12-06 00:11:39 +01:00
struct Foooj {
2020-05-22 17:36:09 +02:00
a [5]int // c_n
2019-12-06 00:11:39 +01:00
}
fn test_fixed() {
mut nums := [4]int{}
2020-12-20 10:34:13 +01:00
// x := nums[1..3]
// assert x.len == 2
2019-09-14 22:48:30 +02:00
assert nums[0] == 0
assert nums[1] == 0
assert nums[2] == 0
assert nums[3] == 0
nums[1] = 7
assert nums[1] == 7
nums2 := [5]int{} // c_n
assert nums2[c_n - 1] == 0
2019-09-14 22:48:30 +02:00
}
2020-06-04 10:35:40 +02:00
fn modify(mut numbers []int) {
2020-01-21 18:49:30 +01:00
numbers[0] = 777
}
2019-09-14 22:48:30 +02:00
fn test_mut_slice() {
2020-01-21 18:49:30 +01:00
mut n := [1, 2, 3]
// modify(mut n)
modify(mut n[..2])
2019-09-14 22:48:30 +02:00
assert n[0] == 777
modify(mut n[2..])
2019-09-14 22:48:30 +02:00
assert n[2] == 777
2019-08-07 12:57:07 +02:00
println(n)
}
2019-08-12 13:41:34 +02:00
2020-06-04 10:35:40 +02:00
fn double_up(mut a []int) {
for i := 0; i < a.len; i++ {
2020-12-20 10:34:13 +01:00
a[i] = a[i] * 2
}
}
2020-06-04 10:35:40 +02:00
fn double_up_v2(mut a []int) {
2020-05-01 00:29:54 +02:00
for i, _ in a {
2020-12-20 10:34:13 +01:00
a[i] = a[i] * 2 // or val*2, doesn't matter
}
}
fn test_mut_arg() {
2020-12-20 10:34:13 +01:00
mut arr := [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
double_up(mut arr)
assert arr.str() == '[2, 4, 6, 8, 10, 12, 14, 16, 18, 20]'
2020-12-20 10:34:13 +01:00
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
double_up_v2(mut arr)
assert arr.str() == '[2, 4, 6, 8, 10, 12, 14, 16, 18, 20]'
}
2019-08-12 13:41:34 +02:00
fn test_clone() {
2019-09-14 22:48:30 +02:00
nums := [1, 2, 3, 4, 100]
_ = nums
2019-09-14 22:48:30 +02:00
nums2 := nums.clone()
assert nums2.len == 5
2020-03-26 11:32:29 +01:00
assert nums.str() == '[1, 2, 3, 4, 100]'
2019-09-14 22:48:30 +02:00
assert nums2.str() == '[1, 2, 3, 4, 100]'
assert nums[1..3].str() == '[2, 3]'
2019-09-14 22:48:30 +02:00
}
/*
2020-12-20 10:34:13 +01:00
fn test_copy() {
a := [1, 2, 3]
2020-12-20 10:42:46 +01:00
b := a
assert b[0] == 1
assert b[1] == 2
assert b[2] == 3
2020-12-20 10:34:13 +01:00
}
*/
fn test_multi_array_clone() {
2020-12-20 10:34:13 +01:00
// 2d array_int
mut a2_1 := [[1, 2, 3], [4, 5, 6]]
mut a2_2 := a2_1.clone()
a2_1[0][1] = 0
a2_2[1][0] = 0
assert a2_1 == [[1, 0, 3], [4, 5, 6]]
assert a2_2 == [[1, 2, 3], [0, 5, 6]]
// 2d array_string
mut b2_1 := [['1', '2', '3'], ['4', '5', '6']]
mut b2_2 := b2_1.clone()
b2_1[0][1] = '0'
b2_2[1][0] = '0'
assert b2_1 == [['1', '0', '3'], ['4', '5', '6']]
assert b2_2 == [['1', '2', '3'], ['0', '5', '6']]
// 3d array_int
mut a3_1 := [[[1, 1], [2, 2], [3, 3]], [[4, 4], [5, 5], [6, 6]]]
2020-12-20 10:34:13 +01:00
mut a3_2 := a3_1.clone()
a3_1[0][0][1] = 0
a3_2[0][1][0] = 0
assert a3_1 == [[[1, 0], [2, 2], [3, 3]], [[4, 4], [5, 5],
[6, 6]]]
assert a3_2 == [[[1, 1], [0, 2], [3, 3]], [[4, 4], [5, 5],
[6, 6]]]
2020-12-20 10:34:13 +01:00
// 3d array_string
mut b3_1 := [[['1', '1'], ['2', '2'], ['3', '3']], [['4', '4'],
['5', '5'], ['6', '6']]]
2020-12-20 10:34:13 +01:00
mut b3_2 := b3_1.clone()
b3_1[0][0][1] = '0'
b3_2[0][1][0] = '0'
assert b3_1 == [[['1', '0'], ['2', '2'], ['3', '3']], [['4', '4'],
['5', '5'], ['6', '6']]]
assert b3_2 == [[['1', '1'], ['0', '2'], ['3', '3']], [['4', '4'],
['5', '5'], ['6', '6']]]
2020-06-19 13:32:55 +02:00
}
fn test_doubling() {
mut nums := [1, 2, 3, 4, 5]
2020-12-20 10:34:13 +01:00
for i in 0 .. nums.len {
nums[i] *= 2
}
println(nums.str())
assert nums.str() == '[2, 4, 6, 8, 10]'
}
struct Test2 {
one int
two int
}
struct Test {
a string
mut:
b []Test2
}
// TODO: default array/struct str methods
2020-05-02 15:26:58 +02:00
fn (ta []Test2) str() string {
mut s := '['
for i, t in ta {
s += t.str()
2020-12-20 10:34:13 +01:00
if i < ta.len - 1 {
s += ', '
}
}
s += ']'
return s
}
2020-05-02 15:26:58 +02:00
fn (t Test2) str() string {
return '{$t.one $t.two}'
}
2020-05-02 15:26:58 +02:00
fn (t Test) str() string {
return '{$t.a $t.b}'
}
fn test_struct_print() {
2020-01-21 18:49:30 +01:00
mut a := Test{
a: 'Test'
2019-11-14 08:00:22 +01:00
b: []
}
2020-01-21 18:49:30 +01:00
b := Test2{
one: 1
two: 2
}
a.b << b
a.b << b
2020-04-23 04:23:19 +02:00
assert a.str() == '{Test [{1 2}, {1 2}]}'
assert b.str() == '{1 2}'
assert a.b.str() == '[{1 2}, {1 2}]'
}
2019-09-28 12:52:02 +02:00
fn test_single_element() {
mut a := [1]
a << 2
assert a.len == 2
assert a[0] == 1
assert a[1] == 2
println(a)
2019-12-06 00:11:39 +01:00
}
2019-10-04 22:07:19 +02:00
fn test_find_index() {
// string
a := ['v', 'is', 'great']
assert a.index('v') == 0
assert a.index('is') == 1
2020-01-21 18:49:30 +01:00
assert a.index('gre') == -1
2019-10-04 22:07:19 +02:00
// int
b := [1, 2, 3, 4]
assert b.index(1) == 0
assert b.index(4) == 3
assert b.index(5) == -1
// byte
c := [0x22, 0x33, 0x55]
assert c.index(0x22) == 0
assert c.index(0x55) == 2
assert c.index(0x99) == -1
// char
d := [`a`, `b`, `c`]
assert d.index(`b`) == 1
assert d.index(`c`) == 2
assert d.index(`u`) == -1
}
2019-10-05 15:03:13 +02:00
fn test_multi() {
2020-01-21 18:49:30 +01:00
a := [[1, 2, 3], [4, 5, 6]]
2019-10-05 15:03:13 +02:00
assert a.len == 2
assert a[0].len == 3
assert a[0][0] == 1
assert a[0][2] == 3
assert a[1][2] == 6
// TODO
2020-01-21 18:49:30 +01:00
// b := [ [[1,2,3],[4,5,6]], [[1,2]] ]
// assert b[0][0][0] == 1
}
fn test_in() {
2020-01-21 18:49:30 +01:00
a := [1, 2, 3]
assert 1 in a
assert 2 in a
assert 3 in a
assert 4 !in a
assert 0 !in a
2020-04-11 23:01:50 +02:00
assert 0 !in a
assert 4 !in a
b := [1, 4, 0]
c := [3, 6, 2, 0]
assert 0 in b
assert 0 in c
}
2019-10-08 12:23:17 +02:00
2019-10-11 03:12:40 +02:00
fn sum(prev int, curr int) int {
return prev + curr
}
fn sub(prev int, curr int) int {
return prev - curr
}
fn test_reduce() {
a := [1, 2, 3, 4, 5]
b := a.reduce(sum, 0)
c := a.reduce(sum, 5)
d := a.reduce(sum, -1)
assert b == 15
assert c == 20
assert d == 14
e := [1, 2, 3]
f := e.reduce(sub, 0)
g := e.reduce(sub, -1)
assert f == -6
assert g == -7
}
fn filter_test_helper_1(a int) bool {
return a > 3
}
2019-10-17 09:44:20 +02:00
fn test_filter() {
a := [1, 2, 3, 4, 5, 6]
b := a.filter(it % 2 == 0)
2019-10-26 13:32:56 +02:00
assert b.len == 3
2019-10-17 09:44:20 +02:00
assert b[0] == 2
assert b[1] == 4
assert b[2] == 6
c := ['v', 'is', 'awesome']
d := c.filter(it.len > 1)
assert d[0] == 'is'
assert d[1] == 'awesome'
2020-04-11 23:01:50 +02:00
////////
2020-12-20 10:34:13 +01:00
arr := [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
2020-04-11 23:01:50 +02:00
println(arr.filter(it % 2 == 0 || it % 3 == 0))
assert true
2020-12-20 10:34:13 +01:00
assert [1, 2, 3].len == 3
mut mut_arr := [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
mut_arr = mut_arr.filter(it < 4)
assert mut_arr.len == 3
2020-12-20 10:34:13 +01:00
assert a.filter(filter_test_helper_1) == [4, 5, 6]
assert [1, 5, 10].filter(filter_test_helper_1) == [5, 10]
2020-04-11 23:01:50 +02:00
// TODO
2020-12-20 10:34:13 +01:00
// assert arr.filter(arr % 2).len == 5
2019-12-06 00:11:39 +01:00
}
2019-10-25 21:40:52 +02:00
2020-06-10 13:18:59 +02:00
fn test_anon_fn_filter() {
filter_num := fn (i int) bool {
return i % 2 == 0
}
2020-12-20 10:34:13 +01:00
assert [1, 2, 3, 4, 5].filter(filter_num) == [2, 4]
2020-06-10 13:18:59 +02:00
}
fn test_anon_fn_arg_filter() {
2020-12-20 10:34:13 +01:00
a := [1, 2, 3, 4].filter(fn (i int) bool {
2020-06-10 13:18:59 +02:00
return i % 2 == 0
})
2020-12-20 10:34:13 +01:00
assert a == [2, 4]
2020-06-10 13:18:59 +02:00
}
2020-05-02 15:26:58 +02:00
fn map_test_helper_1(i int) int {
return i * i
}
2020-12-20 10:34:13 +01:00
2020-05-02 15:26:58 +02:00
fn map_test_helper_2(i int, b string) int {
return i + b.len
}
2020-12-20 10:34:13 +01:00
2020-05-02 15:26:58 +02:00
fn map_test_helper_3(i int, b []string) int {
return i + b.map(it.len)[i % b.len]
}
2019-10-26 13:32:56 +02:00
fn test_map() {
2020-05-01 00:29:54 +02:00
nums := [1, 2, 3, 4, 5, 6]
strs := ['v', 'is', 'awesome']
2020-12-20 10:34:13 +01:00
// assert nums.map() == <error>
// assert nums.map(it, 'excessive') == <error>
2020-05-02 15:26:58 +02:00
// identity
assert nums.map(it) == [1, 2, 3, 4, 5, 6]
assert strs.map(it) == ['v', 'is', 'awesome']
2020-12-20 10:34:13 +01:00
assert nums.map(it - it) == [0, 0, 0, 0, 0, 0]
2020-05-02 15:26:58 +02:00
assert nums.map(it - it)[0] == 0
// type switch
2020-05-01 00:29:54 +02:00
assert nums.map(it * 10) == [10, 20, 30, 40, 50, 60]
2020-05-02 15:26:58 +02:00
assert nums.map(it * it) == [1, 4, 9, 16, 25, 36]
2020-05-01 00:29:54 +02:00
assert nums.map('$it') == ['1', '2', '3', '4', '5', '6']
assert nums.map(it % 2 == 0) == [false, true, false, true, false, true]
assert strs.map(it.to_upper()) == ['V', 'IS', 'AWESOME']
assert strs.map(it == 'awesome') == [false, false, true]
2020-05-02 15:26:58 +02:00
assert strs.map(it.len in nums) == [true, true, false]
assert strs.map(int(7)) == [7, 7, 7]
2020-05-02 15:26:58 +02:00
// external func
assert nums.map(map_test_helper_1(it)) == [1, 4, 9, 16, 25, 36]
assert nums.map(map_test_helper_2(it, 'bb')) == [3, 4, 5, 6, 7, 8]
assert nums.map(map_test_helper_3(it, strs)) == [3, 9, 4, 6, 12, 7]
// empty array as input
2020-12-20 10:34:13 +01:00
assert []int{len: 0}.map(it * 2) == []
2020-05-02 15:26:58 +02:00
// nested maps (where it is of same type)
2020-12-20 10:34:13 +01:00
assert nums.map(strs.map(int(7)) == [7, 7, 7]) == [true, true, true, true, true, true]
assert nums.map('$it' + strs.map('a')[0]) == ['1a', '2a', '3a', '4a', '5a', '6a']
assert nums.map(it + strs.map(int(7))[0]) == [8, 9, 10, 11, 12, 13]
assert nums.map(it + strs.map(it.len)[0]) == [2, 3, 4, 5, 6, 7]
assert strs.map(it.len + strs.map(it.len)[0]) == [2, 3, 8]
2020-05-02 15:26:58 +02:00
// nested (different it types)
2022-04-15 13:58:56 +02:00
assert strs.map(it[nums.map(it - it)[0]]) == [u8(`v`), `i`, `a`]
2020-12-20 10:34:13 +01:00
assert nums[0..3].map('$it' + strs.map(it)[it - 1]) == ['1v', '2is', '3awesome']
assert nums.map(map_test_helper_1) == [1, 4, 9, 16, 25, 36]
assert [1, 5, 10].map(map_test_helper_1) == [1, 25, 100]
2020-05-01 00:29:54 +02:00
assert nums == [1, 2, 3, 4, 5, 6]
assert strs == ['v', 'is', 'awesome']
2019-12-06 00:11:39 +01:00
}
2019-10-26 13:32:56 +02:00
2020-06-09 10:14:10 +02:00
fn test_anon_fn_map() {
add_num := fn (i int) int {
return i + 1
}
2020-12-20 10:34:13 +01:00
assert [1, 2, 3].map(add_num) == [2, 3, 4]
2020-06-09 10:14:10 +02:00
}
fn test_multi_anon_fn_map() {
2020-12-20 10:34:13 +01:00
a := [1, 2, 3].map(fn (i int) int {
return i + 1
})
2020-12-20 10:34:13 +01:00
b := [1, 2, 3].map(fn (i int) int {
return i + 2
})
2020-12-20 10:34:13 +01:00
assert a == [2, 3, 4]
assert b == [3, 4, 5]
}
2020-06-09 12:50:56 +02:00
fn test_anon_fn_arg_map() {
2020-12-20 10:34:13 +01:00
a := [1, 2, 3].map(fn (i int) int {
2020-06-09 12:50:56 +02:00
return i + 1
})
2020-12-20 10:34:13 +01:00
assert a == [2, 3, 4]
2020-06-09 12:50:56 +02:00
}
fn test_anon_fn_arg_different_type_map() {
i_to_str := fn (i int) string {
return i.str()
}
a := [1, 2, 3].map(i_to_str)
assert a == ['1', '2', '3']
}
fn test_anon_fn_inline_different_type_map() {
a := [1, 2, 3].map(fn (i int) string {
return i.str()
})
assert a == ['1', '2', '3']
}
2019-10-25 21:40:52 +02:00
fn test_array_str() {
numbers := [1, 2, 3]
2020-12-20 10:34:13 +01:00
assert numbers == [1, 2, 3]
2019-10-25 21:40:52 +02:00
numbers2 := [numbers, [4, 5, 6]] // dup str() bug
println(numbers2)
2019-10-25 21:40:52 +02:00
assert true
assert numbers.str() == '[1, 2, 3]'
2020-04-23 04:23:19 +02:00
// QTODO
2020-12-20 10:34:13 +01:00
// assert numbers2.str() == '[[1, 2, 3], [4, 5, 6]]'
}
struct User {
2020-12-20 10:34:13 +01:00
age int
name string
}
fn test_eq() {
assert [5, 6, 7] != [6, 7]
assert [`a`, `b`] == [`a`, `b`]
assert [User{
age: 22
name: 'bob'
}] == [User{
age: 22
name: 'bob'
}]
assert [{
'bob': 22
}, {
'tom': 33
}] == [{
'bob': 22
}, {
'tom': 33
}]
assert [[1, 2, 3], [4]] == [[1, 2, 3], [4]]
}
fn test_fixed_array_eq() {
a1 := [1, 2, 3]!
assert a1 == [1, 2, 3]!
assert a1 != [2, 3, 4]!
a2 := [[1, 2]!, [3, 4]!]!
assert a2 == [[1, 2]!, [3, 4]!]!
assert a2 != [[3, 4]!, [1, 2]!]!
a3 := [[1, 2], [3, 4]]!
assert a3 == [[1, 2], [3, 4]]!
assert a3 != [[1, 1], [2, 2]]!
a4 := [[`a`, `b`], [`c`, `d`]]!
assert a4 == [[`a`, `b`], [`c`, `d`]]!
assert a4 != [[`c`, `a`], [`a`, `b`]]!
a5 := [['aaa', 'bbb'], ['ccc', 'ddd']]!
assert a5 == [['aaa', 'bbb'], ['ccc', 'ddd']]!
assert a5 != [['abc', 'def'], ['ccc', 'ddd']]!
a6 := [['aaa', 'bbb']!, ['ccc', 'ddd']!]!
assert a6 == [['aaa', 'bbb']!, ['ccc', 'ddd']!]!
assert a6 != [['aaa', 'bbb']!, ['aaa', 'ddd']!]!
a7 := [[1, 2]!, [3, 4]!]
assert a7 == [[1, 2]!, [3, 4]!]
assert a7 != [[2, 3]!, [1, 2]!]
a8 := [['aaa', 'bbb']!, ['ccc', 'ddd']!]
assert a8 == [['aaa', 'bbb']!, ['ccc', 'ddd']!]
assert a8 != [['bbb', 'aaa']!, ['cccc', 'dddd']!]
}
fn test_fixed_array_literal_eq() {
assert [1, 2, 3]! == [1, 2, 3]!
assert [1, 1, 1]! != [1, 2, 3]!
assert [[1, 2], [3, 4]]! == [[1, 2], [3, 4]]!
assert [[1, 1], [2, 2]]! != [[1, 2], [3, 4]]!
assert [[1, 1]!, [2, 2]!]! == [[1, 1]!, [2, 2]!]!
assert [[1, 1]!, [2, 2]!]! != [[1, 2]!, [2, 3]!]!
assert [[1, 1]!, [2, 2]!] == [[1, 1]!, [2, 2]!]
assert [[1, 1]!, [2, 2]!] != [[1, 2]!, [2, 3]!]
}
2019-11-18 22:27:27 +01:00
fn test_sort() {
mut a := ['hi', '1', '5', '3']
a.sort()
assert a == ['1', '3', '5', 'hi']
2019-11-18 22:27:27 +01:00
mut nums := [67, -3, 108, 42, 7]
nums.sort()
assert nums == [-3, 7, 42, 67, 108]
2020-08-12 05:54:51 +02:00
nums.sort(a < b)
assert nums == [-3, 7, 42, 67, 108]
nums.sort(b < a)
assert nums == [108, 67, 42, 7, -3]
mut users := [User{22, 'Peter'}, User{20, 'Bob'}, User{25, 'Alice'}]
users.sort(a.age < b.age)
assert users[0].age == 20
assert users[1].age == 22
assert users[2].age == 25
assert users[0].name == 'Bob'
assert users[1].name == 'Peter'
assert users[2].name == 'Alice'
users.sort(a.age > b.age)
assert users[0].age == 25
assert users[1].age == 22
assert users[2].age == 20
users.sort(b.age > a.age)
assert users[0].age == 20
assert users[1].age == 22
assert users[2].age == 25
users.sort(a.name < b.name)
assert users[0].name == 'Alice'
assert users[1].name == 'Bob'
assert users[2].name == 'Peter'
2019-12-06 00:11:39 +01:00
}
2019-11-18 22:27:27 +01:00
fn test_sort_with_compare() {
mut a := ['hi', '1', '5', '3']
a.sort_with_compare(fn (a &string, b &string) int {
if a < b {
return -1
}
if a > b {
return 1
}
return 0
})
assert a == ['1', '3', '5', 'hi']
}
2021-04-02 16:28:27 +02:00
fn test_rune_sort() {
mut bs := [`f`, `e`, `d`, `b`, `c`, `a`]
bs.sort()
println(bs)
assert bs == [`a`, `b`, `c`, `d`, `e`, `f`]
2021-04-02 16:28:27 +02:00
bs.sort(a > b)
println(bs)
assert bs == [`f`, `e`, `d`, `c`, `b`, `a`]
2021-04-02 16:28:27 +02:00
bs.sort(a < b)
println(bs)
assert bs == [`a`, `b`, `c`, `d`, `e`, `f`]
2021-04-02 16:28:27 +02:00
}
fn test_sort_by_different_order_of_a_b() {
mut x := [1, 2, 3]
x.sort(a < b)
println(x)
assert x == [1, 2, 3]
mut y := [1, 2, 3]
y.sort(b < a)
println(y)
assert y == [3, 2, 1]
}
2020-01-21 18:49:30 +01:00
fn test_f32_sort() {
mut f := [f32(50.0), 15, 1, 79, 38, 0, 27]
2021-04-03 12:16:18 +02:00
f.sort()
assert f == [f32(0.0), 1, 15, 27, 38, 50, 79]
f.sort(a < b)
assert f == [f32(0.0), 1, 15, 27, 38, 50, 79]
f.sort(b > a)
assert f == [f32(0.0), 1, 15, 27, 38, 50, 79]
f.sort(b < a)
assert f == [f32(79.0), 50, 38, 27, 15, 1, 0]
f.sort(a > b)
assert f == [f32(79.0), 50, 38, 27, 15, 1, 0]
2020-01-21 18:49:30 +01:00
}
2019-11-18 22:27:27 +01:00
2020-01-21 18:49:30 +01:00
fn test_f64_sort() {
mut f := [50.0, 15, 1, 79, 38, 0, 27]
2021-04-03 12:16:18 +02:00
f.sort()
2020-01-21 18:49:30 +01:00
assert f[0] == 0.0
assert f[1] == 1.0
assert f[6] == 79.0
}
fn test_i64_sort() {
mut f := [i64(50), 15, 1, 79, 38, 0, 27]
2021-04-03 12:16:18 +02:00
f.sort()
2020-01-21 18:49:30 +01:00
assert f[0] == 0
assert f[1] == 1
assert f[6] == 79
}
2019-11-18 22:27:27 +01:00
fn test_sort_index_expr() {
mut f := [[i64(50), 48], [i64(15)], [i64(1)], [i64(79)], [i64(38)],
[i64(0)], [i64(27)]]
// TODO This currently gives "indexing pointer" error without unsafe
unsafe {
f.sort(a[0] < b[0])
}
assert f == [[i64(0)], [i64(1)], [i64(15)], [i64(27)], [i64(38)],
[i64(50), 48], [i64(79)]]
}
2021-08-04 05:31:57 +02:00
fn test_a_b_paras_sort() {
mut arr_i := [1, 3, 2]
arr_i.sort(a < b)
println(arr_i)
assert arr_i == [1, 2, 3]
arr_i.sort(b < a)
println(arr_i)
assert arr_i == [3, 2, 1]
mut arr_f := [1.1, 3.3, 2.2]
arr_f.sort(a < b)
println(arr_f)
assert arr_f == [1.1, 2.2, 3.3]
arr_f.sort(b < a)
println(arr_f)
assert arr_f == [3.3, 2.2, 1.1]
}
2019-11-06 16:19:03 +01:00
/*
fn test_for_last() {
numbers := [1, 2, 3, 4]
mut s := '['
for num in numbers {
s += '$num'
if !last {
s += ', '
2019-12-06 00:11:39 +01:00
}
}
s += ']'
assert s == '[1, 2, 3, 4]'
2019-12-06 00:11:39 +01:00
}
2019-11-06 16:19:03 +01:00
*/
2019-12-27 04:09:09 +01:00
struct Foo {
2020-01-21 18:49:30 +01:00
mut:
2019-12-27 04:09:09 +01:00
bar []int
}
fn test_in_struct() {
mut baz := Foo{
bar: [0, 0, 0]
}
baz.bar[0] += 2
baz.bar[0]++
assert baz.bar[0] == 3
}
2020-12-07 17:13:19 +01:00
[direct_array_access]
fn test_direct_modification() {
mut foo := [2, 0, 5]
foo[1] = 3
foo[0] *= 7
foo[1]--
foo[2] -= 2
assert foo[0] == 14
assert foo[1] == 2
assert foo[2] == 3
}
fn test_bools() {
println('test b')
mut a := [true, false]
a << true
println(a)
}
2020-01-19 22:15:37 +01:00
fn test_push_many_self() {
2020-12-20 10:34:13 +01:00
mut actual_arr := [1, 2, 3, 4]
actual_arr << actual_arr
expected_arr := [1, 2, 3, 4, 1, 2, 3, 4]
assert actual_arr.len == expected_arr.len
for i in 0 .. actual_arr.len {
assert actual_arr[i] == expected_arr[i]
}
2020-01-19 22:15:37 +01:00
}
fn test_for() {
2020-12-20 10:34:13 +01:00
nums := [1, 2, 3]
mut sum := 0
2020-03-25 14:24:48 +01:00
for num in nums {
sum += num
}
assert sum == 6
}
2020-01-27 22:31:48 +01:00
fn test_clear() {
2020-12-20 10:34:13 +01:00
mut arr := [1, 2, 3]
2020-01-27 22:31:48 +01:00
assert arr.len == 3
arr.clear()
assert arr.len == 0
arr << 3
arr << 2
arr << 1
arr << 0
assert arr.len == 4
assert arr[0] == 3
assert arr[1] == 2
assert arr[2] == 1
assert arr[3] == 0
arr.clear()
assert arr.len == 0
}
2020-02-17 20:31:40 +01:00
fn test_trim() {
2020-12-20 10:34:13 +01:00
mut arr := [1, 2, 3, 4, 5, 6, 7, 8, 9]
2020-02-17 20:31:40 +01:00
assert arr.len == 9
arr.trim(9)
assert arr.len == 9
assert arr.last() == 9
arr.trim(7)
assert arr.len == 7
assert arr.last() == 7
arr.trim(2)
assert arr.len == 2
assert arr.last() == 2
}
[manualfree]
fn test_drop() {
mut a := [1, 2]
a << 3 // pushing assures reallocation; a.cap now should be bigger:
assert a.cap > 3
// eprintln('>>> a.cap: $a.cap | a.len: $a.len')
a.drop(-1000)
assert a == [1, 2, 3] // a.drop( negative ) should NOT modify the array
// eprintln('>>> a.cap: $a.cap | a.len: $a.len')
a.drop(2)
assert a == [3]
assert a.cap > a.len
// eprintln('>>> a.cap: $a.cap | a.len: $a.len')
a.drop(10)
assert a == []
assert a.cap > a.len
// eprintln('>>> a.cap: $a.cap | a.len: $a.len')
a << 123
a << 456
a << 789
// eprintln('>>> a.cap: $a.cap | a.len: $a.len')
assert a == [123, 456, 789]
a.drop(10)
assert a == []
// eprintln('>>> a.cap: $a.cap | a.len: $a.len')
unsafe { a.free() } // test offset OK
}
2020-12-20 10:34:13 +01:00
fn test_hex() {
// array hex
2022-04-15 13:58:56 +02:00
st := [u8(`V`), `L`, `A`, `N`, `G`]
2020-12-20 10:34:13 +01:00
assert st.hex() == '564c414e47'
assert st.hex().len == 10
2022-04-15 13:58:56 +02:00
st1 := [u8(0x41)].repeat(100)
2020-12-20 10:34:13 +01:00
assert st1.hex() == '41'.repeat(100)
}
fn test_left_shift_precendence() {
mut arr := []int{}
arr << 1 + 1
arr << 1 - 1
arr << 2 / 1
arr << 2 * 1
assert arr[0] == 2
assert arr[1] == 0
assert arr[2] == 2
assert arr[3] == 2
}
fn test_array_with_cap() {
2020-12-20 10:34:13 +01:00
a4 := []int{len: 1, cap: 10}
2020-04-26 17:28:43 +02:00
assert a4.len == 1
assert a4.cap == 10
2020-12-20 10:34:13 +01:00
a5 := []int{len: 1, cap: 10}
assert a5.len == 1
assert a5.cap == 10
}
2020-06-12 01:24:25 +02:00
fn test_multi_array_index() {
2020-12-20 10:34:13 +01:00
mut a := [][]int{len: 2, init: []int{len: 3, init: 0}}
2020-06-12 01:24:25 +02:00
a[0][0] = 1
assert '$a' == '[[1, 0, 0], [0, 0, 0]]'
mut b := [[0].repeat(3)].repeat(2)
b[0][0] = 1
assert '$b' == '[[1, 0, 0], [0, 0, 0]]'
2020-06-12 01:24:25 +02:00
}
2020-06-28 11:16:16 +02:00
fn test_plus_assign_string() {
mut a := ['']
a[0] += 'abc'
assert a == ['abc']
}
fn mut_arr_with_eq_in_fn(mut a []int) {
2020-12-20 10:34:13 +01:00
if a == [1, 2, 3, 4] {
a[0] = 0
}
2020-12-20 10:34:13 +01:00
if [0, 2, 3, 4] == a {
a[1] = 0
}
2020-12-20 10:34:13 +01:00
if !(a != [0, 0, 3, 4]) {
a[2] = 0
}
2020-12-20 10:34:13 +01:00
if !([0, 0, 0, 4] != a) {
a[3] = 0
}
}
fn test_mut_arr_with_eq_in_fn() {
2020-12-20 10:34:13 +01:00
mut a := [1, 2, 3, 4]
mut_arr_with_eq_in_fn(mut a)
2020-12-20 10:34:13 +01:00
assert a == [0, 0, 0, 0]
}
fn array_in_mut(mut a []int) {
if 1 in a {
a[0] = 2
}
}
fn test_array_in_mut() {
2020-12-20 10:34:13 +01:00
mut a := [1, 2]
array_in_mut(mut a)
2020-12-20 10:34:13 +01:00
assert a == [2, 2]
}
// test array delete in function with mut argument
fn delete_nums(mut arr []int) {
arr.delete(0)
}
fn test_array_delete_in_mut() {
mut nums := [1, 2, 3]
delete_nums(mut nums)
assert nums == [2, 3]
}
// test array add in function with mut argument
fn add_nums(mut arr []int) {
arr << 4
}
fn test_array_add_in_mut() {
mut nums := [1, 2, 3]
add_nums(mut nums)
assert nums == [1, 2, 3, 4]
}
fn test_reverse_in_place() {
mut a := [1, 2, 3, 4]
a.reverse_in_place()
assert a == [4, 3, 2, 1]
mut b := ['a', 'b', 'c']
b.reverse_in_place()
assert b == ['c', 'b', 'a']
mut c := [[1, 2], [3, 4], [5, 6]]
c.reverse_in_place()
assert c == [[5, 6], [3, 4], [1, 2]]
}
2020-07-14 18:55:44 +02:00
fn test_array_int_pop() {
2020-12-20 10:34:13 +01:00
mut a := [1, 2, 3, 4, 5]
2020-07-14 18:55:44 +02:00
assert a.len == 5
x := a.last()
y := a.pop()
assert x == y
assert a.len == 4
z := a.pop()
assert a.len == 3
assert z == 4
2021-03-19 08:34:41 +01:00
x1 := a.pop()
x2 := a.pop()
2020-07-14 18:55:44 +02:00
final := a.pop()
assert final == 1
}
fn test_array_string_pop() {
mut a := ['abc', 'def', 'xyz']
assert a.len == 3
assert a.pop() == 'xyz'
assert a.pop() == 'def'
assert a.pop() == 'abc'
assert a.len == 0
assert a.cap == 3
}
2020-08-24 09:04:50 +02:00
2020-12-02 04:35:00 +01:00
fn test_array_first() {
a := [3]
assert a.first() == 3
b := [1, 2, 3, 4]
assert b.first() == 1
c := ['abc', 'def']
assert c.first()[0] == `a`
s := [Chunk{'a'}]
assert s.first().val == 'a'
}
fn test_array_last() {
a := [3]
assert a.last() == 3
b := [1, 2, 3, 4]
assert b.last() == 4
c := ['abc', 'def']
assert c.last()[0] == `d`
s := [Chunk{'a'}]
assert s.last().val == 'a'
}
2020-08-24 09:04:50 +02:00
[direct_array_access]
fn test_direct_array_access() {
2020-12-20 10:34:13 +01:00
mut a := [11, 22, 33, 44]
2020-08-24 09:04:50 +02:00
assert a[0] == 11
assert a[2] == 33
x := a[0]
a[0] = 21
a[1] += 2
a[2] = x + 3
a[3] -= a[1]
assert a == [21, 24, 14, 20]
}
[direct_array_access]
fn test_direct_array_access_via_ptr() {
2020-12-20 10:34:13 +01:00
mut b := [11, 22, 33, 44]
2020-08-24 09:04:50 +02:00
unsafe {
mut a := &b
assert a[0] == 11
assert a[2] == 33
x := a[0]
a[0] = 21
a[1] += 2
a[2] = x + 3
a[3] -= a[1]
assert a == [21, 24, 14, 20]
}
}
fn test_push_arr_string_free() {
mut lines := ['hi']
s := 'a' + 'b'
lines << s
// make sure the data in the array is valid after freeing the string
unsafe { s.free() }
//
println(lines)
assert lines.len == 2
assert lines[0] == 'hi'
assert lines[1] == 'ab'
}
const (
grid_size_1 = 2
grid_size_2 = 3
grid_size_3 = 4
cell_value = 123
)
fn test_multidimensional_array_initialization_with_consts() {
mut data := [][][]int{len: grid_size_1, init: [][]int{len: grid_size_2, init: []int{len: grid_size_3, init: cell_value}}}
assert data.len == grid_size_1
assert data[0].len == grid_size_2
assert data[0][0].len == grid_size_3
assert data[0][0][0] == cell_value
assert data[1][1][1] == cell_value
}
fn test_byteptr_vbytes() {
unsafe {
bp := malloc(5)
bp[0] = 1
bp[1] = 2
bp[2] = 3
bp[3] = 4
bp[4] = 255
bytes := bp.vbytes(5)
println(bytes)
assert bytes.len == 5
assert bytes[0] == 1
assert bytes[1] == 2
assert bytes[2] == 3
assert bytes[3] == 4
assert bytes[4] == 255
}
}
fn test_voidptr_vbytes() {
unsafe {
bp := malloc(3)
bp[0] = 4
bp[1] = 5
bp[2] = 6
bytes := voidptr(bp).vbytes(3)
assert bytes.len == 3
assert bytes[0] == 4
assert bytes[1] == 5
assert bytes[2] == 6
println(bytes)
}
}
fn test_multi_array_prepend() {
mut a := [][]int{}
a.prepend([1, 2, 3])
assert a == [[1, 2, 3]]
mut b := [][]int{}
b.prepend([[1, 2, 3]])
assert b == [[1, 2, 3]]
}
fn test_multi_array_insert() {
mut a := [][]int{}
a.insert(0, [1, 2, 3])
assert a == [[1, 2, 3]]
mut b := [][]int{}
b.insert(0, [[1, 2, 3]])
assert b == [[1, 2, 3]]
}
fn test_multi_array_in() {
a := [[1]]
println([1] in a)
assert [1] in a
}
fn test_any_type_array_contains() {
a := [true, false]
assert a.contains(true)
assert true in a
assert a.contains(false)
assert false in a
b := [i64(2), 3, 4]
assert b.contains(i64(3))
assert 5 !in b
c := [[1], [2]]
assert c.contains([1])
assert [2] in c
assert [3] !in c
}
struct Person {
name string
nums []int
kv map[string]string
}
fn test_struct_array_of_multi_type_in() {
ivan := Person{
name: 'ivan'
nums: [1, 2, 3]
kv: {
'aaa': '111'
}
}
people := [
Person{
name: 'ivan'
nums: [1, 2, 3]
kv: {
'aaa': '111'
}
},
Person{
name: 'bob'
nums: [2]
kv: {
'bbb': '222'
}
},
]
println(ivan in people)
assert ivan in people
}
fn test_struct_array_of_multi_type_index() {
ivan := Person{
name: 'ivan'
nums: [1, 2, 3]
kv: {
'aaa': '111'
}
}
people := [
Person{
name: 'ivan'
nums: [1, 2, 3]
kv: {
'aaa': '111'
}
},
Person{
name: 'bob'
nums: [2]
kv: {
'bbb': '222'
}
},
]
println(people.index(ivan))
assert people.index(ivan) == 0
}
struct Coord {
x int
y int
z int
}
fn test_array_struct_contains() {
mut coords := []Coord{}
coord_1 := Coord{
x: 1
y: 2
z: -1
}
coords << coord_1
exists := coord_1 in coords
not_exists := coord_1 !in coords
println('`exists`: $exists and `not exists`: $not_exists')
assert exists == true
assert not_exists == false
}
fn test_array_struct_ref_contains() {
mut coords := []&Coord{}
coord_1 := &Coord{
x: 1
y: 2
z: -1
}
coords << coord_1
exists := coord_1 in coords
println(exists)
assert exists == true
}
fn test_array_struct_ref_index() {
mut coords := []&Coord{}
coord_1 := &Coord{
x: 1
y: 2
z: -1
}
coords << coord_1
println(coords.index(coord_1))
assert coords.index(coord_1) == 0
}
fn test_array_of_array_append() {
mut x := [][]int{len: 4}
println(x) // OK
x[2] << 123 // RTE
println(x)
assert '$x' == '[[], [], [123], []]'
}
fn test_array_of_map_insert() {
mut x := []map[string]int{len: 4}
println(x) // OK
x[2]['123'] = 123 // RTE
println(x)
assert '$x' == "[{}, {}, {'123': 123}, {}]"
}
fn test_multi_fixed_array_init() {
a := [3][3]int{}
assert '$a' == '[[0, 0, 0], [0, 0, 0], [0, 0, 0]]'
}
struct Numbers {
odds []int
evens []int
}
fn test_array_of_multi_filter() {
arr := [1, 2, 3, 4, 5]
nums := Numbers{
odds: arr.filter(it % 2 == 1)
evens: arr.filter(it % 2 == 0)
}
println(nums)
assert nums.odds == [1, 3, 5]
assert nums.evens == [2, 4]
}
fn test_array_of_multi_map() {
arr := [1, 3, 5]
nums := Numbers{
odds: arr.map(it + 2)
evens: arr.map(it * 2)
}
println(nums)
assert nums.odds == [3, 5, 7]
assert nums.evens == [2, 6, 10]
}
fn test_multi_fixed_array_with_default_init() {
a := [3][3]int{init: [3]int{init: 10}}
println(a)
assert a == [[10, 10, 10]!, [10, 10, 10]!, [10, 10, 10]!]!
}
struct Abc {
mut:
x i64
y i64
z i64
}
fn test_clone_of_same_elem_size_array() {
mut arr := []Abc{}
arr << Abc{1, 2, 3}
arr << Abc{2, 3, 4}
arr2 := arr.clone()
println(arr2)
assert arr2 == [Abc{1, 2, 3}, Abc{2, 3, 4}]
}
pub fn example<T>(mut arr []T) []T {
return arr.clone()
}
fn test_generic_mutable_arrays() {
mut arr := [1, 2, 3]
assert example(mut arr) == [1, 2, 3]
}
struct Ok {}
fn test_inline_array_element_access() {
println([Ok{}][0])
a1 := [Ok{}][0]
assert a1 == Ok{}
println([1][0])
a2 := [1][0]
assert a2 == 1
}
//
fn f(x int, y int) []int {
return [x, y]
}
fn test_2d_array_init_with_it() {
a := [][]int{len: 6, init: f(it, 2 * it)}
assert a == [[0, 0], [1, 2], [2, 4], [3, 6], [4, 8], [5, 10]]
}