Context Navigation

source: inundation/ga/storm_surge/pmesh/meshHarness.py @ 643

Last change on this file since 643 was 431, checked in by duncan, 20 years ago
increased functionality of ungenerate
File size: 34.3 KB

Rev	Line
[349]	1	#!/usr/bin/env python
	2	#
	3	import tempfile
	4	import unittest
	5	from mesh import *
	6	from load_mesh.loadASCII import *
	7
	8	class meshTestCase(unittest.TestCase):
	9	def setUp(self):
	10	pass
	11
	12
	13
	14	def tearDown(self):
	15	pass
	16
	17	def testPointDistance(self):
	18	a = Point(0.0, 0.0)
	19	b = Point(0.0, 10.0)
	20
	21	self.failUnless( a.DistanceToPoint(b) == 10.0,
	22	'Point DistanceToPoint is wrong!')
	23
	24	def testVertexDistance(self):
	25	a = Vertex (0.0, 0.0)
	26	b = Vertex (0.0, 10.0)
	27
	28	self.failUnless( a.DistanceToPoint(b) == 10.0,
	29	'Point DistanceToPoint is wrong!')
	30
	31	def testTriangle(self):
	32	a = Vertex (0.0, 0.0)
	33	b = Vertex (0.0, 2.0)
	34	c = Vertex (2.0,0.0)
	35	d = Vertex (0.0, 4.0)
	36	e = Vertex (2.0, 2.0)
	37	f = Vertex (4.0,0.0)
	38
	39	t1 = Triangle(b,a,c)
	40	t2 = Triangle(b,c,e)
	41	t3 = Triangle(e,c,f)
	42	t4 = Triangle(d,b,e)
	43	t2.setNeighbors(t3,t4,t1)
	44
	45	self.failUnless( t2.neighbors[2].vertices[0] == b, 'Triangle initialisation is wrong!')
	46
	47
	48	def testSegment(self):
	49	a = Vertex (0.0, 0.0)
	50	b = Vertex (0.0, 10.0)
	51	s = Segment(a,b, marker = 20)
	52
	53	self.failUnless( s.vertices[0].DistanceToPoint(s.vertices[1]) == 10.0,
	54	'vertices in a segment are wrong')
	55
	56	self.failUnless( s.marker == 20.0,
	57	'marker in a segment are wrong')
	58
	59	def testdeleteUserVertex(self):
	60
	61
	62	mesh = Mesh()
	63	a = mesh.addUserVertex(0.0, 0.0)
	64	b = mesh.addUserVertex (0.0, 2.0)
	65	c = mesh.addUserVertex (2.0,0.0)
	66
	67	s1 = mesh.addUserSegment(a,b)
	68	s2 = mesh.addUserSegment(a,c)
	69	s3 = mesh.addUserSegment(c,b)
	70
	71	mesh.deleteMeshObject (a)
	72	self.failUnless(mesh.userSegments[0] == s3,
	73	'Bad segment. ')
	74	self.failUnless(len(mesh.userSegments) ==1,
	75	'Segments not deleted.')
	76	self.failUnless(len(mesh.userVertices) == 2,
	77	'Vertex not deleted.')
	78
	79
	80
	81	def testgenerateMesh(self):
	82	a = Vertex (0.0, 0.0)
	83	d = Vertex (0.0, 4.0)
	84	f = Vertex (4.0,0.0)
	85
	86	s1 = Segment(a,d)
	87	s2 = Segment(d,f)
	88	s3 = Segment(a,f)
	89
	90	r1 = Region(0.3, 0.3,tag = 1.3,maxArea = .6)
	91	#print r1
	92	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3], regions=[r1] )
	93
	94	m.generateMesh("QApz", maxArea = 2.1 )
	95
	96	#print m
	97
	98	#m.plotMeshTriangle()
	99
	100	result = 1.414214
	101	delta = 0.00001
	102
	103	self.failUnless((m.meshTriangles[1].vertices[0].x < result + delta) or
	104	(m.meshTriangles[1].vertices[0].x > result - delta),
	105	'generated mesh is wrong!')
	106
	107	def test_regionalMaxArea(self):
	108	v0 = Vertex (0.0, 0.0)
	109	v1 = Vertex (6.0, 0.0)
	110	v2 = Vertex (6.0,6.0)
	111	v3 = Vertex (0.0,6.0)
	112
	113	s1 = Segment(v0,v1)
	114	s2 = Segment(v1,v2)
	115	s3 = Segment(v3,v2)
	116	s4 = Segment(v3,v0)
	117	s5 = Segment(v2,v0)
	118
	119	r1 = Region(3, 1,tag = 1.3)
	120	#print r1
	121	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5], regions=[r1] )
	122
	123	m.generateMesh("Apz", maxArea = 36 )
	124
	125	#m.plotMeshTriangle()
	126	#print "len(m.meshTriangles)",len(m.meshTriangles)
	127
	128	self.failUnless(len(m.meshTriangles) == 2,
	129	'test_regionalMaxArea 1:generated mesh is wrong!')
	130
	131	## Another test case
	132	r1 = Region(3, 1,tag = 1.3)
	133	r2 = Region(1, 3,tag = 1.3, maxArea = 8)
	134	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	135	regions=[r1,r2] )
	136	m.generateMesh("Apz", maxArea = 36 )
	137	self.failUnless(len(m.meshTriangles) == 6,
	138	'test_regionalMaxArea 2:generated mesh is wrong!')
	139
	140	## Another test case
	141	r1 = Region(3, 1, tag = 1.3, maxArea = 8)
	142	r2 = Region(1, 3, tag = 1.3, maxArea = 8)
	143	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	144	regions=[r1,r2] )
	145	m.generateMesh("Apz", maxArea = 36 )
	146	print "len(m.meshTriangles)",len(m.meshTriangles)
	147	self.failUnless(len(m.meshTriangles) == 8,
	148	'test_regionalMaxArea 3:generated mesh is wrong!')
	149
	150	## Another test case
	151	r1 = Region(3, 1, tag = 1.3 )
	152	r2 = Region(1, 3, tag = 1.3, maxArea = 8)
	153	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	154	regions=[r1,r2] )
	155	m.generateMesh("Apz", maxArea = 8 )
	156	self.failUnless(len(m.meshTriangles) == 8,
	157	'test_regionalMaxArea 4:generated mesh is wrong!')
	158
	159	## Another test case
	160	r1 = Region(3, 1,tag = 1.3, maxArea = 8)
	161	r2 = Region(1, 3,tag = 1.3, maxArea = 8)
	162	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	163	regions=[r1,r2] )
	164	m.generateMesh("Apz", maxArea = 36,isRegionalMaxAreas = False )
	165	self.failUnless(len(m.meshTriangles) == 2,
	166	'test_regionalMaxArea 5:generated mesh is wrong!')
	167
	168	def testdeleteUserVertex(self):
	169	mesh = Mesh()
	170	a = mesh.addUserVertex(0.0, 0.0)
	171	b = mesh.addUserVertex (0.0, 2.0)
	172	c = mesh.addUserVertex (2.0,0.0)
	173
	174	s1 = mesh.addUserSegment(a,b)
	175	s2 = mesh.addUserSegment(a,c)
	176	s3 = mesh.addUserSegment(c,b)
	177
	178	mesh.deleteMeshObject (s2)
	179
	180	#print ",s2 in mesh.userSegments" ,s2 in mesh.userSegments
	181	self.failUnless(not(s2 in mesh.userSegments),
	182	'Bad segment. ')
	183	self.failUnless(len(mesh.userSegments) ==2,
	184	'Segments not deleted.')
	185	self.failUnless(len(mesh.userVertices) == 3,
	186	'Vertex deleted, instead of segment.')
	187
	188	def testTriangleArea(self):
	189	a = Vertex (10.0, 10.0)
	190	b = Vertex (10.0, 20.0)
	191	c = Vertex (20.0,10.0)
	192
	193	d = Vertex (-20.0, 0.0)
	194	e = Vertex (-20.0, -20.0)
	195	f = Vertex (20.0,-20.0)
	196
	197	t1 = Triangle(b,a,c)
	198	t2 = Triangle(e,d,f)
	199
	200	# print "t1", t1
	201	# print "t1 area ", t1.calcArea()
	202	# print "t2", t2
	203	# print "t2 area ", t2.calcArea()
	204	self.failUnless( t1.calcArea() == 50 and t2.calcArea() == 400, 'Triangle area is wrong!')
	205	def testisUserSegmentNew (self):
	206	mesh = Mesh()
	207	a = mesh.addUserVertex(0.0, 0.0)
	208	b = mesh.addUserVertex (0.0, 2.0)
	209	c = mesh.addUserVertex (2.0,0.0)
	210	d = mesh.addUserVertex (2.0,3.0)
	211
	212	s1 = mesh.addUserSegment(a,b)
	213	s2 = mesh.addUserSegment(a,c)
	214	s3 = mesh.addUserSegment(c,b)
	215
	216	self.failUnless(mesh.isUserSegmentNew(a,d) ,
	217	'Segment should be new. ')
	218	self.failUnless(not(mesh.isUserSegmentNew(a,b)) ,
	219	'Segment should not be new. ')
	220
	221
	222	def testautoSegment(self):
	223	a = Vertex (0.0, 0.0)
	224	d = Vertex (0.0, 4.0)
	225	f = Vertex (4.0,4.0)
	226	g = Vertex (4.0,0.0)
	227
	228	s1 = Segment(a,d)
	229	#s2 = Segment(a,f)
	230	#s3 = Segment(f,d)
	231
	232	m = Mesh(userVertices=[a,d,f,g], userSegments=[s1] )
	233
	234	#print m
	235
	236	m.autoSegment()
	237
	238	#print m
	239
	240	#print 'Len', len(m.userSegments)
	241
	242	self.failUnless(len(m.userSegments) == 4 ,
	243	'userSegments is wrong!')
	244
	245	def testRegions(self):
	246	a = Vertex (0.0, 0.0)
	247	b = Vertex (0.0, 2.0)
	248	c = Vertex (2.0,0.0)
	249	d = Vertex (0.0, 4.0)
	250	e = Vertex (2.0, 2.0)
	251	f = Vertex (4.0,0.0)
	252	g = Vertex (0.0,-2.0)
	253
	254	Segment.set_default_tag("")
	255	s1 = Segment(a,b)
	256	s2 = Segment(b,e)
	257	s3 = Segment(e,c)
	258	s4 = Segment(c,a)
	259	s5 = Segment(b,d)
	260	s6 = Segment(e,d)
	261	s7 = Segment(e,f)
	262	s8 = Segment(c,f)
	263	s9 = Segment(g,c)
	264	s10 = Segment(g,a)
	265
	266	r1 = Region(0.1,0.1,tag="22")
	267	r2 = Region(0.1,2.1,tag="11")
	268	r3 = Region(2.1,0.1)
	269
	270	m = Mesh(userVertices=[a,b,c,d,e,f,g], userSegments=[s1,s2,s3,s4,s5,s6,s7,s8,s9,s10], regions=[r1,r2,r3] )
	271	m.generateMesh("QApz", maxArea = 2.1 )
	272	#print m
	273	Triangulation = m.getTriangulation()
	274	#print Triangulation[0].attribute
	275	#print Triangulation[1].attribute
	276	#print Triangulation[2].attribute
	277	#print Triangulation[3].attribute
	278	#print Triangulation[4].attribute
	279
	280	self.failUnless(Triangulation[0].attribute == "" and
	281	Triangulation[1].attribute == "22" and
	282	Triangulation[2].attribute == "" and
	283	Triangulation[3].attribute == "11" and
	284	Triangulation[4].attribute == "22" ,
	285	'region attributes are wrong!')
	286
	287	def test_vertexAttribs(self):
	288	a = Vertex (0.0, 0.0, attributes = [12.0,2.0])
	289	d = Vertex (0.0, 4.0, attributes = [9.0,7.0])
	290	f = Vertex (4.0,0.0, attributes = [14.0,3.0])
	291
	292	Segment.set_default_tag("")
	293	s1 = Segment(a,d)
	294	s2 = Segment(d,f)
	295	s3 = Segment(a,f)
	296
	297	r1 = Region(0.3, 0.3, tag = 88.9)
	298
	299	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3], regions=[r1])
	300
	301	m.generateMesh("QApz", maxArea = 2.1)
	302
	303	vert = m.getMeshVertices()
	304
	305	self.failUnless(vert[0].attributes == [12.0, 2.0] and
	306	vert[1].attributes == [9.0, 7.0] and
	307	vert[2].attributes == [14.0,3.0] and
	308	vert[3].attributes == [12.232233047033631, 4.4142135623730949] and
	309	vert[4].attributes == [13.0, 2.5] ,
	310	'vertex attributes are wrong!')
	311
	312
	313	def test_vertexAttribs2(self):
	314
	315	a = Vertex (0.0, 0.0)
	316	d = Vertex (0.0, 4.0)
	317	f = Vertex (4.0,0.0)
	318
	319	Segment.set_default_tag("")
	320	s1 = Segment(a,d)
	321	s2 = Segment(d,f)
	322	s3 = Segment(a,f)
	323
	324	r1 = Region(0.3, 0.3, tag = 88.9)
	325
	326	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3], regions=[r1])
	327
	328	m.generateMesh("QApz", maxArea = 2.1 )
	329
	330	vert = m.getMeshVertices()
	331	self.failUnless(vert[0].attributes == [] and
	332	vert[1].attributes == [] and
	333	vert[2].attributes == [] and
	334	vert[3].attributes == [] and
	335	vert[4].attributes == [],
	336	'vertex attributes are wrong!')
	337
	338	def test_segmarker(self):
	339
	340	a = Vertex (0.0, 0.0)
	341	d = Vertex (0.0, 4.0)
	342	f = Vertex (4.0,0.0)
	343
	344	s1 = Segment(a,d,marker = 5)
	345	s2 = Segment(d,f,marker = 7)
	346	s3 = Segment(a,f,marker = 9)
	347
	348	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3])
	349
	350	m.generateMesh("QApz", maxArea = 2.1 )
	351
	352	seg = m.getMeshSegments()
	353
	354	#print "seg[0].marker"
	355	#print seg[0].marker
	356	#print "seg[0].marker"
	357	self.failUnless(seg[0].marker == 5 and
	358	seg[1].marker == 7 and
	359	seg[2].marker == 9 and
	360	seg[3].marker == 7 and
	361	seg[4].marker == 9 and
	362	seg[5].marker == 7 and
	363	seg[6].marker == 5,
	364	'seg markers are wrong')
	365
	366	def test_segmarker2(self):
	367
	368	a = Vertex (0.0, 0.0)
	369	d = Vertex (0.0, 4.0)
	370	f = Vertex (4.0,0.0)
	371	e = Vertex (1.0,1.0)
	372
	373	s1 = Segment(a,d)
	374	s2 = Segment(d,f)
	375	s3 = Segment(a,f)
	376	s4 = Segment(a,e)
	377
	378	m = Mesh(userVertices=[a,d,f,e], userSegments=[s1,s2,s3,s4])
	379
	380	m.generateMesh("QApz", maxArea = 2.1)
	381
	382	seg = m.getMeshSegments()
	383	self.failUnless(seg[0].marker == "external" and
	384	seg[1].marker == "external" and
	385	seg[2].marker == "external" and
	386	seg[3].marker == "internal" and
	387	seg[4].marker == "external",
	388	'2nd seg markers are wrong')
	389
	390	def test_asciiFile(self):
	391
