R Under development (unstable) (2024-08-21 r87038 ucrt) -- "Unsuffered Consequences"
Copyright (C) 2024 The R Foundation for Statistical Computing
Platform: x86_64-w64-mingw32/x64

R is free software and comes with ABSOLUTELY NO WARRANTY.
You are welcome to redistribute it under certain conditions.
Type 'license()' or 'licence()' for distribution details.

R is a collaborative project with many contributors.
Type 'contributors()' for more information and
'citation()' on how to cite R or R packages in publications.

Type 'demo()' for some demos, 'help()' for on-line help, or
'help.start()' for an HTML browser interface to help.
Type 'q()' to quit R.

> library(testthat)
>  Sys.setenv('OMP_THREAD_LIMIT'=2)
>  library(rlibkriging)

Attaching package: 'rlibkriging'

The following objects are masked from 'package:base':

    load, save

> 
> ##library(rlibkriging, lib.loc="bindings/R/Rlibs")
> ##library(testthat)
> 
> f <- function(x) {
+     1 - 1 / 2 * (sin(12 * x) / (1 + x) + 2 * cos(7 * x) * x^5 + 0.7)
+ }
> plot(f)
> n <- 5
> X_o <- seq(from = 0, to = 1, length.out = n)
> y_o <- f(X_o)
> points(X_o, y_o,pch=16)
> 
> lk <- Kriging(y = matrix(y_o, ncol = 1),
+               X = matrix(X_o, ncol = 1),
+               kernel = "gauss",
+               regmodel = "linear",
+               optim = "none",
+               #normalize = TRUE,
+               parameters = list(theta = matrix(0.1), sigma2=0.001))
> 
> ## Ckeck consistency bw predict & simulate
> 
> lp = lk$predict(seq(0,1,,21)) # libK predict
> lines(seq(0,1,,21),lp$mean,col='red')
> polygon(c(seq(0,1,,21),rev(seq(0,1,,21))),c(lp$mean+2*lp$stdev,rev(lp$mean-2*lp$stdev)),col=rgb(1,0,0,0.2),border=NA)
> 
> ls = lk$simulate(1000, 123, seq(0,1,,21)) # libK simulate
> for (i in 1:min(100,ncol(ls))) {
+     lines(seq(0,1,,21),ls[,i],col=rgb(1,0,0,.1),lwd=4)
+ }
> 
> for (i in 1:21) {
+     if (lp$stdev[i,] > 1e-3) # otherwise means that density is ~ dirac, so don't test
+     test_that(desc="simulate sample follows predictive distribution",
+         expect_true(ks.test(ls[i,], "pnorm", mean = lp$mean[i,],sd = lp$stdev[i,])$p.value > 0.01))
+ }
Test passed 🥳
Test passed 🥳
Test passed 🥇
Test passed 🌈
Test passed 🎉
Test passed 🎉
Test passed 🥳
Test passed 😀
Test passed 😀
Test passed 🥳
Test passed 😸
Test passed 🎊
Test passed 😸
Test passed 🥇
Test passed 🥇
Test passed 😀
> 
> ## Check consistency when update
> 
> X_u = c(.4,.6)
> y_u = f(X_u)
> 
> # new Kriging model from scratch
> l2 = Kriging(y = matrix(c(y_o,y_u),ncol=1),
+              X = matrix(c(X_o,X_u),ncol=1),
+               kernel = "gauss",
+               regmodel = "linear",
+               optim = "none",
+               parameters = list(theta = matrix(0.1), sigma2=0.001))
> 
> lu = copy(lk)
> lu$update(y_u,X_u)
> 
> ## Update, predict & simulate
> 
> lp2 = l2$predict(seq(0,1,,21))
> lpu = lu$predict(seq(0,1,,21))
> 
> plot(f)
> points(X_o,y_o)
> lines(seq(0,1,,21),lp2$mean,col='red')
> polygon(c(seq(0,1,,21),rev(seq(0,1,,21))),c(lp2$mean+2*lp2$stdev,rev(lp2$mean-2*lp2$stdev)),col=rgb(1,0,0,0.2),border=NA)
> lines(seq(0,1,,21),lpu$mean,col='blue')
> polygon(c(seq(0,1,,21),rev(seq(0,1,,21))),c(lpu$mean+2*lpu$stdev,rev(lpu$mean-2*lpu$stdev)),col=rgb(0,0,1,0.2),border=NA)
> 
> ls2 = l2$simulate(1000, 123, seq(0,1,,21))
> lsu = lu$simulate(1000, 123, seq(0,1,,21))
> for (i in 1:100) {
+     lines(seq(0,1,,21),ls2[,i],col=rgb(1,0,0,.1),lwd=4)
+     lines(seq(0,1,,21),lsu[,i],col=rgb(0,0,1,.1),lwd=4)
+ }
> 
> for (i in 1:21) {
+     #test_that(desc="simulate sample follows predictive distribution",
+     #    expect_true(ks.test(ls2[i,],lsu[i,])$p.value > 0.01))
+ 
+     # random gen is the same so we expect strict equality of samples !
+     test_that(desc="simulate sample are the same",
+         expect_true(all.equal(ls2[i,],lsu[i,])))
+ }
Test passed 🎉
Test passed 🥳
Test passed 🌈
Test passed 🥇
Test passed 🥇
Test passed 🥇
Test passed 🥇
Test passed 😸
Test passed 🎉
Test passed 🥇
Test passed 🌈
Test passed 😀
Test passed 😀
Test passed 😀
Test passed 🥳
Test passed 😸
Test passed 😀
Test passed 😀
Test passed 🎉
Test passed 🎉
Test passed 🥇
> 
> ## Update simulate
> 
> X_n = seq(0,1,,21)
> i_u = c(9,13)
> X_u = X_n[i_u]# c(.4,.6)
> y_u = f(X_u)
> 
> X_n = sort(c(X_u+1e-2,X_n)) # add some noise to avoid degenerate cases
> 
> ls = lk$simulate(1000, 123, X_n, will_update=TRUE)
> #y_u = rs[i_u,1] # force matching 1st sim
> lus = NULL
> lus = lk$update_simulate(y_u, X_u)
> 
> lu = copy(lk)
> lu$update(y_u, matrix(X_u,ncol=1), refit=FALSE)
> lsu = NULL
> lsu = lu$simulate(1000, 123, X_n)
> 
> plot(f)
> points(X_o,y_o,pch=16)
> points(X_u,y_u,col='red',pch=16)
> for (i in 1:ncol(lus)) {
+     lines(X_n,ls[,i],col=rgb(0,0,0,.1),lwd=4)
+     lines(X_n,lus[,i],col=rgb(1,0,0,.1),lwd=4)
+     lines(X_n,lsu[,i],col=rgb(0,0,1,.1),lwd=4)
+ }
> 
> for (i in 1:length(X_n)) {
+     dsu=density(lsu[i,])
+     dus=density(lus[i,])
+     polygon(
+         X_n[i] + dsu$y/100,
+         dsu$x,
+         col=rgb(0,0,1,0.2),border=NA)
+     polygon(
+         X_n[i] + dus$y/100,
+         dus$x,
+         col=rgb(1,0,0,0.2),border=NA)
+     #test_that(desc="updated,simulated sample follows simulated,updated distribution",
+     #    expect_true(ks.test(lus[i,],lsu[i,])$p.value  > 0.01))
+ }
> 
> for (i in 1:length(X_n)) {
+     plot(density(ls[i,]))
+     lines(density(lsu[i,]),col='orange')
+     lines(density(lus[i,]),col='red')
+     if (sd(lsu[i,])>1e-3 && sd(lus[i,])>1e-3) # otherwise means that density is ~ dirac, so don't test
+     test_that(desc="updated,simulated sample follows simulated,updated distribution",
+         expect_gt(ks.test(lus[i,],lsu[i,])$p.value, 0.001))
+ }
Test passed 🥳
Test passed 😀
Test passed 🎉
Test passed 🥳
Test passed 🥇
Test passed 🥇
Test passed 🌈
Test passed 😸
Test passed 😀
Test passed 🌈
Test passed 😀
Test passed 🥳
Test passed 🥳
Test passed 😸
Test passed 😀
Test passed 🎊
> 
> 
> 
> 
> 
> 
> 
> 
> ############################## 2D ########################################
> 
> f <- function(X) apply(X, 1,
+                        function(x)
+                          prod(
+                            sin(2*pi*
+                                  ( x * (seq(0,1,l=1+length(x))[-1])^2 )
+                            )))
> n <- 100
> d <- 2
> 
> set.seed(1234)
> X_o <- matrix(runif(n*d),ncol=d) #seq(from = 0, to = 1, length.out = n)
> y_o <- f(X_o)
> #points(X_o, y_o)
> 
> lkd <- Kriging(y = y_o,
+               X = X_o,
+               kernel = "gauss",
+               regmodel = "linear",
+               optim = "none",
+               #normalize = TRUE,
+               parameters = list(theta = matrix(rep(0.1,d))))
> 
> ## Predict & simulate
> X_n = matrix(runif(100),ncol=d) #seq(0,1,,)
> 
> lpd = lkd$predict(X_n) # libK predict
> #lines(seq(0,1,,21),lp$mean,col='red')
> #polygon(c(seq(0,1,,21),rev(seq(0,1,,21))),c(lp$mean+2*lp$stdev,rev(lp$mean-2*lp$stdev)),col=rgb(1,0,0,0.2),border=NA)
> 
> lsd = lkd$simulate(1000, 123, X_n) # libK simulate
> #for (i in 1:100) {
> #    lines(seq(0,1,,21),ls[,i],col=rgb(1,0,0,.1),lwd=4)
> #}
> 
> for (i in 1:nrow(X_n)) {
+     m = lpd$mean[i,]
+     s = lpd$stdev[i,]
+     if (s > 1e-2) # otherwise means that density is ~ dirac, so don't test
+     test_that(desc="simulate sample follows predictive distribution",
+         expect_true(ks.test(lsd[i,],"pnorm",mean=m,sd=s)$p.value > 0.01))
+ }
Test passed 🎊
Test passed 😀
Test passed 🥇
Test passed 🥇
Test passed 🥳
Test passed 🎉
Test passed 🥇
Test passed 😸
Test passed 🥳
Test passed 🎊
Test passed 🎊
Test passed 🥳
Test passed 🎉
Test passed 😀
Test passed 🥇
Test passed 🎉
Test passed 🎉
Test passed 🥇
Test passed 😸
Test passed 🎉
Test passed 🌈
Test passed 🌈
Test passed 😀
Test passed 🌈
Test passed 🌈
Test passed 😀
Test passed 😀
Test passed 🎊
Test passed 🎉
Test passed 🎉
Test passed 🥇
Test passed 🥳
Test passed 🥳
Test passed 😸
Test passed 😀
Test passed 🥳
Test passed 🌈
Test passed 😸
Test passed 😸
Test passed 🥇
Test passed 😀
Test passed 🥳
> 
> ### Update
> #
> #X_u = c(.4,.6)
> #y_u = f(X_u)
> #
> ## new Kriging model from scratch
> #l2 = Kriging(y = matrix(c(y_o,y_u),ncol=1),
> #             X = matrix(c(X_o,X_u),ncol=1),
> #              kernel = "gauss",
> #              regmodel = "linear",
> #              optim = "none",
> #              parameters = list(theta = matrix(0.1)))
> #
> #lu = copy(lk)
> #update(lu, y_u,X_u)
> #
> ### Update, predict & simulate
> #
> #lp2 = l2$predict(seq(0,1,,21))
> #lpu = lu$predict(seq(0,1,,21))
> #
> #plot(f)
> #points(X_o,y_o)
> #lines(seq(0,1,,21),lp2$mean,col='red')
> #polygon(c(seq(0,1,,21),rev(seq(0,1,,21))),c(lp2$mean+2*lp2$stdev,rev(lp2$mean-2*lp2$stdev)),col=rgb(1,0,0,0.2),border=NA)
> #lines(seq(0,1,,21),lpu$mean,col='blue')
> #polygon(c(seq(0,1,,21),rev(seq(0,1,,21))),c(lpu$mean+2*lpu$stdev,rev(lpu$mean-2*lpu$stdev)),col=rgb(0,0,1,0.2),border=NA)
> #
> #ls2 = l2$simulate(100, 123, seq(0,1,,21))
> #lsu = lu$simulate(100, 123, seq(0,1,,21))
> #for (i in 1:100) {
> #    lines(seq(0,1,,21),ls2[,i],col=rgb(1,0,0,.1),lwd=4)
> #    lines(seq(0,1,,21),lsu[,i],col=rgb(0,0,1,.1),lwd=4)
> #}
> #
> #for (i in 1:21) {
> #    m2 = lp2$mean[i,]
> #    s2 = lp2$stdev[i,]
> #    mu = lpu$mean[i,]
> #    su = lpu$stdev[i,]
> #    test_that(desc="simulate sample follows predictive distribution",
> #        expect_true(ks.test(ls2[i,] - m2,"pnorm",mean=m2,sd=s2)$p.value  > 0.01))
> #    test_that(desc="simulate sample follows predictive distribution",
> #        expect_true(ks.test(lsu[i,] - mu,"pnorm",mean=mu,sd=su)$p.value  > 0.01))
> #}
> #
> #
> #
> ## Update simulate
> 
> i_u = c(9,13,25,43,42,35,24)
> X_u = X_n[i_u,,drop=FALSE]# c(.4,.6)
> y_u = f(X_u)
> 
> X_n = rbind(X_u+1e-2,X_n) # add some noise to avoid degenerate cases
> 
> #lkd0 = rlibkriging:::load.Kriging("/tmp/lkd.json")
> lsd = lkd$simulate(1000, 123, X_n, will_update=TRUE)
> #y_u = rs[i_u,1] # force matching 1st sim
> lusd = NULL
> lusd = lkd$update_simulate(y_u, X_u)
> 
> lud = copy(lkd)
> lud$update(matrix(y_u,ncol=1), X_u, refit=FALSE)
> #lud0 = rlibkriging:::load.Kriging("/tmp/lud.json")
> lsud = NULL
> lsud = lud$simulate(1000, 123, X_n)
> 
> #lkd$save("/tmp/lkd.json")
> #lud$save("/tmp/lud.json")
> 
> #plot(f)
> #points(X_o,y_o,pch=20)
> #points(X_u,y_u,col='red',pch=20)
> #for (i in 1:ncol(lus)) {
> #    lines(seq(0,1,,21),lus[,i],col=rgb(1,0,0,.1),lwd=4)
> #    lines(seq(0,1,,21),lsu[,i],col=rgb(0,0,1,.1),lwd=4)
> #}
> #
> #for (i in 1:length(X_n)) {
> #    dsu=density(lsu[i,])
> #    dus=density(lus[i,])
> #    polygon(
> #        X_n[i] + dsu$y/100,
> #        dsu$x,
> #        col=rgb(0,0,1,0.2),border=NA)
> #    polygon(
> #        X_n[i] + dus$y/100,
> #        dus$x,
> #        col=rgb(1,0,0,0.2),border=NA)
> #    #test_that(desc="updated,simulated sample follows simulated,updated distribution",
> #    #    expect_true(ks.test(lus[i,],lsu[i,])$p.value  > 0.01))
> #}
> 
> for (i in 1:nrow(X_n)) {
+     plot(density(lsd[i,]))
+     lines(density(lsud[i,]),col='orange')
+     lines(density(lusd[i,]),col='red')
+     if (sd(lsud[i,])>1e-3 && sd(lusd[i,])>1e-3) {# otherwise means that density is ~ dirac, so don't test
+     test_that(desc=paste0("updated,simulated sample follows simulated,updated distribution ",sd(lsud[i,]),",",sd(lusd[i,])),
+         expect_gt(ks.test(lusd[i,],lsud[i,])$p.value, 0.01))
+     }
+ }
Test passed 🎉
Test passed 🥳
Test passed 🥇
Test passed 🎉
Test passed 🎊
Test passed 🥇
Test passed 🥳
Test passed 🌈
Test passed 🥇
Test passed 😀
Test passed 🎊
Test passed 😸
Test passed 🎉
Test passed 🎊
Test passed 🎉
Test passed 🌈
Test passed 😀
Test passed 🥳
Test passed 😸
Test passed 🌈
Test passed 🥇
Test passed 😸
Test passed 🎊
Test passed 😸
Test passed 🎊
Test passed 🌈
Test passed 🎉
Test passed 🥇
Test passed 🎊
Test passed 🎊
Test passed 🌈
Test passed 🎊
Test passed 🌈
Test passed 🥇
Test passed 🌈
Test passed 🥳
Test passed 😸
Test passed 🥳
Test passed 😸
Test passed 🌈
Test passed 🥳
Test passed 🌈
Test passed 🥳
Test passed 🥳
Test passed 😀
Test passed 🌈
Test passed 🎉
Test passed 🥳
Test passed 😸
> 
> proc.time()
   user  system elapsed 
  15.84    2.06   17.93