netgen/tests/pytest/test_tutorials.py

164 lines
6.0 KiB
Python
Raw Normal View History

2019-08-08 13:12:33 +05:00
import os, pytest
from netgen.meshing import meshsize, MeshingParameters, SetMessageImportance
2019-08-08 13:12:33 +05:00
import netgen.csg as csg
import netgen.stl as stl
2019-10-23 14:27:05 +05:00
import netgen.geom2d as geom2d
2019-10-04 15:25:14 +05:00
from pyngcore import TaskManager
import json
try:
import netgen.occ as occ
has_occ = occ.occ_version >= "7.4.0"
2021-07-26 12:37:07 +05:00
has_occ = False # skip occ tests (current development on interface)
except ImportError:
has_occ = False
2019-08-08 13:12:33 +05:00
SetMessageImportance(0)
def round(x, digits=11):
try:
return float(("{:."+str(digits)+"g}").format(x))
except: #list
return [float(("{:."+str(digits)+"g}").format(y)) for y in x]
def getData(mesh, mp):
out = {}
out['ne1d'] = len(mesh.Elements1D())
out['ne2d'] = len(mesh.Elements2D())
out['ne3d'] = len(mesh.Elements3D())
# round badness to avoid fluctuations in last digits
out["total_badness"] = round(mesh.CalcTotalBadness(mp))
angles = mesh.CalcMinMaxAngle()
out["angles_trig"] = round(angles["trig"], 5)
out["angles_tet"] = round(angles["tet"], 5)
out["quality_histogram"] = str(list(mesh.GetQualityHistogram()))
return out
def checkData(mesh, mp, ref):
data = getData(mesh, mp)
assert ref['ne1d'] == data['ne1d']
assert ref['ne2d'] == data['ne2d']
assert ref['ne3d'] == data['ne3d']
assert json.loads(ref['quality_histogram']) == pytest.approx(json.loads(data['quality_histogram']), abs=1, rel=0.4)
assert ref['total_badness'] == pytest.approx(data['total_badness'], rel=1e-5)
assert ref['angles_trig'] == pytest.approx(data['angles_trig'], rel=1e-4)
assert ref['angles_tet'] == pytest.approx(data['angles_tet'], rel=1e-4)
2019-10-07 16:41:15 +05:00
# get tutorials
2019-08-08 13:12:33 +05:00
def getFiles(fileEnding):
r, d, files = next(os.walk(os.path.join("..","..","tutorials")))
2019-10-23 14:27:05 +05:00
return [f for f in files if f.endswith(fileEnding)]
2019-08-08 13:12:33 +05:00
2019-10-07 16:41:15 +05:00
# get additional tests
def getAdditionalFiles(fileEnding):
r, d, files = next(os.walk("geofiles"))
2019-10-23 14:27:05 +05:00
return [f for f in files if f.endswith(fileEnding)]
2019-10-07 16:41:15 +05:00
@pytest.fixture
def refdata():
return json.load(open('results.json','r'))
2019-08-08 13:12:33 +05:00
def getMeshingparameters(filename):
2019-10-04 15:25:14 +05:00
standard = [MeshingParameters()] + [MeshingParameters(ms) for ms in (meshsize.very_coarse, meshsize.coarse, meshsize.moderate, meshsize.fine, meshsize.very_fine)]
2019-08-08 13:12:33 +05:00
if filename == "shell.geo":
return [] # do not test this example cause it needs so long...
if filename == "manyholes2.geo":
return [standard[1]] # this gets too big for finer meshsizes
if filename in ("manyholes.geo", "frame.step"):
return standard[:3] # this gets too big for finer meshsizes
if filename == "extrusion.geo":
return standard[:-1]
if filename == "screw.step":
return standard[3:] # coarser meshes don't work here
if filename == "cylinder.geo":
2024-03-12 01:10:56 +05:00
return [] # gives inconsistent reults
if filename == "cylsphere.geo":
return standard[0:2] + standard[3:] # coarse gives inconsistent reults (other mesh on MacOS)
if filename == "part1.stl":
return standard[0:1] + standard[2:] # very coarse does not work
2019-08-08 13:12:33 +05:00
return standard
2019-10-23 14:27:05 +05:00
_geofiles = getFiles(".in2d") + getFiles(".geo") + getFiles(".stl")
if has_occ:
2019-10-23 14:27:05 +05:00
_geofiles += getFiles(".step")
2019-10-04 15:25:14 +05:00
_geofiles.sort()
2019-10-23 14:27:05 +05:00
_additional_testfiles = getAdditionalFiles(".stl")
2019-10-07 16:41:15 +05:00
if has_occ:
2019-10-23 14:27:05 +05:00
_additional_testfiles += getAdditionalFiles(".step")
2019-10-07 16:41:15 +05:00
_additional_testfiles.sort()
2019-10-04 15:25:14 +05:00
2019-08-08 13:12:33 +05:00
def generateMesh(filename, mp):
2019-10-07 16:41:15 +05:00
folder = os.path.join("..","..","tutorials") if filename in _geofiles else "geofiles"
2019-08-08 13:12:33 +05:00
if filename.endswith(".geo"):
2019-10-07 16:41:15 +05:00
geo = csg.CSGeometry(os.path.join(folder, filename))
2019-08-08 13:12:33 +05:00
elif filename.endswith(".stl"):
2019-10-07 16:41:15 +05:00
geo = stl.STLGeometry(os.path.join(folder, filename))
2019-08-08 13:12:33 +05:00
elif filename.endswith(".step"):
2019-10-07 16:41:15 +05:00
geo = occ.OCCGeometry(os.path.join(folder, filename))
2019-10-23 14:27:05 +05:00
elif filename.endswith(".in2d"):
geo = geom2d.SplineGeometry(os.path.join(folder, filename))
2019-10-04 15:25:14 +05:00
return geo.GenerateMesh(mp)
2019-08-08 13:12:33 +05:00
2019-08-29 20:56:47 +05:00
def isSlowTest(filename):
return filename in ["cubemcyl.geo", "frame.step", "revolution.geo", "manyholes.geo", "torus.geo",
"cubemsphere.geo", "manyholes2.geo", "matrix.geo", "trafo.geo", "ellipticcone.geo",
"period.geo", "shaft.geo", "cubeandring.geo", "ellipticcyl.geo",
2019-10-07 16:41:15 +05:00
"ellipsoid.geo", "cone.geo", "plane.stl"]
2019-08-29 20:56:47 +05:00
def getParameters():
res = []
2019-10-07 16:41:15 +05:00
for f in _geofiles + _additional_testfiles:
for i,mp in enumerate(getMeshingparameters(f)):
if isSlowTest(f):
res.append( pytest.param(f, mp, i, marks=pytest.mark.slow ) )
else:
res.append( (f, mp, i) )
return res
@pytest.mark.parametrize(("filename", "mp", "i"), getParameters())
def test_geoFiles(filename, mp, i, refdata):
ref = refdata[filename]
2019-10-04 15:25:14 +05:00
import filecmp
print("load geo", filename)
mp = MeshingParameters(mp, parallel_meshing=False)
mesh = generateMesh(filename, mp)
mesh.Save(filename+'_seq.vol.gz')
with TaskManager():
mesh_par = generateMesh(filename, mp)
mesh_par.Save(filename+'_par.vol.gz')
2019-09-09 20:05:06 +05:00
assert filecmp.cmp(filename+'_seq.vol.gz', filename+'_par.vol.gz')
checkData(mesh, mp, ref[i])
2019-09-09 20:05:06 +05:00
2019-08-08 13:12:33 +05:00
2024-03-12 01:10:56 +05:00
def generateResultFile(output_file='results.json'):
import time
data = {}
with TaskManager():
2019-10-07 16:41:15 +05:00
for _file in _geofiles + _additional_testfiles:
print("generate "+_file)
2019-08-08 13:12:33 +05:00
start = time.time()
mps = getMeshingparameters(_file)
if not mps:
continue
meshdata = []
for mp in mps:
2021-04-13 15:11:10 +05:00
try:
mesh = generateMesh(_file, mp)
except Exception as e:
print("Meshingparameters: ", mp)
raise e
meshdata.append( getData(mesh, mp) )
data[_file] = meshdata
2019-10-07 16:41:15 +05:00
print("needed", time.time() - start, "seconds")
2019-08-08 13:12:33 +05:00
s = json.dumps(data, sort_keys=True, indent=4)
2024-03-12 01:10:56 +05:00
open(output_file, "w").write(s)
2019-10-07 16:41:15 +05:00
print("done")
2019-08-08 13:12:33 +05:00
if __name__ == "__main__":
2024-03-12 01:10:56 +05:00
import sys
generateResultFile(*sys.argv[1:])