[364]	392	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	393	d = Vertex (0.0, 4.0) #, attributes = [1.2])
	394	f = Vertex (4.0,0.0) #, attributes = [1.3])
	395	e = Vertex (1.0,1.0) #, attributes = [1.4])
[349]	396
	397	s1 = Segment(a,d)
	398	s2 = Segment(d,f)
	399	s3 = Segment(a,f)
	400	s4 = Segment(a,e)
	401
	402	m = Mesh(userVertices=[a,d,f,e], userSegments=[s1,s2,s3,s4])
	403
	404	m.generateMesh("QApz", maxArea = 2.1 )
	405
	406	seg = m.getMeshSegments()
	407
	408	fileName = tempfile.mktemp(".txt")
	409	m.exportASCIItrianglulationfile(fileName)
	410	file = open(fileName)
	411	lFile = file.read().split('\n')
	412	file.close()
	413	os.remove(fileName)
	414
	415	#print "@^@^"
	416	#for l in lFile:
	417	# print l,"<"
	418	#print "@^@^"
[364]	419
	420	self.failUnless(lFile[0] == "5 0 # <vertex #> <x> <y> [attributes] ...Triangulation Vertices..."
[349]	421	,
	422	'Ascii file is wrong, vertex title')
[364]	423	self.failUnless(lFile[1] == "0 0.0 0.0 " and #1.1 " and
	424	lFile[2] == "1 0.0 4.0 " and #1.2 " and
	425	lFile[3] == "2 4.0 0.0 " and #1.3 " and
	426	lFile[4] == "3 1.0 1.0 " and #1.4 " and
	427	lFile[5] == "4 2.0 2.0 " #1.25 "
[349]	428	,
	429	'Ascii file is wrong, vertex')
