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
|
$:.push File.dirname(__FILE__) + "/../.."
require "lapack_test"
class GelsyTest < Test::Unit::TestCase
include LapackTest
def setup
@a = Hash.new
@b = Hash.new
@jpvt = Hash.new
@b_exp = Hash.new
@rank_exp = Hash.new
@a[:r] = NMatrix[[-0.09, 0.14, -0.46, 0.68, 1.29],
[-1.56, 0.20, 0.29, 1.09, 0.51],
[-1.48, -0.43, 0.89, -0.71, -0.96],
[-1.09, 0.84, 0.77, 2.11, -1.27],
[ 0.08, 0.55, -1.13, 0.14, 1.74],
[-1.59, -0.72, 1.06, 1.24, 0.34]].to_lm
@b[:r] = NVector[[7.4, 4.2, -8.3, 1.8, 8.6, 2.1]]
@jpvt[:r] = NArray.int(@a[:r].shape[1])
@b_exp[:r] = NArray[[0.6344, 0.9699, -1.4403, 3.3678, 3.3992]]
@rank_exp[:r] = 4
i = Complex::I
@a[:c] = NMatrix[[ 0.47-0.34*I, -0.40+0.54*I, 0.60+0.01*I, 0.80-1.02*I],
[-0.32-0.23*I, -0.05+0.20*I, -0.26-0.44*I, -0.43+0.17*I],
[ 0.35-0.60*I, -0.52-0.34*I, 0.87-0.11*I, -0.34-0.09*I],
[ 0.89+0.71*I, -0.45-0.45*I, -0.02-0.57*I, 1.14-0.78*I],
[-0.19+0.06*I, 0.11-0.85*I, 1.44+0.80*I, 0.07+1.14*I]].to_lm
@b[:c] = NVector[[-1.08-2.59*I, -2.61-1.49*I, 3.13-3.61*I, 7.33-8.01*I, 9.12+7.63*I]]
@jpvt[:c] = NArray.int(@a[:c].shape[1])
@b_exp[:c] = NArray[[1.1669-3.3224*I, 1.3486+5.5027*I, 4.1764+2.3435*I, 0.6467+0.0107*I]]
@rank_exp[:c] = 3
@rcond = 0.01
end
%w(s d c z).each do |x|
method = "#{x}gelsy"
rc = LapackTest.get_rc(x)
define_method("test_#{method}") do
rank, work, info, a, b, jpvt = NumRu::Lapack.send(method, @a[rc], @b[rc], @jpvt[rc], @rcond)
assert_equal 0, info
assert_narray @b_exp[rc], b, 1e-4
assert_equal @rank_exp[rc], rank
end
define_method("test_#{method}_inquiring_lwork") do
rank, work, info, = NumRu::Lapack.send(method, @a[rc], @b[rc], @jpvt[rc], @rcond, :lwork => -1)
assert_equal 0, info
lwork = get_int(work[0])
rank, work, info, a, b, jpvt = NumRu::Lapack.send(method, @a[rc], @b[rc], @jpvt[rc], @rcond, :lwork => lwork)
assert_equal 0, info
assert_equal lwork, get_int(work[0])
assert_narray @b_exp[rc], b, 1e-4
assert_equal @rank_exp[rc], rank
end
define_method("test_#{method}_inquiring_lwork_oldargstyle") do
rank, work, info, = NumRu::Lapack.send(method, @a[rc], @b[rc], @jpvt[rc], @rcond, :lwork => -1)
assert_equal 0, info
lwork = get_int(work[0])
rank, work, info, a, b, jpvt = NumRu::Lapack.send(method, @a[rc], @b[rc], @jpvt[rc], @rcond, -1)
assert_equal 0, info
assert_equal lwork, get_int(work[0])
end
end
end
|
