-
Notifications
You must be signed in to change notification settings - Fork 1
/
test_dvector.t
135 lines (118 loc) · 3.41 KB
/
test_dvector.t
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
-- SPDX-FileCopyrightText: 2024 René Hiemstra <rrhiemstar@gmail.com>
-- SPDX-FileCopyrightText: 2024 Torsten Keßler <t.kessler@posteo.de>
--
-- SPDX-License-Identifier: MIT
import "terratest/terratest"
local Alloc = require('alloc')
local Complex = require('complex')
local Nfloat = require("nfloat")
local DVector = require('dvector')
local Allocator = Alloc.Allocator
local DefaultAllocator = Alloc.DefaultAllocator()
local float128 = Nfloat.FixedFloat(128)
local float1024 = Nfloat.FixedFloat(1024)
for _, is_complex in pairs({false, true}) do
for _, S in pairs({int, uint, int64, uint64, float, double, float128, float1024}) do
local T
if is_complex then
T = Complex.complex(S)
else
T = S
end
testenv(T) "DynamicVector" do
local dvector = DVector.DynamicVector(T)
terracode
var alloc: DefaultAllocator
end
testset "new" do
terracode
var v = dvector.new(&alloc, 3)
end
test v:size() == 3
end
testset "all" do
terracode
var v = dvector.all(&alloc, 2, 4)
end
test v:size() == 2
test v:get(0) == 4
test v:get(1) == 4
end
testset "zeros" do
terracode
var v = dvector.zeros(&alloc, 2)
end
test v:size() == 2
test v:get(0) == 0
test v:get(1) == 0
end
testset "zeros_like" do
terracode
var w = dvector.new(&alloc, 3)
var v = dvector.zeros_like(&alloc, &w)
end
test w:size() == 3
test v:size() == 3
for i = 0, 2 do
test v:get(i) == 0
end
end
testset "ones" do
terracode
var v = dvector.ones(&alloc, 2)
end
test v:size() == 2
test v:get(0) == 1
test v:get(1) == 1
end
testset "ones_like" do
terracode
var w = dvector.new(&alloc, 2)
var v = dvector.ones_like(&alloc, &w)
end
test w:size() == 2
test v:size() == 2
test v:get(0) == 1
test v:get(1) == 1
end
testset "from" do
terracode
var v = dvector.from(&alloc, 3, 2, 1)
end
test v:size() == 3
test v:get(0) == 3
test v:get(1) == 2
test v:get(2) == 1
end
testset "copy" do
terracode
var w = dvector.from(&alloc, 3, 2, 1)
var v = dvector.like(&alloc, &w)
v:copy(&w)
end
test v:size() == 3
test v:get(0) == 3
test v:get(1) == 2
test v:get(2) == 1
end
testset "iterator" do
terracode
var x = dvector.from(&alloc, 3, 2, 1)
var y = dvector.from(&alloc, 1, 2, 3, 4)
y.inc = 2
y.size = 4 / y.inc
var sumx: T = 0
for xx in x do
sumx = sumx + xx
end
var prody: T = 1
for yy in y do
prody = prody * yy
end
end
test sumx == 6
test prody == 3
end
end
end
end