[364]	430
	431	self.failUnless(lFile[6] == "# attribute column titles ...Triangulation Vertex Titles..."
[349]	432	,
[364]	433	'Ascii file is wrong, attribute column title')
	434	self.failUnless(lFile[7] == "4 # <triangle #> [<vertex #>] [<neigbouring triangle #>] [attribute of region] ...Triangulation Triangles..." and
	435	lFile[8] == "0 3 2 4 -1 2 3 " and
	436	lFile[9] == "1 1 0 3 3 2 -1 " and
	437	lFile[10] == "2 3 4 1 -1 1 0 " and
	438	lFile[11] == "3 2 3 0 1 -1 0 "
	439	,
[349]	440	'Ascii file is wrong, triangle')
	441
[364]	442	self.failUnless(lFile[12] == "5 # <segment #> <vertex #> <vertex #> [boundary marker] ...Triangulation Segments..." and
	443	lFile[13] == "0 0 1 external" and
	444	lFile[14] == "1 1 4 external" and
	445	lFile[15] == "2 2 0 external" and
	446	lFile[16] == "3 0 3 internal" and
	447	lFile[17] == "4 4 2 external" ,
[349]	448	'Ascii file is wrong, segment')
	449
[364]	450	self.failUnless(lFile[18] == '4 0 # <vertex #> <x> <y> [attributes] ...Mesh Vertices...',
[349]	451	'Ascii file is wrong, Mesh Vertices Title')
	452
