Skip to content

Commit 9ec962f

Browse files
juhyun-kim-uclaHua-Zhou
authored andcommitted
add test functions for gradient! and fisher_B!
1 parent e5468d8 commit 9ec962f

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

test/two_variance_component_test.jl

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ vcmodelrot = TwoVarCompModelRotate(vcmodel)
5252
= zeros(2d^2)
5353
#@code_warntype gradient!(∇, vcmodelrot, vcdatarot)
5454
@inferred VarianceComponentModels.gradient!(∇, vcmodelrot, vcdatarot)
55+
@inferred VarianceComponentModels.gradient!(∇, vcmodel, vcdatarot)
56+
@inferred VarianceComponentModels.gradient!(∇, vcmodel, vcdata)
5557
@test norm(VarianceComponentModels.gradient(vcmodel, vcdata) -
5658
VarianceComponentModels.gradient(vcmodelrot, vcdatarot)) 0.0
5759
@test norm(VarianceComponentModels.gradient(vcmodel, vcdata) -
@@ -75,9 +77,17 @@ H = zeros(2d^2, 2d^2)
7577
@info "Evaluate Fisher information matrix of B"
7678
H = zeros(p * d, p * d)
7779
#@code_warntype fisher_B!(H, vcmodelrot, vcdatarot)
80+
# @inferred VarianceComponentVariate(Y, X, V)
81+
# vcdata = VarianceComponentVariate(Y, X, V)
82+
# vcmodel = VarianceComponentModel(vcdata)
83+
84+
# @info "Pre-compute eigen-decomposition and rotate data"
85+
# vcdatarot = TwoVarCompVariateRotate(vcdata)
86+
# vcmodelrot = TwoVarCompModelRotate(vcmodel)
7887

7988
@inferred fisher_B!(H, vcmodelrot, vcdatarot)
8089
@inferred fisher_B!(H, vcmodel, vcdatarot)
90+
@inferred fisher_B!(H, vcmodel, vcdata)
8191
@test norm(fisher_B(vcmodel, vcdata) - fisher_B(vcmodelrot, vcdatarot)) 0.0
8292
@test norm(fisher_B(vcmodel, vcdata) - fisher_B(vcmodel, vcdatarot)) 0.0
8393
@test norm(fisher_B(vcmodel, [vcdata vcdata]) -

0 commit comments

Comments
 (0)