forked from pydata/patsy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathget-R-bs-test-vectors.R
More file actions
92 lines (83 loc) · 2.13 KB
/
get-R-bs-test-vectors.R
File metadata and controls
92 lines (83 loc) · 2.13 KB
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
# R code to dump the output from 'bs()' in many situations, for
# checking compatibility.
#
# Usage:
# Rscript tools/get-R-bs-test-vectors.R > patsy/test_splines_bs_data.py
cat("# This file auto-generated by tools/get-R-bs-test-vectors.R\n")
cat(sprintf("# Using: %s\n", R.Version()$version.string))
cat("import numpy as np\n")
options(digits=20)
library(splines)
x <- (1.5)^(0:19)
MISSING <- "MISSING"
is.missing <- function(obj) {
length(obj) == 1 && obj == MISSING
}
pyprint <- function(arr) {
if (is.missing(arr)) {
cat("None\n")
} else {
cat("np.array([")
for (val in arr) {
cat(val)
cat(", ")
}
cat("])")
if (!is.null(dim(arr))) {
cat(".reshape((")
for (size in dim(arr)) {
cat(sprintf("%s, ", size))
}
cat("), order=\"F\")")
}
cat("\n")
}
}
num.tests <- 0
dump.bs <- function(degree, df, knots,
intercept, Boundary.knots) {
cat("--BEGIN TEST CASE--\n")
cat(sprintf("degree=%s\n", degree))
cat(sprintf("df=%s\n", if (is.missing(df)) "None" else df))
cat(sprintf("intercept=%s\n", intercept))
cat("Boundary.knots=")
pyprint(Boundary.knots)
cat("knots=")
pyprint(knots)
args <- list(x=x, degree=degree, intercept=intercept)
if (!is.missing(df)) {
args$df <- df
}
if (!is.missing(knots)) {
args$knots <- knots
}
if (!is.missing(Boundary.knots)) {
args$Boundary.knots <- Boundary.knots
}
result <- do.call(bs, args)
cat("output=")
pyprint(result)
cat("--END TEST CASE--\n")
assign("num.tests", num.tests + 1, envir=.GlobalEnv)
}
cat("R_bs_test_x = ")
pyprint(x)
cat("R_bs_test_data = \"\"\"\n")
for (degree in c(1, 3, 5)) {
for (intercept in c(TRUE, FALSE)) {
for (Boundary.knots in list(MISSING, c(0, 3000))) {
for (df in c(3, 5, 12)) {
if (df < degree + 2) {
next
} else {
dump.bs(degree, df, MISSING, intercept, Boundary.knots)
}
}
for (knots in list(c(), c(100), c(1000), c(10, 100, 1000))) {
dump.bs(degree, MISSING, knots, intercept, Boundary.knots)
}
}
}
}
cat("\"\"\"\n")
cat(sprintf("R_bs_num_tests = %s\n", num.tests))