[364]	453	self.failUnless(lFile[19] == '0 0.0 0.0 ' and #1.1 ' and
	454	lFile[20] == '1 0.0 4.0 ' and #1.2 ' and
	455	lFile[21] == '2 4.0 0.0 ' and #1.3 ' and
	456	lFile[22] == '3 1.0 1.0 ' #1.4 '
	457	,
[349]	458	'Ascii file is wrong, Mesh Vertices II')
	459
[364]	460	self.failUnless(lFile[23] == '4 # <segment #> <vertex #> <vertex #> [boundary marker] ...Mesh Segments...' and
	461	lFile[24] == '0 0 1 ' and
	462	lFile[25] == '1 1 2 ' and
	463	lFile[26] == '2 0 2 ' and
	464	lFile[27] == '3 0 3 ' and
	465	lFile[28] == '0 # <Hole #> <x> <y> ...Mesh Holes...' and
	466	lFile[29] == '0 # <Region #> <x> <y> <tag>...Mesh Regions...'
[349]	467	,
	468	'Ascii file is wrong, Mesh Segments')
[364]	469
[349]	470
	471	def test_ascii_file(self):
	472
[364]	473	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	474	d = Vertex (0.0, 4.0) #, attributes = [1.2])
	475	f = Vertex (4.0,0.0) #, attributes = [1.3])
	476	e = Vertex (1.0,1.0) #, attributes = [1.4])
