Skip to content

Commit fb0626f

Browse files
committed
add test for updating one of A or b
1 parent d05ad09 commit fb0626f

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

test/forwarddiff_overloads.jl

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,4 +51,32 @@ cache.b = new_b
5151
x_p = solve!(cache)
5252
backslash_x_p = new_A \ new_b
5353

54+
@test (x_p, backslash_x_p, rtol = 1e-9)
55+
56+
# Just update A
57+
A, b = h([ForwardDiff.Dual(10.0, 1.0, 0.0), ForwardDiff.Dual(10.0, 0.0, 1.0)])
58+
59+
prob = LinearProblem(A, b)
60+
cache = init(prob)
61+
62+
new_A, _ = h([ForwardDiff.Dual(5.0, 1.0, 0.0), ForwardDiff.Dual(5.0, 0.0, 1.0)])
63+
cache.A = new_A
64+
65+
x_p = solve!(cache)
66+
backslash_x_p = new_A \ b
67+
68+
@test (x_p, backslash_x_p, rtol = 1e-9)
69+
70+
# Just update b
71+
A, b = h([ForwardDiff.Dual(5.0, 1.0, 0.0), ForwardDiff.Dual(5.0, 0.0, 1.0)])
72+
73+
prob = LinearProblem(A, b)
74+
cache = init(prob)
75+
76+
_, new_b = h([ForwardDiff.Dual(5.0, 1.0, 0.0), ForwardDiff.Dual(5.0, 0.0, 1.0)])
77+
cache.b = new_b
78+
79+
x_p = solve!(cache)
80+
backslash_x_p = A \ new_b
81+
5482
@test (x_p, backslash_x_p, rtol = 1e-9)

0 commit comments

Comments
 (0)