@@ -481,25 +481,25 @@ test "Trilinear f = x*y*z T: Tensor[float]":
481
481
check abs (spline.eval (i, j, k)[2 ] - 1 ) < 1 e-16
482
482
483
483
test " rbfBase f=x*y*z" :
484
- let pos = meshgrid (arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ))
484
+ let pos = numericalnim. meshgrid (arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ))
485
485
let vals = pos[_, 0 ] *. pos[_, 1 ] *. pos[_, 2 ]
486
486
let rbfObj = newRbfBase (pos, vals)
487
487
488
488
# We want test points in the interior to avoid the edges
489
- let xTest = meshgrid (arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ))
489
+ let xTest = numericalnim. meshgrid (arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ))
490
490
let yTest = rbfObj.eval (xTest)
491
491
let yCorrect = xTest[_, 0 ] *. xTest[_, 1 ] *. xTest[_, 2 ]
492
492
for x in abs (yCorrect - yTest):
493
493
check x < 0.16
494
494
check mean_squared_error (yTest, yCorrect) < 2 e-4
495
495
496
496
test " rbf f=x*y*z" :
497
- let pos = meshgrid (arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ))
497
+ let pos = numericalnim. meshgrid (arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ), arraymancer.linspace (0.0 , 1.0 , 5 ))
498
498
let vals = pos[_, 0 ] *. pos[_, 1 ] *. pos[_, 2 ]
499
499
let rbfObj = newRbf (pos, vals)
500
500
501
501
# We want test points in the interior to avoid the edges
502
- let xTest = meshgrid (arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ))
502
+ let xTest = numericalnim. meshgrid (arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ), arraymancer.linspace (0.1 , 0.9 , 10 ))
503
503
let yTest = rbfObj.eval (xTest)
504
504
let yCorrect = xTest[_, 0 ] *. xTest[_, 1 ] *. xTest[_, 2 ]
505
505
for x in abs (yCorrect - yTest):
0 commit comments