[349]	477
	478	s1 = Segment(a,d)
	479	s2 = Segment(d,f)
	480	s3 = Segment(a,f)
	481	s4 = Segment(a,e)
	482
	483	m = Mesh(userVertices=[a,d,f,e], userSegments=[s1,s2,s3,s4])
	484
	485	m.generateMesh("QApz", maxArea = 2.1 )
	486
	487	seg = m.getMeshSegments()
	488
	489	fileName = tempfile.mktemp(".txt")
[364]	490	m.exportASCIItrianglulationfile(fileName)
[349]	491	file = open(fileName)
	492	lFile = file.read().split('\n')
	493	file.close()
	494	os.remove(fileName)
	495
[364]	496	#print "@^@^"
	497	#for l in lFile:
	498	# print l,"<"
	499	#print "@^@^"
	500	self.failUnless(lFile[0] == "5 0 # <vertex #> <x> <y> [attributes] ...Triangulation Vertices..."
[349]	501	,
	502	'Ascii file is wrong, vertex title')
[364]	503	self.failUnless(lFile[1] == "0 0.0 0.0 " and #1.1 " and
	504	lFile[2] == "1 0.0 4.0 " and #1.2 " and
	505	lFile[3] == "2 4.0 0.0 " and #1.3 " and
	506	lFile[4] == "3 1.0 1.0 " and #1.4 " and
	507	lFile[5] == "4 2.0 2.0 " #1.25 "
[349]	508	,
	509	'Ascii file is wrong, vertex')
[364]	510
	511	self.failUnless(lFile[6] == "# attribute column titles ...Triangulation Vertex Titles..."
[349]	512	,
[364]	513	'Ascii file is wrong, attribute column title')
	514	self.failUnless(lFile[7] == "4 # <triangle #> [<vertex #>] [<neigbouring triangle #>] [attribute of region] ...Triangulation Triangles..." and
	515	lFile[8] == "0 3 2 4 -1 2 3 " and
	516	lFile[9] == "1 1 0 3 3 2 -1 " and
	517	lFile[10] == "2 3 4 1 -1 1 0 " and
	518	lFile[11] == "3 2 3 0 1 -1 0 "
	519	,
[349]	520	'Ascii file is wrong, triangle')
	521
[364]	522	self.failUnless(lFile[12] == "5 # <segment #> <vertex #> <vertex #> [boundary marker] ...Triangulation Segments..." and
	523	lFile[13] == "0 0 1 external" and
	524	lFile[14] == "1 1 4 external" and
	525	lFile[15] == "2 2 0 external" and
	526	lFile[16] == "3 0 3 internal" and
	527	lFile[17] == "4 4 2 external" ,
[349]	528	'Ascii file is wrong, segment')
	529
[364]	530	self.failUnless(lFile[18] == '4 0 # <vertex #> <x> <y> [attributes] ...Mesh Vertices...',
[349]	531	'Ascii file is wrong, Mesh Vertices Title')
	532
[364]	533	self.failUnless(lFile[19] == '0 0.0 0.0 ' and #1.1 ' and
	534	lFile[20] == '1 0.0 4.0 ' and #1.2 ' and
	535	lFile[21] == '2 4.0 0.0 ' and #1.3 ' and
	536	lFile[22] == '3 1.0 1.0 ' #1.4 '
	537	,
[349]	538	'Ascii file is wrong, Mesh Vertices II')
	539
[364]	540	self.failUnless(lFile[23] == '4 # <segment #> <vertex #> <vertex #> [boundary marker] ...Mesh Segments...' and
	541	lFile[24] == '0 0 1 ' and
	542	lFile[25] == '1 1 2 ' and
	543	lFile[26] == '2 0 2 ' and
	544	lFile[27] == '3 0 3 ' and
	545	lFile[28] == '0 # <Hole #> <x> <y> ...Mesh Holes...' and
	546	lFile[29] == '0 # <Region #> <x> <y> <tag>...Mesh Regions...'
[349]	547	,
	548	'Ascii file is wrong, Mesh Segments')
	549
	550
	551	def test_thinoutVertices(self):
	552
	553	v1 = Vertex(-20,-20)
	554	v2 = Vertex(-11,-11)
	555	v3 = Vertex(-10,-10)
	556	v4 = Vertex(-9,-1)
	557	v5 = Vertex(-8,2)
	558	v6 = Vertex(6,3)
	559	v7 = Vertex(12,9)
	560	v8 = Vertex(15,3)
	561	v9 = Vertex(24,3)
	562	m = Mesh(userVertices = [v1,v2,v3,v4,v5,v6,v7,v8,v9])
	563	m.thinoutVertices(10)
	564
	565	self.failUnless(v1 in m.userVertices,
	566	'test_thinoutVertices, test 1 failed')
	567	self.failUnless(v3 in m.userVertices,
	568	'test_thinoutVertices, test 2 failed')
	569	self.failUnless(v4 in m.userVertices,
	570	'test_thinoutVertices, test 3 failed')
	571	self.failUnless(v6 in m.userVertices,
	572	'test_thinoutVertices, test 4 failed')
	573	self.failUnless(v7 in m.userVertices,
	574	'test_thinoutVertices, test 5 failed')
	575	self.failUnless(v9 in m.userVertices,
	576	'test_thinoutVertices, test 6 failed')
	577	self.failUnless(v5 not in m.userVertices,
	578	'test_thinoutVertices, test 7 failed')
	579	self.failUnless(v2 not in m.userVertices,
	580	'test_thinoutVertices, test 8 failed')
	581	self.failUnless(v8 not in m.userVertices,
	582	'test_thinoutVertices, test 9 failed')
	583
	584	def test_same_x_y(self):
	585	v = Point(7,8)
	586	f = Point(7,8)
	587	f.same_x_y(v)
	588
	589	self.failUnless(f.same_x_y(v),
	590	'same_x_y True failed')
	591	e = Point(7,9)
	592	self.failUnless(not f.same_x_y(e),
	593	'same_x_y False failed')
	594
