source: inundation/mesh_engine/test_triangmodule.py @ 3028

Last change on this file since 3028 was 3028, checked in by duncan, 18 years ago

trying to fix a problem with import statements

File size: 14.3 KB
Line 
1#!/usr/bin/env python
2#
3
4import sys
5
6import unittest
7import mesh_engine.triang as triang
8
9
10class triangTestCase(unittest.TestCase):
11    def setUp(self):
12        pass
13
14    def tearDown(self):
15        pass
16
17    def testrectangle(self):
18
19        points = []
20        seglist = []
21        segattlist = []
22        holelist = []
23        regionlist = []
24
25        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
26        # This will cause a seg fault
27        #points = [[0.0,0.0],[0.0,10.0],[3.0,0.0],[3.0,10.0]]
28
29        # automatically generate pointattlist
30        pointattlist = []
31        for point in points:
32            pointattlist.append([])
33        #print "pointattlist",pointattlist
34        #pointattlist = [[],[],[],[]]
35        trilist = []
36        mode = "Qzcn"
37        data = triang.genMesh(points,seglist,holelist,regionlist,
38                              pointattlist,segattlist,trilist, mode)
39        self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)],
40                        'generatedtrianglelist is wrong!')
41        #print "data['generatedsegmentlist']",data['generatedsegmentlist']
42        self.failUnless(data['generatedsegmentlist'] ==[(2, 0), (3, 2), (1, 3),
43                                                        (0, 1)]
44              ,          'generatedsegmentlist is wrong!')
45        self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0),
46                                                      (3.0, 0.0), (3.0, 10.0)],
47                        ' is wrong!')
48
49        #print "data['generatedtriangleneighborlist']",
50        #data['generatedtriangleneighborlist']
51        self.failUnless(data['generatedtriangleneighborlist'] ==[(-1, 1, -1),
52                                                                 (-1, 0, -1)],
53                        'generatedtriangleneighborlist is wrong!')
54
55
56    def testrectangleII(self):
57
58        points = []
59        seglist = []
60        holelist = []
61        regionlist = []
62
63        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
64        pointattlist = [[],[],[],[]]
65        regionlist.append( (1.2,1.2,5.0) )
66        seglist = [(0,1),(1,3),(3,2),(2,0)]
67        segattlist = [0,0,0,0]
68        trilist = []
69        mode = "Qzp"
70        data = triang.genMesh(points,seglist,holelist,regionlist,
71                              pointattlist,segattlist,trilist, mode)
72
73        self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)],
74                        'trianglelist is wrong!')
75        self.failUnless(data['generatedsegmentlist'] ==[(0, 1), (1, 3),
76                                                        (3, 2), (2, 0)],
77                        'segmentlist is wrong!')
78        self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0),
79                                                      (3.0, 0.0), (3.0, 10.0)],
80                        ' is wrong!')
81
82    def testsegmarker(self):
83
84        points = []
85        seglist = []
86        holelist = []
87        regionlist = []
88
89        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
90        pointattlist = [[],[],[],[]]
91        regionlist.append( (1.2,1.2,5.0) )
92        seglist = [(0,1),(1,3),(3,2),(2,0)]
93        segattlist = [1.0,2.0,3.0,4.0]
94        trilist = []
95        mode = "Qzp"
96        data = triang.genMesh(points,seglist,holelist,regionlist,
97                              pointattlist,segattlist,trilist, mode)
98
99        self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)],
100                        'trianglelist is wrong!')
101        self.failUnless(data['generatedsegmentlist'] ==[(0, 1), (1, 3),
102                                                        (3, 2), (2, 0)],
103                        'segmentlist is wrong!')
104        self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0),
105                                                      (3.0, 0.0), (3.0, 10.0)],
106                        ' is wrong!')
107        self.failUnless(data['generatedsegmentmarkerlist'] ==[1,2,3,4],
108                        ' is wrong!')
109
110    def testbad_region(self):
111
112        points = []
113        seglist = []
114        holelist = []
115        regionlist = [(1.2,1.2)]
116
117        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
118        pointattlist = [[],[],[],[]]
119        seglist = [(0,1),(1,3),(3,2),(2,0)]
120        segattlist = [0,0,0,0]
121        trilist = []
122        mode = "Qzpn"
123        try:
124            data = triang.genMesh(points,seglist,holelist,
125                                  regionlist,pointattlist,segattlist,
126                                  trilist, mode)
127
128        except TypeError:
129            pass
130        else:
131            self.failUnless(0 ==1,
132                        'bad region list did not raise error!')
133
134    def testregion_with_maxarea(self):
135
136        points = []
137        seglist = []
138        holelist = []
139        regionlist = [(3,1,1.0)]
140
141        points = [(0.0,0.0),(6.0,0.0),(6.0,6.0),(0.0,6.0)]
142        pointattlist = [[],[],[],[]]
143        seglist = [(0,1),(1,2),(3,2),(3,0),(0,2)]
144        segattlist = [0,0,0,0,0]
145        trilist = []
146        mode = "Qzpna36a"
147        data = triang.genMesh(points,seglist,holelist,regionlist,
148                              pointattlist,segattlist, trilist, mode)
149
150        self.failUnless(len(data['generatedtrianglelist']) == 2,
151                        'testregion_with_maxarea 1: # of tris is wrong!')
152        ## Another test case
153        regionlist = [(3,1,1.0),(1,3,1.0,8.0)]
154        mode = "Qzp21na36a"
155        data = triang.genMesh(points,seglist,holelist,regionlist,
156                              pointattlist,segattlist, trilist, mode)
157        #print "len(data['generatedtrianglelist']",len(data['generatedtrianglelist'])
158        # for Duncan On unix this returns a 7 triangle result.
159        # for Duncan on Windows returns a 6 triangle result.
160        # for Ole on nautilus this returns 6
161        # for Duncan on nautilus this returns 7
162        # ??, it seems to be the results from triangle that is
163        # causing the different results, and we are treating
164        # triangle as a back box.
165
166        self.failUnless(len(data['generatedtrianglelist']) >= 6,
167                        'testregion_with_maxarea 2: # of tris is wrong!')
168        ## Another test case
169        regionlist = [(3,1,1.0,8.0),(1,3,1.0,8.0)]
170        mode = "Qzpna36a"
171        data = triang.genMesh(points,seglist,holelist,regionlist,
172                              pointattlist,segattlist, trilist, mode)
173        #print "len(data['generatedtrianglelist']",len(data['generatedtrianglelist'])
174        # On unix this returns a 10 triangle result.
175        # Windows returns a 8 triangle result.
176        self.failUnless(len(data['generatedtrianglelist']) >= 8,
177                        'testregion_with_maxarea 3: # of tris is wrong!')
178
179        ## Another test case
180        regionlist = [(3,1,1.0),(1,3,1.0,8.0)]
181        mode = "Qzpna8a"
182        data = triang.genMesh(points,seglist,holelist,regionlist,
183                              pointattlist,segattlist, trilist, mode)
184        #print "len(data['generatedtrianglelist']",len(data['generatedtrianglelist'])
185        # On unix this returns a 10 triangle result.
186        # Windows returns a 8 triangle result.
187
188        self.failUnless(len(data['generatedtrianglelist']) >= 8,
189                        'testregion_with_maxarea 4: # of tris is wrong!')
190
191    def testbad_point(self):
192
193        holelist = []
194        regionlist = []
195
196        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0)]
197        pointattlist = [[],[],[],[]]
198        seglist = [(0,1),(1,3),(3,2),(2,0)]
199        segattlist = [0,0,0,0]
200        trilist = []
201        mode = "Qzpn"
202        try:
203            data = triang.genMesh(points,seglist,holelist,regionlist,
204                                  pointattlist,segattlist, trilist, mode)
205
206        except TypeError:
207            pass
208        else:
209            self.failUnless(0 ==1,
210                        'bad point list did not raise error!')
211
212    def testbad_hole(self):
213
214        holelist = [(9.0)]
215        regionlist = []
216
217        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
218        pointattlist = [[],[],[],[]]
219        seglist = [(0,1),(1,3),(3,2),(2,0)]
220        segattlist = [0,0,0,0]
221        trilist = []
222        mode = "Qzpn"
223        try:
224            data = triang.genMesh(points,seglist,holelist,regionlist,
225                                  pointattlist,segattlist, trilist, mode)
226
227        except TypeError:
228            pass
229        else:
230            self.failUnless(0 ==1,
231                        'bad hole list did not raise error!')
232
233    def testbad_segment(self):
234
235        holelist = []
236        regionlist = []
237
238        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
239        pointattlist = [[],[],[],[]]
240        seglist = [(0,1),(1,3),(3,2),(2)]
241        segattlist = [0,0,0,0]
242        trilist = []
243        mode = "Qzpn"
244        try:
245            data = triang.genMesh(points,seglist,holelist,regionlist,
246                                  pointattlist,segattlist, trilist, mode)
247
248        except TypeError:
249            pass
250        else:
251            self.failUnless(0 ==1,
252                        'bad segment list did not raise error!')
253
254    def testbad_segattlist(self):
255
256        holelist = []
257        regionlist = []
258
259        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
260        pointattlist = [[],[],[],[]]
261        seglist = [(0,1),(1,3),(3,2),(2,0)]
262        segattlist = [0,0]
263        trilist = []
264        mode = "Qzpn"
265        try:
266            data = triang.genMesh(points,seglist,holelist,regionlist,
267                                  pointattlist,segattlist, trilist, mode)
268
269            self.failUnless(data['trianglelist'] ==[(1, 0, 2), (2, 3, 1)],
270                        'trianglelist is wrong!')
271        except TypeError:
272            pass
273        else:
274            self.failUnless(0 ==1,
275                        'bad segment attribute list did not raise error!')
276
277    def testrectangle_regions(self):
278
279        points = []
280        seglist = []
281        holelist = []
282        regionlist = []
283
284        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)]
285        pointattlist = [[],[],[],[]]
286        #it seems that
287        #triangle only associates one region with a triangle
288        regionlist.append( (1.3,1.3,88.33) )
289        regionlist.append( (1.2,1.2,77,55) )
290        seglist = [(0,1),(1,3),(3,2),(2,0)]
291        segattlist = [0,0,0,0]
292        trilist = []
293        mode = "QAzpq"
294        data = triang.genMesh(points,seglist,holelist,regionlist,
295                              pointattlist,segattlist, trilist, mode)
296
297        self.failUnless(data['generatedtriangleattributelist'] ==[[77.0], [77.0], [77.0], [77.0]],
298                        'triangleattributelist is wrong!')
299
300
301    def Xtest_lone_verts(self):
302
303        points = []
304        seglist = []
305        holelist = []
306        regionlist = []
307
308        points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0),(0.0,10.0)]
309        pointattlist = [[],[],[],[],[]]
310        regionlist.append( (1.2,1.2,5.0) )
311        seglist = [(0,4),(4,1),(1,3),(3,2),(2,0)]
312        segattlist = [0,0,0,0,0]
313        trilist = []
314        mode = "Qzp"
315        data = triang.genMesh(points,seglist,holelist,regionlist,
316                              pointattlist,segattlist,trilist, mode)
317        #print "data['generatedtrianglelist']", data['generatedtrianglelist']
318        self.failUnless(data['generatedtrianglelist'] ==[(4, 0, 2), (2, 3, 4)],
319                        'trianglelist is wrong!')
320        self.failUnless(data['generatedsegmentlist'] ==[(0, 4), (4, 3),
321                                                        (3, 2), (2, 0)],
322                        'segmentlist is wrong!')
323        self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0),
324                                                      (3.0, 0.0), (3.0, 10.0),
325                                                      (0.0,10.0)],
326                        ' is wrong!')
327        self.failUnless(data['lonepointlist'] ==[1],
328                        'lonepointlist is wrong!')
329        #print "", data['lonepointlist']
330
331    def Xtest_lone_vertsII(self):
332
333        points = []
334        seglist = []
335        holelist = []
336        regionlist = []
337
338        points = [(0.0,0.0),(0.0,0.0),(0.0,10.0),(0.0,10.0),(10.0,10.0),
339                  (10.0,10.0),(0.0,10.0),(10.0,0.0)]
340
341        pointattlist = []
342        for point in points:
343            pointattlist.append([])
344        seglist = [(0,1),(1,2),(2,3),(3,4),(4,5),(5,7),(7,0)]
345        segattlist = []
346        for seg in seglist:
347            segattlist.append(0)
348        trilist = []
349        mode = "Qzp"
350        data = triang.genMesh(points,seglist,holelist,regionlist,
351                              pointattlist,segattlist,trilist, mode)
352        #print "data['generatedtrianglelist']", data['generatedtrianglelist']
353        self.failUnless(data['generatedtrianglelist'] ==[(6, 1, 7), (7, 5, 6)],
354                        'trianglelist is wrong!')
355        self.failUnless(data['generatedsegmentlist'] ==[(1, 6), (6, 5),
356                                                        (5, 7), (7, 1)],
357                        'segmentlist is wrong!')
358        self.failUnless(data['generatedpointlist'] ==[(0.0,0.0),(0.0,0.0),
359                                                      (0.0,10.0),(0.0,10.0),
360                                                      (10.0,10.0),
361                                                      (10.0,10.0),(0.0,10.0),
362                                                      (10.0,0.0)],
363                        ' is wrong!')
364        self.failUnless(data['lonepointlist'] ==[0,2,3,4],
365                        'lonepointlist is wrong!')
366
367    def Xtest_lone_vertsIII(self):
368
369        points = []
370        seglist = []
371        holelist = []
372        regionlist = []
373
374        points = [(0.0,0.0),(0.0,0.0),(0.0,10.0),(0.0,10.0),(10.0,10.0),
375                  (10.0,10.0),(0.0,10.0),(10.0,0.0)]
376
377        pointattlist = []
378        for point in points:
379            pointattlist.append([])
380        seglist = []
381        segattlist = []
382        for seg in seglist:
383            segattlist.append(0)
384        trilist = []
385        mode = "Qzp"
386        data = triang.genMesh(points,seglist,holelist,regionlist,
387                              pointattlist,segattlist,trilist, mode)
388
389        self.failUnless(data['lonepointlist'] ==[0,1,2,3,4,5,6,7],
390                        'lonepointlist is wrong!')
391if __name__ == "__main__":
392
393    suite = unittest.makeSuite(triangTestCase,'test')
394    runner = unittest.TextTestRunner()
395    runner.run(suite)
Note: See TracBrowser for help on using the repository browser.