Source code for sfepy.tests.test_mesh_generators

import os.path as op
import numpy as nm

import sfepy.base.testing as tst

tolerance = 1e-6

[docs] def test_gen_block_mesh(output_dir): from sfepy.mesh.mesh_generators import gen_block_mesh mesh = gen_block_mesh([1, 2, 3], [4, 3, 5], [2, 1, 0], verbose=False) filename = op.join(output_dir, 'gen_block.mesh') mesh.write(filename) tst.report('block mesh generated') csum = nm.sum(mesh.coors - nm.min(mesh.coors, axis=0), axis=0) assert nm.linalg.norm(csum - nm.array([30, 60, 90])) < tolerance
[docs] def test_gen_cylinder_mesh(output_dir): from sfepy.mesh.mesh_generators import gen_cylinder_mesh mesh = gen_cylinder_mesh([0.5, 1, 2, 1.5, 3], [5, 4, 3], [0, 2, 1], axis='z', non_uniform=True, verbose=False) filename = op.join(output_dir, 'gen_cylinder.mesh') mesh.write(filename) tst.report('cylinder mesh generated') csum = nm.sum(mesh.coors - nm.min(mesh.coors, axis=0), axis=0) assert nm.linalg.norm(csum - nm.array([120, 90, 90])) < tolerance
[docs] def test_gen_extended_block_mesh(output_dir): from sfepy.mesh.mesh_generators import gen_extended_block_mesh mesh = gen_extended_block_mesh([2, 3, 1], [5, 3, 4], [14, 10, 20], 5, [1, 0, 2]) filename = op.join(output_dir, 'gen_extended_block.mesh') mesh.write(filename) tst.report('extended block mesh generated') csum = nm.sum(mesh.coors - nm.min(mesh.coors, axis=0), axis=0) assert nm.linalg.norm(csum - nm.array([1932, 1380, 2760])) < tolerance
[docs] def test_gen_tiled_mesh(output_dir): from sfepy.mesh.mesh_generators import gen_block_mesh, gen_tiled_mesh mesh0 = gen_block_mesh([1, 2, 3], [4, 3, 5], [2, 1, 0], verbose=False) mesh = gen_tiled_mesh(mesh0, [3, 1, 5]) filename = op.join(output_dir, 'gen_tiled.mesh') mesh.write(filename) tst.report('tiled mesh generated') csum = nm.sum(mesh.coors - nm.min(mesh.coors, axis=0), axis=0) assert nm.linalg.norm(csum - nm.array([945, 630, 4725])) < tolerance
[docs] def test_gen_mesh_from_geom(output_dir): from shutil import which from sfepy.mesh.geom_tools import geometry from sfepy.mesh.mesh_generators import gen_mesh_from_geom nx, ny, nz = 1, 2, 3 ok = True # 2D if which('triangle'): geo = geometry(2) geo.addpoint(0, [0, 0]) geo.addpoint(1, [nx, 0]) geo.addpoint(2, [0, ny]) geo.addline(3, [0, 1]) geo.addline(4, [1, 2]) geo.addline(5, [2, 0]) geo.addsurface(6, [3, 4, 5]) geo.addphysicalsurface(1, [6]) mesh = gen_mesh_from_geom(geo, a=0.01, refine=True) filename = op.join(output_dir, 'gen_triangle.mesh') mesh.write(filename) tst.report('mesh generated by "triangle" generator') orign = 98 actn = mesh.coors.shape[0] ok = ok and nm.abs((actn - orign) / float(orign)) < 0.05 # 3D if which('tetgen'): geo = geometry(3) geo.addpoint(0, [0, 0 , 0]) geo.addpoint(1, [nx, 0, 0]) geo.addpoint(2, [0, ny, 0]) geo.addpoint(3, [0, 0, nz]) geo.addline(4, [0, 1]) geo.addline(5, [1, 2]) geo.addline(6, [2, 0]) geo.addline(7, [0, 3]) geo.addline(8, [1, 3]) geo.addline(9, [2, 3]) geo.addsurface(10, [4, 8, -7]) geo.addsurface(11, [5, 9, -8]) geo.addsurface(12, [6, 7, -9]) geo.addsurface(13, [4, 5, 6]) #Volume geo.addvolume(14, [10, 11, 12, 13]) geo.addphysicalvolume(1, [14]) mesh = gen_mesh_from_geom(geo, a=0.001, refine=True) filename = op.join(output_dir, 'gen_tetgen.mesh') mesh.write(filename) tst.report('mesh generated by "tetgen" generator') orign = 898 actn = mesh.coors.shape[0] ok = ok and nm.abs((actn - orign) / float(orign)) < 0.05 assert ok
[docs] def test_gen_mesh_from_voxels(output_dir): from sfepy.mesh.mesh_generators import gen_mesh_from_voxels voxels = nm.array([[[0, 0, 0, 0, 1], [0, 0, 0, 1, 1], [0, 0, 0, 1, 1], [0, 0, 0, 0, 1]], [[0, 0, 0, 1, 1], [0, 1, 1, 1, 1], [0, 1, 1, 1, 1], [0, 0, 0, 1, 1]], [[1, 0, 0, 1, 1], [1, 1, 1, 1, 1], [1, 1, 1, 1, 1], [1, 0, 0, 1, 1]]]) mesh = gen_mesh_from_voxels(voxels, [0.5, 0.3, 1.]) filename = op.join(output_dir, 'gen_voxels.mesh') mesh.write(filename) tst.report('voxel based mesh generated') csum = nm.sum(mesh.coors - nm.min(mesh.coors, axis=0), axis=0) assert nm.linalg.norm(csum - nm.array([90, 48.3, 265])) < tolerance