[431]	595	def test_import_mesh(self):
[349]	596
	597	a_att = [5.0,2.0]
	598	d_att =[4.0,2.0]
	599	f_att = [3.0,2.0]
	600	e_att = [2.0,2.0]
	601	a_xy = [0.0, 0.0]
[364]	602	a = Vertex ( a_xy[0],a_xy[1]) #, attributes =a_att)
	603	d = Vertex (0.0, 4.0) #, attributes =d_att)
	604	f = Vertex (4.0,0.0) #, attributes =f_att)
	605	e = Vertex (1.0,1.0) #, attributes =e_att)
[349]	606
	607	s1 = Segment(a,d, marker = "50")
	608	s2 = Segment(d,f, marker = "40")
	609	s3 = Segment(a,f, marker = "30")
	610	s4 = Segment(a,e, marker = "20")
	611
	612	r1 = Region(0.3, 0.3,tag = "1.3")
	613	m = Mesh(userVertices=[a,d,f,e],
	614	userSegments=[s1,s2,s3,s4],
	615	regions=[r1])
	616
	617	m.generateMesh("QApz", maxArea = 2.1)
	618	fileName = tempfile.mktemp(".tsh")
	619	#print "dgs!!!"
	620	#print "****************** fileName", fileName
	621	m.exportASCIItrianglulationfile(fileName)
[406]	622	#print "******************"
	623	#print "m", m
	624	#print "******************"
[349]	625	m_returned = importMeshFromFile(fileName)
[406]	626	#print "m_returned",m_returned
	627	#print "******************"
[349]	628	#print "****************** fileName", fileName
	629	os.remove(fileName)
	630	self.failUnless(0 == m.__cmp__(m_returned),
	631	'loading and saving of a mesh failed')
	632
	633
	634	def test_normaliseMesh(self):
	635
	636	a_att = [5.0,2.0]
	637	d_att =[4.0,2.0]
	638	f_att = [3.0,2.0]
	639	e_att = [2.0,2.0]
	640	a_xy = [10.0, 10.0]
	641	a = Vertex ( a_xy[0],a_xy[1], attributes =a_att)
	642	d = Vertex (15.0, 10.0, attributes =d_att)
	643	f = Vertex (10.0,20.0, attributes =f_att)
	644	e = Vertex (15.0,20.0, attributes =e_att)
	645
	646	s1 = Segment(a,d, marker = 50)
	647	s2 = Segment(d,e, marker = 40)
	648	s3 = Segment(e,f, marker = 30)
	649	s4 = Segment(f,a, marker = 20)
	650
	651	r1 = Region(0.3, 0.3,tag = 1.3)
	652	m = Mesh(userVertices=[a,d,f,e],
	653	userSegments=[s1,s2,s3,s4],
	654	regions=[r1])
	655	m.normaliseMesh(1,0,1)
	656	[xmin, ymin, xmax, ymax] = m.boxsize()
	657	[attmin, attmax] = m.maxMinVertAtt(0)
	658	self.failUnless(attmin == 0.0 and attmax == 1.0,
	659	'normalise failed')
	660	self.failUnless(xmin == 0.0 and ymin == 0.0 and xmax == 0.5 and ymax == 1.0,
	661	'normalise failed')
	662	m.normaliseMesh(200,-100,5)
	663	[xmin, ymin, xmax, ymax] = m.boxsize()
	664	[attmin, attmax] = m.maxMinVertAtt(0)
	665	self.failUnless(attmin == 0.0 and attmax == 5.0,
	666	'normalise failed')
	667	self.failUnless(xmin == -100.0 and ymin == -100.0 and xmax == 0.0 and ymax == 100.0,
	668	'normalise failed')
	669
	670	def test_exportASCIIsegmentoutlinefile(self):
	671	a = Vertex (0,0)
	672	b = Vertex (0,3)
	673	c = Vertex (3,3)
	674	d = Vertex (1,2)
	675	e = Vertex (3,1)
	676
	677	s1 = Segment(a,b, marker = "50")
	678	s2 = Segment(b,c, marker = "40")
	679	s3 = Segment(c,a, marker = "30")
	680
	681	r1 = Region(2, 1,tag = "1.3")
	682	h1 = Hole(1,4)
	683	m = Mesh(userVertices=[a,b,c,d,e],
	684	userSegments=[s1,s2,s3],
	685	regions=[r1],
	686	holes = [h1])
	687
	688	m.generateMesh("QApz", maxArea = 2.1)
	689	#print "mesh **************dsg", m
	690
	691	fileName = tempfile.mktemp(".tsh")
	692	m.exportASCIIsegmentoutlinefile(fileName)
	693
	694	m_returned = importMeshFromFile(fileName)
	695
	696	#print "m_returned ****",m_returned
	697	#print "****************** fileName", fileName
	698	os.remove(fileName)
	699
	700	#Trim mesh, so it should like like m_returned
	701	m.meshVertices = []
	702	m.meshTriangles = []
	703	m.meshSegments = []
	704	m.userVertices=[a,b,c]
	705	#print "(m.__cmp__(m_returned)", m.__cmp__(m_returned)
	706	self.failUnless(0 == m.__cmp__(m),
	707	'test_exportASCIIsegmentoutlinefile:loading and saving of a mesh failed')
	708	self.failUnless(0 == m.__cmp__(m_returned),
	709	'test_exportASCIIsegmentoutlinefile:loading and saving of a mesh failed')
	710
	711
	712	def test_exportxyafile(self):
	713	a = Vertex (0,0)
	714	b = Vertex (0,3)
	715	c = Vertex (3,3)
	716	d = Vertex (1,2)
	717	e = Vertex (3,1)
	718	f = Vertex (3,1)
	719
	720	s1 = Segment(a,b, marker = 50)
	721	s2 = Segment(b,c, marker = 40)
	722	s3 = Segment(c,a, marker = 30)
	723
	724	r1 = Region(2, 1,tag = 1.3)
	725	h1 = Hole(1,4)
	726	# Warning mesh can't produce this type of data structure its self
	727	m = Mesh(userVertices=[a,b,c,d,e],
	728	userSegments=[s1,s2,s3],
	729	regions=[r1],
	730	holes = [h1])
	731
	732	fileName = tempfile.mktemp(".txt")
	733	m.exportxyafile(fileName)
	734	file = open(fileName)
	735	lFile = file.read().split('\n')
	736	file.close()
