def setUp(self):
''' ThreadTest::setUp
Addition to the TestBase.setUp, creates a user to use in the thread
actions
'''
TestBase.setUp(self)
self.register(self.user)
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.n = 100
self.titles = ['Array 1', 'Array 2', 'Array 3']
self.header = 'A header line'
##### Now generate output for testing ####
filename = os.path.join(self.test_dir, 'test.dat')
self.data = np.random.random((self.n, len(self.titles)))
header = [self.header, '%d' % len(self.titles)]
for ln in self.titles:
header.append(ln + '\n')
header = '\n'.join(header)
np.savetxt(filename,
self.data,
delimiter=' ',
header=header,
comments='')
# Set up the reader
reader = GSLibReader()
reader.AddFileName(filename)
# Perform the read
reader.Update()
self.HEADER = reader.get_file_header()
self.TABLE = reader.GetOutput()
def setUp(self):
''' ClassTest::setUp
Overloaded setup call, this is response for creating a root user by
default.
'''
TestBase.setUp(self)
self.register(self.root_user)
def setUp(self):
''' QuestTest::setUp
Overloaded setup call, this is responsible for creating a root user by
default. Mostly useful because of the requirement of a DM user for
most of the tests.
'''
TestBase.setUp(self)
self.register(self.root_user)
def setUp(self):
''' QuestTest::setUp
Overloaded setup call, this is responsible for creating a root user by
default. Mostly useful because of the requirement of a DM user for
most of the tests.
'''
TestBase.setUp(self)
self.register(self.root_user)
def setUp(self):
TestBase.setUp(self)
self.test_dir = tempfile.mkdtemp()
treeMesh = """16 16 16
0.0000 0.0000 48.0000
1.000 2.000 3.000
29
1 1 1 8
9 1 1 8
1 9 1 8
9 9 1 8
1 1 9 4
5 1 9 4
9 1 9 8
1 5 9 4
5 5 9 4
1 9 9 8
9 9 9 8
1 1 13 2
3 1 13 2
5 1 13 4
1 3 13 2
3 3 13 2
1 5 13 4
5 5 13 4
1 1 15 1
2 1 15 1
3 1 15 2
1 2 15 1
2 2 15 1
1 3 15 2
3 3 15 2
1 1 16 1
2 1 16 1
1 2 16 1
2 2 16 1
"""
# Write out mesh file
filename = os.path.join(self.test_dir, 'octree.msh')
self.meshFileName = filename
with open(filename, 'w') as f:
f.write(treeMesh)
# write out model file(s)
self.nt = 5
self.modelFileNames = ['model%d.mod' % i for i in range(self.nt)]
self.modelFileNames = [
os.path.join(self.test_dir, self.modelFileNames[i])
for i in range(self.nt)
]
self.arrs = [None] * self.nt
for i in range(self.nt):
self.arrs[i] = np.random.random(29)
np.savetxt(self.modelFileNames[i],
self.arrs[i],
delimiter=' ',
comments='! ')
return
def setUp(self):
''' PlayerTest::setUp
Overloaded setup call, This is responsible for creating a root user by
default and logging them out. Mostly useful because all of the users
created subsequently will have default permissions.
'''
TestBase.setUp(self)
self.register(self.root_user)
self.logout()
def setUp(self):
''' ForumTest::setUp
Additions to the TestBase.setUp, creates a user to use in the forum
actions
'''
TestBase.setUp(self)
self.register(self.root_user)
self.logout()
self.register(self.user)
def setUp(self):
TestBase.setUp(self)
# Create some input tables
self.t0 = vtk.vtkTable()
# Populate the tables
self.n = 400
self.title = 'Array 0'
self.arr = np.random.random(self.n) # Table 0
self.t0.AddColumn(interface.convertArray(self.arr, self.title))
return
def setUp(self):
TestBase.setUp(self)
self.RTOL = 0.00001 # As higi as rotation precision can get
x = np.array([0.0, 1.0, 0.0])
y = np.array([0.0, 0.0, 1.0])
z = np.array([0.0, 0.0, 0.0])
x = np.reshape(x, (len(x), -1))
y = np.reshape(y, (len(y), -1))
z = np.reshape(z, (len(z), -1))
self.pts = np.concatenate((x, y, z), axis=1)
self.vtkpoints = interface.pointsToPolyData(self.pts)
return
def setUp(self):
TestBase.setUp(self)
# create a volumetric data set
self.grid = PVGeo.model_build.CreateTensorMesh().apply()
# create an unudulating surface in the grid domain
# make XY random in the grid bounds
# make Z ranome within a very small domain inside the grid
bnds = self.grid.GetBounds()
x = np.random.uniform(bnds[0], bnds[1], 5000)
y = np.random.uniform(bnds[2], bnds[3], 5000)
z = np.random.uniform(-200, -100, 5000)
self.points = interface.points_to_poly_data(np.c_[x,y,z])
def setUp(self):
TestBase.setUp(self)
# Create some input tables
self.t0 = vtk.vtkTable()
# Populate the tables
self.arrs = [None, None]
self.n = 400
self.titles = ('Array 0', 'Array 1')
self.arrs[0] = np.random.random(self.n) # Table 0
self.arrs[1] = np.random.random(self.n) # Table 0
self.t0.AddColumn(interface.convertArray(self.arrs[0], self.titles[0]))
self.t0.AddColumn(interface.convertArray(self.arrs[1], self.titles[1]))
return
def setUp(self):
TestBase.setUp(self)
# Create some input tables
self.idi = vtk.vtkImageData()
self.idi.SetDimensions(20, 2, 10)
self.idi.SetSpacing(1, 1, 1)
self.idi.SetOrigin(100, 100, 100)
# Populate the tables
self.n = 400
self.title = 'Array 0'
self.arr = np.random.random(self.n)
self.idi.GetPointData().AddArray(interface.convert_array(self.arr, self.title))
return
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.mesh = """9
-300.0 -180.0 1
-130.0 1
-110.0 1
-100.0 1
100.0 40
110.0 1
130.0 1
180.0 1
300.0 1
14
-10.0 10.0 5
22.0 4
42.0 5
57.0 3
63.0 1
71.0 1
81.0 1
95.0 1
115.0 1
140.0 1
170.0 1
205.0 1
245.0 1
300.0 1
"""
self.origin = (-300, 0, 10)
self.nx = 48
self.nz = 27
self.shape = (self.nx, 1, self.nz)
self.extent = (0, self.shape[0], 0, self.shape[1], 0, self.shape[2])
self.data_name = 'foo'
##### Now generate output for testing ####
# Produce data and write out files:
meshname = self._write_mesh()
modname, self.data = self._write_model()
# Set up the reader:
reader = TensorMeshReader()
reader.set_mesh_filename(meshname)
# Get and test output:
reader.Update() # Test the read up front for the mesh
reader.add_model_file_name(modname)
reader.set_data_name(self.data_name)
reader.Update() # Now read the models upfront
self.GRID = reader.GetOutput()
return
def gen_kernel_for_vector(op_desc,
_,
index_table=None,
config: NamedTuple = None,
idx=None,
gen_tiling_spaces=False):
"""Compile kernel module for vector"""
test_base = TestBase()
test_base.params_init(op_desc[0][0:4] + str(idx), os.getcwd())
kernel_name = "poly_"
if idx is not None:
kernel_name += str(idx)
if config is None:
attrs = {'dim': ""}
else:
tiling = [[getattr(config, name), 1]
for name in getattr(config, '_fields')
if name.startswith('tiling')]
tiling_param = []
for i, element in enumerate(tiling):
tiling_param.append(index_table[i] + element)
dim_info = ct_util.set_dims(tuple(tiling_param))
attrs = {'dim': dim_info}
_, func, args, kwargs = test_base.ana_args(op_desc)
if 'attrs' in kwargs.keys():
kwargs['attrs']['dim'] = attrs['dim']
kwargs['attrs']['tuning'] = gen_tiling_spaces
kwargs['attrs']['kernel_name'] = kernel_name
else:
for _, arg_ in enumerate(args):
if isinstance(arg_, dict):
arg_['dim'] = attrs['dim']
arg_['tuning'] = gen_tiling_spaces
arg_['kernel_name'] = kernel_name
break
try:
if gen_tiling_spaces:
mod, expect, param_for_mod = func(*args, **kwargs)
mod = list(mod)
mod.append(expect)
mod.append(param_for_mod)
else:
mod = func(*args, **kwargs)
except BaseException as e:
print("Compile ERROR message:", e)
print(func)
print("Compile ERROR")
raise Exception("Compile ERROR")
return mod
def setUp(self):
TestBase.setUp(self)
# create a volumetric data set
self.grid = PVGeo.model_build.CreateTensorMesh().Apply()
# create a spline throught the data set
def path1(y):
"""Equation: x = a(y-h)^2 + k"""
a = -0.0001
x = a * y**2 + 1000
idxs = np.argwhere(x > 0)
return x[idxs][:, 0], y[idxs][:, 0]
x, y = path1(np.arange(-500.0, 1500.0, 25.0))
zo = np.linspace(9.0, 11.0, num=len(y))
coords = np.vstack((x, y, zo)).T
self.points = interface.pointsToPolyData(coords)
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.n = 100
self.shape = (150, 200, 20)
self.extent = (0, self.shape[0], 0, self.shape[1], 0, self.shape[2])
self.titles = ['Array 1', 'Array 2', 'Array 3']
##### Now generate output for testing ####
fname = os.path.join(self.test_dir, 'test.dat')
self.data = np.random.random((self.n, len(self.titles)))
header = ['%d %d %d' % self.shape, '%d' % len(self.titles)]
for ln in self.titles:
header.append(ln + '\n')
header = '\n'.join(header)
np.savetxt(fname, self.data, delimiter=' ', header=header, comments='')
# Set up the reader
self.GRID = SGeMSGridReader().Apply(fname)
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.filename = os.path.join(self.test_dir, 'test.dat')
sample = """ncols 4
nrows 6
xllcorner 100.0
yllcorner 50.0
cellsize 50.0
NODATA_value -9999
-9999 -9999 5 2
-9999 20 100 36
3 8 35 10
32 42 50 6
88 75 27 9
13 5 1 -9999
"""
with open(self.filename, 'w') as f:
f.write(sample)
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.filename = os.path.join(self.test_dir, 'test.xyz')
# Make a temporary file to test:
self.nrows = 100
self.ncols = 8 # LEAVE ALONE
self.header = 'X, dx, Y, dy, Z, dz, approximate distance, cell index'
self.data = np.random.random((self.nrows, self.ncols))
np.savetxt(self.filename,
self.data,
header=self.header,
comments='! ',
fmt='%.6e')
reader = XYZTextReader()
reader.AddFileName(self.filename)
reader.Update()
self.TABLE = reader.GetOutput()
return
def setUp(self):
TestBase.setUp(self)
# Create some input tables
self.t0 = vtk.vtkTable()
self.t1 = vtk.vtkTable()
# Populate the tables
self.n = 100
self.titles = ('Array 0', 'Array 1', 'Array 2')
self.arrs = [None, None, None]
self.arrs[0] = np.random.random(self.n) # Table 0
self.arrs[1] = np.random.random(self.n) # Table 0
self.arrs[2] = np.random.random(self.n) # Table 1
self.t0.AddColumn(interface.convertArray(self.arrs[0], self.titles[0]))
self.t0.AddColumn(interface.convertArray(self.arrs[1], self.titles[1]))
self.t1.AddColumn(interface.convertArray(self.arrs[2], self.titles[2]))
# Now use the `CombineTables` filter:
f = CombineTables()
f.SetInputDataObject(0, self.t0)
f.SetInputDataObject(1, self.t1)
f.Update()
self.TABLE = f.GetOutputDataObject(0)
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.commafilename = os.path.join(self.test_dir, 'comma.txt')
self.tabfilename = os.path.join(self.test_dir, 'tab.txt')
# Make a temporary delimited text file to test:
lines = [
'This is a header line to skip',
'int,str,float ! Comment,this line has the data array names',
'5,foo,6.9', '1,bar,8.5 ! another comment', '3,oof,7.7'
]
# Append newlines
lines = [ln + '\n' for ln in lines]
# Now write contents to files
f = open(self.commafilename, 'w')
f.writelines(lines)
f.close()
f = open(self.tabfilename, 'w')
f.writelines([ln.replace(',', '\t') for ln in lines])
f.close()
return
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.origin = (-350, -400, 0)
self.xCells = '200 100 50 20*50.0 50 100 200'
self.yCells = '200 100 50 21*50.0 50 100 200'
self.zCells = '20*25.0 50 100 200'
self.shape = (26, 27, 23)
self.n = self.shape[0] * self.shape[1] * self.shape[2]
self.extent = (0, self.shape[0], 0, self.shape[1], 0, self.shape[2])
self.data_name = 'foo'
##### Now generate output for testing ####
# Produce data and write out files:
self.meshname = self._write_mesh()
self.modname, self.data = self._write_model()
self.modname_multi, self.data_multi = self._write_model_multi()
# Set up the reader:
reader = TensorMeshReader()
reader.set_mesh_filename(self.meshname)
# Get and test output:
reader.Update() # Read only mesh upfront
reader.add_model_file_name(self.modname)
reader.set_data_name(self.data_name)
reader.Update() # Read models upfront
self.GRID = reader.GetOutput()
#### Now read mesh with multi component data
# Set up the reader:
reader = TensorMeshReader()
reader.set_mesh_filename(self.meshname)
# Get and test output:
reader.Update() # Read only mesh upfront
reader.add_model_file_name(self.modname_multi)
reader.set_data_name(self.data_name)
reader.Update() # Read models upfront
self.GRID_MULTI = reader.GetOutput()
def setUp(self):
TestBase.setUp(self)
self.test_dir = tempfile.mkdtemp()
self.filename = os.path.join(os.path.dirname(__file__), 'data/surfer-grid.grd')
def setUp(self):
TestBase.setUp(self)
self.tree = etree.fromstring(TestBase.TRACK_DETAILS_SAMPLE)
def __init__(self, methodName="runTest"):
TestBase.__init__(self, methodName)
def tearDown(self):
TestBase.tearDown(self)
os.system("rm -rf '%s'" % os.path.join(dbdir, self.dbname))
os.unlink(os.path.join(dbdir, '%s.ini' % self.dbname))
def setUp(self):
TestBase.setUp(self)
ans = self.client.createdb(name=self.dbname, conf=config)
self.client.setdb(self.dbname)
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.filename = os.path.join(os.path.dirname(__file__), 'data',
'vd2.asc')
def setUp(self):
TestBase.setUp(self)
self.filename = os.path.join(os.path.dirname(__file__),
'data/Craig-Chile/LdM_topo.topo')
def __init__(self, methodName='runTest'):
TestBase.__init__(self, methodName)
def setUp(self):
TestBase.setUp(self)
self.RTOL = 0.00001 # As high as rotation precision can get
return
def setUp(self):
TestBase.setUp(self)
self.register()
def setUp(self):
TestBase.setUp(self)
# create a volumetric data set
self.grid = PVGeo.model_build.CreateTensorMesh().Apply()
def setUp(self):
TestBase.setUp(self)
def setUp(self):
TestBase.setUp(self)
self.test_dir = tempfile.mkdtemp()
self.filename = os.path.join(self.test_dir, 'test.txt')
def setUp(self):
TestBase.setUp(self)
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
self.n = 100
return
def tearDown(self):
# Remove the test data directory after the test
shutil.rmtree(self.test_dir)
TestBase.tearDown(self)
def setUp(self):
TestBase.setUp(self)
self.filename = os.path.join(os.path.dirname(__file__),
'data/Craig-Chile/LdM_grav_obs.grv')
