netgen/tests/pytest/test_csg2d.py
2020-08-28 18:35:35 +02:00

61 lines
1.3 KiB
Python

from netgen.geom2d import *
import pytest
import math
from pytest import approx
def test_two_circles():
c1 = Circle(center=(0,0), radius=1)
c2 = c1.Rotate(deg=45)
s = c1*c2
geo = CSG2d()
geo.Add(s)
m = geo.GenerateMesh()
assert len(m.Elements2D()) > 0
ngs = pytest.importorskip("ngsolve")
mesh = ngs.Mesh(m)
mesh.Curve(5)
assert ngs.Integrate(1.0, mesh) == approx(math.pi)
ngs.Draw(mesh)
def test_two_edge():
s = Solid2d( [(-1,0), cp(0,1), (1,0), cp(0,2)] )
geo = CSG2d()
geo.Add(s)
m = geo.GenerateMesh()
assert len(m.Elements2D()) > 0
ngs = pytest.importorskip("ngsolve")
g = geo.GenerateSplineGeometry()
ngs.Draw(g)
mesh = ngs.Mesh(m)
mesh.Curve(5)
ngs.Draw(mesh)
def test_trig_and_circle():
g = CSG2d()
trig = Solid2d( [(0,0), (1,1), (-1,1) ] ).BC("diamond")
circle = Circle( center=(0,0.101), radius=0.1).BC("circle") # TODO: Failing with center=(0,0.1)
d = trig-circle
g.Add(d)
g.Add(circle)
m = g.GenerateMesh(maxh=0.1)
assert len(m.Elements2D()) > 0
ngs = pytest.importorskip("ngsolve")
geo = g.GenerateSplineGeometry()
ngs.Draw(geo)
mesh = ngs.Mesh(m)
mesh.Curve(3)
ngs.Draw(mesh)
if __name__ == "__main__":
test_two_circles()
test_two_edge()
test_trig_and_circle()