[355]	737
	738	#print "**********************"
	739	#print lFile
	740	#print "**********************"
	741
[349]	742	os.remove(fileName)
	743	self.failUnless(lFile[0] == "5 0 # <vertex #> <x> <y> [attributes]" and
	744	lFile[1] == "0 0 " and
	745	lFile[2] == "0 3 " and
	746	lFile[3] == "3 3 " and
	747	lFile[4] == "1 2 " and
	748	lFile[5] == "3 1 "
	749	,
	750	'exported Ascii xya file is wrong')
	751
	752	m.generateMesh("QApz", maxArea = 2.1)
	753	fileName = tempfile.mktemp(".txt")
	754	m.exportxyafile(fileName)
	755	file = open(fileName)
	756	lFile = file.read().split('\n')
	757	file.close()
	758	os.remove(fileName)
	759	self.failUnless(lFile[0] == "6 0 # <vertex #> <x> <y> [attributes]" and
	760	lFile[1] == "0.0 0.0 " and
	761	lFile[2] == "0.0 3.0 " and
	762	lFile[3] == "3.0 3.0 " and
	763	lFile[4] == "1.0 2.0 " and
	764	lFile[5] == "3.0 1.0 " and
	765	lFile[6] == "1.5 1.5 "
	766	,
	767	'exported Ascii xya file is wrong')
	768
	769	def test_strings2ints(self):
	770	list = ["sea","river inlet","","sea","","moat"]
	771	preset = ["moat", "internal boundary"]
	772	[intlist, converter] = segment_strings2ints(list,preset )
	773	self.failUnless(intlist == [2,3 ,0 ,2 ,0 ,0 ]
	774	,
	775	'test_strings2ints produces bad intlist')
	776	self.failUnless(converter == ['moat', 'internal boundary',
	777	'sea', 'river inlet']
	778	,
	779	'test_strings2ints produces bad converter')
	780
	781	def test_ints2strings(self):
	782	list = ["internal boundary","sea","river inlet",
	783	"","sea","","moat","internal boundary"]
	784	outlist = ['internal boundary', 'sea', 'river inlet', 'moat',
	785	'sea', 'moat', 'moat', 'internal boundary']
	786	preset = ["moat", "internal boundary"]
	787	[intlist, converter] = segment_strings2ints(list,preset )
	788	newlist = segment_ints2strings(intlist, converter)
	789	self.failUnless(outlist == newlist
	790	,
	791	'test_strings2ints produces bad intlist')
	792	self.failUnless(converter == ['moat', 'internal boundary',
	793	'sea', 'river inlet']
	794	,
	795	'test_strings2ints produces bad converter')
	796
	797	def test_ints2strings2(self):
	798	list = ["","",""]
	799	preset = ["moat", "internal boundary"]
	800	[intlist, converter] = segment_strings2ints(list,preset )
	801	newlist = segment_ints2strings(intlist, converter)
	802	outlist = ['moat', 'moat', 'moat']
	803	self.failUnless(outlist == newlist
	804	,
	805	'test_strings2ints produces bad intlist')
	806	self.failUnless(converter == ['moat', 'internal boundary']
	807	,
	808	'test_strings2ints produces bad converter')
	809
	810
	811	def test_removeDuplicatedVertices(self):
	812	a = Vertex (0,0)
	813	a.index = 0
	814	b = Vertex (0,3)
	815	b.index = 1
	816	c = Vertex (3,3)
	817	c.index = 2
	818	d = Vertex (1,1)
	819	d.index = 3
	820	e = Vertex (3,1)
	821	e.index = 4
	822	f = Vertex (1,1)
	823	f.index = 5
	824	g = Vertex (1,1)
	825	g.index = 6
	826	inputVerts_noDups = [a,b,c,d,e]
	827
	828	m = Mesh(userVertices=[a,b,c,d,e,f,g])
	829	counter = m.removeDuplicatedUserVertices()
	830	UserVerts = m.getUserVertices()
	831
	832
	833	self.failUnless(UserVerts == inputVerts_noDups,
	834	'duplicate verts not removed')
	835	#for userVert, inputVert in map(None, UserVerts, inputVerts_noDups):
	836	# self.failUnless(userVert.x == inputVert.x,
	837	# 'x duplicate verts not removed')
	838	# self.failUnless(userVert.y == inputVert.y,
	839	# 'y duplicate verts not removed')
