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
|
#!/usr/bin/python3
# Copyright (C) 2016 EDF
# All Rights Reserved
# This code is published under the GNU Lesser General Public License (GNU LGPL)
import numpy as np
import unittest
import random
import math
import StOptGrids
# unit test for Legendre grids
#############################
class testGrids(unittest.TestCase):
# test Legendre grids
def testLegendreGrids(self):
# low value for the mesh
lowValues =np.array([1.,2.,3.],dtype=float)
# size of the mesh
step = np.array([0.7,2.3,1.9],dtype=float)
# number of step
nbStep = np.array([4,5,6], dtype=np.int32)
# degree of the polynomial in each direction
degree = np.array([2,1,3], dtype=np.int32)
# create the Legendre grid
grid = StOptGrids.RegularLegendreGrid(lowValues,step,nbStep,degree )
iterGrid = grid.getGridIterator()
# array to store
data = np.empty(grid.getNbPoints())
# iterates on point
while( iterGrid.isValid()):
#get coordinates of the point
pointCoord = iterGrid.getCoordinate()
data[iterGrid.getCount()] = math.log(1. + pointCoord.sum())
iterGrid.next()
# get back an interpolator
ptInterp = np.array([2.3,3.2,5.9],dtype=float)
interpol = grid.createInterpolator(ptInterp)
# calculate interpolated value
interpValue = interpol.apply(data)
print(("Interpolated value Legendre" , interpValue))
# test grids function
iDim = grid.getDimension()
pt = grid.getExtremeValues()
if __name__ == '__main__':
unittest.main()
|
