def test_BC_flags(self):
"""
Testing the flags of shapes and their in their domain
"""
nb_shapes = 3
domain2D = create_domain2D()
domain3D = create_domain3D()
domain2DRANS = create_domain2D()
domain3DRANS = create_domain3D()
flags_v2D = []
flags_s2D = []
flags_v3D = []
flags_f3D = []
flags_v2DRANS = []
flags_s2DRANS = []
flags_v3DRANS = []
flags_f3DRANS = []
maxf = 0
for i in range(nb_shapes):
# 2D
a = create_custom2D(domain2D)
if flags_v2D:
maxf = np.max([np.max(flags_v2D), np.max(flags_s2D)])
flags_v2D += (a.vertexFlags+maxf).tolist()
flags_s2D += (a.segmentFlags+maxf).tolist()
# 3D
a = create_custom3D(domain3D)
if flags_v3D:
maxf = np.max([np.max(flags_v3D), np.max(flags_f3D)])
flags_v3D += (a.vertexFlags+maxf).tolist()
flags_f3D += (a.facetFlags+maxf).tolist()
# 2D RANS
a = create_custom2D(domain2DRANS, folder='mprans')
if flags_v2DRANS:
maxf = np.max([np.max(flags_v2DRANS), np.max(flags_s2DRANS)])
flags_v2DRANS += (a.vertexFlags+maxf).tolist()
flags_s2DRANS += (a.segmentFlags+maxf).tolist()
# 3D RANS
a = create_custom3D(domain3DRANS, folder='mprans')
if flags_v3DRANS:
maxf = np.max([np.max(flags_v3DRANS), np.max(flags_f3DRANS)])
flags_v3DRANS += (a.vertexFlags+maxf).tolist()
flags_f3DRANS += (a.facetFlags+maxf).tolist()
assembleDomain(domain2D)
assembleDomain(domain3D)
assembleDomainRANS(domain2DRANS)
assembleDomainRANS(domain3DRANS)
# test flags
npt.assert_equal(domain2D.vertexFlags, flags_v2D)
npt.assert_equal(domain2D.segmentFlags, flags_s2D)
npt.assert_equal(domain3D.vertexFlags, flags_v3D)
npt.assert_equal(domain3D.facetFlags, flags_f3D)
npt.assert_equal(domain2DRANS.vertexFlags, flags_v2DRANS)
npt.assert_equal(domain2DRANS.segmentFlags, flags_s2DRANS)
npt.assert_equal(domain3DRANS.vertexFlags, flags_v3DRANS)
npt.assert_equal(domain3DRANS.facetFlags, flags_f3DRANS)
def test_BC_flags(self):
"""
Testing the flags of shapes and their in their domain
"""
nb_shapes = 3
domain2D = create_domain2D()
domain3D = create_domain3D()
domain2DRANS = create_domain2D()
domain3DRANS = create_domain3D()
flags_v2D = []
flags_s2D = []
flags_v3D = []
flags_f3D = []
flags_v2DRANS = []
flags_s2DRANS = []
flags_v3DRANS = []
flags_f3DRANS = []
maxf = 0
for i in range(nb_shapes):
# 2D
a = create_custom2D(domain2D)
if flags_v2D:
maxf = np.max([np.max(flags_v2D), np.max(flags_s2D)])
flags_v2D += (a.vertexFlags+maxf).tolist()
flags_s2D += (a.segmentFlags+maxf).tolist()
# 3D
a = create_custom3D(domain3D)
if flags_v3D:
maxf = np.max([np.max(flags_v3D), np.max(flags_f3D)])
flags_v3D += (a.vertexFlags+maxf).tolist()
flags_f3D += (a.facetFlags+maxf).tolist()
# 2D RANS
a = create_custom2D(domain2DRANS, folder='mprans')
if flags_v2DRANS:
maxf = np.max([np.max(flags_v2DRANS), np.max(flags_s2DRANS)])
flags_v2DRANS += (a.vertexFlags+maxf).tolist()
flags_s2DRANS += (a.segmentFlags+maxf).tolist()
# 3D RANS
a = create_custom3D(domain3DRANS, folder='mprans')
if flags_v3DRANS:
maxf = np.max([np.max(flags_v3DRANS), np.max(flags_f3DRANS)])
flags_v3DRANS += (a.vertexFlags+maxf).tolist()
flags_f3DRANS += (a.facetFlags+maxf).tolist()
assembleDomain(domain2D)
assembleDomain(domain3D)
assembleDomainRANS(domain2DRANS)
assembleDomainRANS(domain3DRANS)
# test flags
npt.assert_equal(domain2D.vertexFlags, flags_v2D)
npt.assert_equal(domain2D.segmentFlags, flags_s2D)
npt.assert_equal(domain3D.vertexFlags, flags_v3D)
npt.assert_equal(domain3D.facetFlags, flags_f3D)
npt.assert_equal(domain2DRANS.vertexFlags, flags_v2DRANS)
npt.assert_equal(domain2DRANS.segmentFlags, flags_s2DRANS)
npt.assert_equal(domain3DRANS.vertexFlags, flags_v3DRANS)
npt.assert_equal(domain3DRANS.facetFlags, flags_f3DRANS)
def test_assemble_domain(self):
"""
Testing the assembleDomain() for different domains with multiple shapes
"""
nb_shapes = 10
domain2D = create_domain2D()
domain2DRANS = create_domain2D()
domain3D = create_domain3D()
domain3DRANS = create_domain3D()
dim2D = [1., 1.]
dim3D = [1., 1., 1.]
coords2D = np.array([0.5, 0.5])
coords3D = np.array([0.5, 0.5, 0.5])
nb_bc2D = 0
nb_bc2DRANS = 0
nb_bc3D = 0
nb_bc3DRANS = 0
for shape in range(nb_shapes):
coords2D += 1.5
a = create_rectangle(domain2D, dim=dim2D, coords=coords2D)
nb_bc2D += len(a.BC_list)
a = create_cuboid(domain3D, dim=dim3D, coords=coords3D)
nb_bc3D += len(a.BC_list)
a = create_rectangle(domain2DRANS,
dim=dim2D,
coords=coords2D,
folder='mprans')
nb_bc2DRANS += len(a.BC_list)
a = create_cuboid(domain3DRANS,
dim=dim3D,
coords=coords3D,
folder='mprans')
nb_bc3DRANS += len(a.BC_list)
a = create_tank2D(domain2DRANS, dim=[50., 50.], coords=[25., 25.])
nb_bc2DRANS += len(a.BC_list)
a = create_tank3D(domain3DRANS,
dim=[50., 50., 50.],
coords=[25., 25., 25.])
nb_bc3DRANS += len(a.BC_list)
assembleDomain(domain2D)
assembleDomain(domain3D)
assembleDomainRANS(domain2DRANS)
assembleDomainRANS(domain3DRANS)
# check that each domain has the right number of shapes
npt.assert_equal(len(domain2D.shape_list), nb_shapes)
npt.assert_equal(len(domain3D.shape_list), nb_shapes)
npt.assert_equal(len(domain2DRANS.shape_list), nb_shapes + 1)
npt.assert_equal(len(domain3DRANS.shape_list), nb_shapes + 1)
# check that the number of boundary conditions is right
npt.assert_equal(len(domain2D.bc), nb_bc2D + 1)
npt.assert_equal(len(domain3D.bc), nb_bc3D + 1)
npt.assert_equal(len(domain2DRANS.bc), nb_bc2DRANS + 1)
npt.assert_equal(len(domain3DRANS.bc), nb_bc3DRANS + 1)
def test_assemble_domain(self):
"""
Testing the assembleDomain() for different domains with multiple shapes
"""
nb_shapes = 10
domain2D = create_domain2D()
domain2DRANS = create_domain2D()
domain3D = create_domain3D()
domain3DRANS = create_domain3D()
dim2D = [1., 1.]
dim3D = [1., 1., 1.]
coords2D = np.array([0.5, 0.5])
coords3D = np.array([0.5, 0.5, 0.5])
nb_bc2D = 0
nb_bc2DRANS = 0
nb_bc3D = 0
nb_bc3DRANS = 0
for shape in range(nb_shapes):
coords2D += 1.5
a = create_rectangle(domain2D, dim=dim2D, coords=coords2D)
nb_bc2D += len(a.BC_list)
a = create_cuboid(domain3D, dim=dim3D, coords=coords3D)
nb_bc3D += len(a.BC_list)
a = create_rectangle(domain2DRANS, dim=dim2D, coords=coords2D,
folder='mprans')
nb_bc2DRANS += len(a.BC_list)
a = create_cuboid(domain3DRANS, dim=dim3D, coords=coords3D,
folder='mprans')
nb_bc3DRANS += len(a.BC_list)
a = create_tank2D(domain2DRANS, dim=[50., 50.], coords=[25., 25.])
nb_bc2DRANS += len(a.BC_list)
a = create_tank3D(domain3DRANS, dim=[50., 50., 50.],
coords=[25., 25., 25.])
nb_bc3DRANS += len(a.BC_list)
assembleDomain(domain2D)
assembleDomain(domain3D)
assembleDomainRANS(domain2DRANS)
assembleDomainRANS(domain3DRANS)
# check that each domain has the right number of shapes
npt.assert_equal(len(domain2D.shape_list), nb_shapes)
npt.assert_equal(len(domain3D.shape_list), nb_shapes)
npt.assert_equal(len(domain2DRANS.shape_list), nb_shapes+1)
npt.assert_equal(len(domain3DRANS.shape_list), nb_shapes+1)
# check that the number of boundary conditions is right
npt.assert_equal(len(domain2D.bc), nb_bc2D+1)
npt.assert_equal(len(domain3D.bc), nb_bc3D+1)
npt.assert_equal(len(domain2DRANS.bc), nb_bc2DRANS+1)
npt.assert_equal(len(domain3DRANS.bc), nb_bc3DRANS+1)
def test_BC_flags(self):
"""
Testing the flags of shapes and their in their domain
"""
nb_shapes = 3
domain2D = create_domain2D()
domain3D = create_domain3D()
domain2DRANS = create_domain2D()
domain3DRANS = create_domain3D()
domainSTL = create_domain3D()
flags_v2D = []
flags_s2D = []
flags_v3D = []
flags_f3D = []
flags_v2DRANS = []
flags_s2DRANS = []
flags_v3DRANS = []
flags_f3DRANS = []
maxf = 0
for i in range(nb_shapes):
# 2D
a = create_custom2D(domain2D)
if flags_v2D:
maxf = np.max([np.max(flags_v2D), np.max(flags_s2D)])
flags_v2D += (a.vertexFlags + maxf).tolist()
flags_s2D += (a.segmentFlags + maxf).tolist()
# 3D
a = create_custom3D(domain3D)
if flags_v3D:
maxf = np.max([np.max(flags_v3D), np.max(flags_f3D)])
flags_v3D += (a.vertexFlags + maxf).tolist()
flags_f3D += (a.facetFlags + maxf).tolist()
# 2D RANS
a = create_custom2D(domain2DRANS, folder='mprans')
if flags_v2DRANS:
maxf = np.max([np.max(flags_v2DRANS), np.max(flags_s2DRANS)])
flags_v2DRANS += (a.vertexFlags + maxf).tolist()
flags_s2DRANS += (a.segmentFlags + maxf).tolist()
# 3D RANS
a = create_custom3D(domain3DRANS, folder='mprans')
if flags_v3DRANS:
maxf = np.max([np.max(flags_v3DRANS), np.max(flags_f3DRANS)])
flags_v3DRANS += (a.vertexFlags + maxf).tolist()
flags_f3DRANS += (a.facetFlags + maxf).tolist()
assembleDomain(domain2D)
assembleDomain(domain3D)
assembleDomainRANS(domain2DRANS)
assembleDomainRANS(domain3DRANS)
# test flags
npt.assert_equal(domain2D.vertexFlags, flags_v2D)
npt.assert_equal(domain2D.segmentFlags, flags_s2D)
npt.assert_equal(domain3D.vertexFlags, flags_v3D)
npt.assert_equal(domain3D.facetFlags, flags_f3D)
npt.assert_equal(domain2DRANS.vertexFlags, flags_v2DRANS)
npt.assert_equal(domain2DRANS.segmentFlags, flags_s2DRANS)
npt.assert_equal(domain3DRANS.vertexFlags, flags_v3DRANS)
npt.assert_equal(domain3DRANS.facetFlags, flags_f3DRANS)
#stl flags
path1 = getpath()
stl = create_stl3D(domainSTL, os.path.join(path1, "STLBlocks.stl"))
assembleDomainRANS(domainSTL)
STLnames = ["Bed0", "Concrete0", "Inlet0", "Outlet0", "Top0", "Wall0"]
nSTLs = len(STLnames)
j = 0
print
for key, value in stl.boundaryTags.items():
self.assertTrue(STLnames[value - 1] == key)
for key, value in domainSTL.boundaryTags.items():
self.assertTrue(stl.name + '_' + STLnames[value - 1] == key)
npt.assert_equal(nSTLs, len(stl.boundaryTags))
def test_assemble_domain(self):
"""
Testing the assembleDomain() for different domains with multiple shapes
"""
nb_shapes = 10
domain2D = create_domain2D()
domain2DRANS = create_domain2D()
domain3D = create_domain3D()
domain3DRANS = create_domain3D()
domainSTL = create_domain3D()
dim2D = np.array([1., 1.])
dim3D = np.array([1., 1., 1.])
coords2D = np.array([0.5, 0.5])
coords3D = np.array([0.5, 0.5, 0.5])
nb_bc2D = 0
nb_bc2DRANS = 0
nb_bc3D = 0
nb_bc3DRANS = 0
for shape in range(nb_shapes):
coords2D += 1.5
coords3D += 1.5
a = create_rectangle(domain2D, dim=dim2D, coords=coords2D)
nb_bc2D += len(a.BC_list)
a = create_cuboid(domain3D, dim=dim3D, coords=coords3D)
nb_bc3D += len(a.BC_list)
a = create_rectangle(domain2DRANS,
dim=dim2D,
coords=coords2D,
folder='mprans')
nb_bc2DRANS += len(a.BC_list)
a = create_cuboid(domain3DRANS,
dim=dim3D,
coords=coords3D,
folder='mprans')
nb_bc3DRANS += len(a.BC_list)
a = create_tank2D(domain2DRANS, dim=[50., 50.])
nb_bc2DRANS += len(a.BC_list)
a = create_tank3D(domain3DRANS, dim=[50., 50., 50.])
nb_bc3DRANS += len(a.BC_list)
assembleDomain(domain2D)
assembleDomain(domain3D)
assembleDomainRANS(domain2DRANS)
assembleDomainRANS(domain3DRANS)
x2D = domain2D.x
x3D = domain3D.x
x2DRANS = domain2DRANS.x
x3DRANS = domain3DRANS.x
L2D = domain2D.L
L3D = domain3D.L
L2DRANS = domain2DRANS.L
L3DRANS = domain3DRANS.L
# check that each domain has the right number of shapes
npt.assert_equal(len(domain2D.shape_list), nb_shapes)
npt.assert_equal(len(domain3D.shape_list), nb_shapes)
npt.assert_equal(len(domain2DRANS.shape_list), nb_shapes + 1)
npt.assert_equal(len(domain3DRANS.shape_list), nb_shapes + 1)
# check that the number of boundary conditions is right
npt.assert_equal(len(domain2D.bc), nb_bc2D + 1)
npt.assert_equal(len(domain3D.bc), nb_bc3D + 1)
npt.assert_equal(len(domain2DRANS.bc), nb_bc2DRANS + 1)
npt.assert_equal(len(domain3DRANS.bc), nb_bc3DRANS + 1)
# check that bounding boxes are rightly calculated
npt.assert_equal(L2D, [14.5, 14.5])
npt.assert_equal(L3D, [14.5, 14.5, 14.5])
npt.assert_equal(L2DRANS, [50., 50.])
npt.assert_equal(L3DRANS, [50., 50., 50.])
npt.assert_equal(x2D, [1.5, 1.5])
npt.assert_equal(x3D, [1.5, 1.5, 1.5])
npt.assert_equal(x2DRANS, [0., 0.])
npt.assert_equal(x3DRANS, [0., 0., 0.])
#stl domain checks
path1 = getpath()
stl = create_stl3D(domainSTL, os.path.join(path1, "STLBlocks.stl"))
assembleDomainRANS(domainSTL)
STLnames = ["Bed0", "Concrete0", "Inlet0", "Outlet0", "Top0", "Wall0"]
nSTLs = len(STLnames)
npt.assert_equal(nSTLs, max(stl.vertexFlags))
npt.assert_equal(nSTLs, max(stl.facetFlags))
npt.assert_equal(nSTLs, max(domainSTL.vertexFlags))
npt.assert_equal(nSTLs, max(domainSTL.facetFlags))