[406]	840
	841
[407]	842	def test_ungenerateFileLoading(self):
[349]	843
[406]	844	fileName = tempfile.mktemp(".txt")
	845	file = open(fileName,"w")
	846	file.write(" 1 ?? ??\n\
	847	0.0 0.0\n\
	848	1.0 0.0\n\
	849	1.0 1.0\n\
	850	0.0 1.0\n\
	851	0.0 0.0\n\
	852	END\n\
	853	2 ?? ??\n\
	854	10.0 10.0\n\
	855	10.0 20.0\n\
	856	20.0 20.0\n\
	857	10.0 10.0\n\
	858	END\n\
	859	END\n")
	860	file.close()
	861
	862
	863	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	864	b = Vertex (0.0, 40.0) #, attributes = [1.2])
	865	c = Vertex (40.0,40.0) #, attributes = [1.3])
	866	d = Vertex (40.0,0.0) #, attributes = [1.4])
	867
	868	s1 = Segment(a,b)
	869	s2 = Segment(b,c)
	870	s3 = Segment(c,d)
	871	s4 = Segment(d,a)
	872
	873	m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])
	874	dict = importUngenerateFile(fileName)
[407]	875
	876	tag = "DSG"
	877	Segment.set_default_tag(tag)
[406]	878	m.addVertsSegs(dict)
[407]	879
	880	# have to reset this , since it's a class attribute
	881	Segment.set_default_tag("")
[406]	882
	883	self.failUnless(len(m.userSegments) ==11,
[407]	884	'Wrong segment list length.')
[406]	885	self.failUnless(len(m.userVertices) == 11,
[407]	886	'Wrong vertex list length.')
[406]	887	self.failUnless(m.userSegments[10].vertices[0] == m.userVertices[10],
[407]	888	'bad vertex on segment.')
[406]	889	self.failUnless(m.userSegments[10].vertices[1] == m.userVertices[8],
[407]	890	'Bad segment.')
	891	self.failUnless(m.userSegments[10].marker == tag,
	892	'wrong tag.')
[431]	893
	894	def test_ungenerateFileLoadingII(self):
[406]	895
[431]	896	fileName = tempfile.mktemp(".txt")
	897	file = open(fileName,"w")
	898	file.write(" 1 ?? ??\n\
	899	0.0 0.0\n\
	900	1.0 0.0\n\
	901	1.0 1.0\n\
	902	0.0 1.0\n\
	903	0.0 0.0\n\
	904	END\n\
	905	2 ?? ??\n\
	906	10.0 10.0\n\
	907	10.0 20.0\n\
	908	20.0 20.0\n\
	909	END\n\
	910	END\n")
	911	file.close()
	912
	913
	914	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	915	b = Vertex (0.0, 40.0) #, attributes = [1.2])
	916	c = Vertex (40.0,40.0) #, attributes = [1.3])
	917	d = Vertex (40.0,0.0) #, attributes = [1.4])
	918
	919	s1 = Segment(a,b)
	920	s2 = Segment(b,c)
	921	s3 = Segment(c,d)
	922	s4 = Segment(d,a)
	923
	924	m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])
	925	dict = importUngenerateFile(fileName)
	926
	927	tag = "DSG"
	928	Segment.set_default_tag(tag)
	929	m.addVertsSegs(dict)
	930
	931	# have to reset this , since it's a class attribute
	932	Segment.set_default_tag("")
	933
	934	self.failUnless(len(m.userSegments) ==10,
	935	'Wrong segment list length.')
	936	self.failUnless(len(m.userVertices) == 11,
	937	'Wrong vertex list length.')
	938
[349]	939	#-------------------------------------------------------------
	940	if __name__ == "__main__":
[406]	941	suite = unittest.makeSuite(meshTestCase,'test')
[349]	942	runner = unittest.TextTestRunner()
	943	runner.run(suite)
	944

Note: See TracBrowser for help on using the repository browser.

Download in other formats: