Ignore:
Timestamp:
Jan 13, 2009, 3:13:33 PM (15 years ago)
Author:
rwilson
Message:

Change Numeric imports to general form - ready to change to NumPy?.

File:
1 edited

Legend:

Unmodified
Added
Removed

  • anuga_core/source/anuga/mesh_engine/test_generate_mesh.py

    r5950 r6155  
    1212from anuga.mesh_engine.mesh_engine import generate_mesh
    1313
    14 from Numeric import array, Float, Int, alltrue, allclose
     14import Numeric as num
    1515
    1616from anuga.utilities.numerical_tools import ensure_numeric
    … …  
    4343                              pointattlist,segattlist, mode, points)
    4444
    45         correct = array([(1, 0, 2), (2, 3, 1)])
    46         self.failUnless(alltrue(data['generatedtrianglelist'].flat == \
    47                                 correct.flat),
     45        correct = num.array([(1, 0, 2), (2, 3, 1)])
     46        self.failUnless(num.alltrue(data['generatedtrianglelist'].flat == \
     47                                    correct.flat),
    4848                        'trianglelist is wrong!')
    49         correct = array([(0, 1), (1, 3), (3, 2), (2, 0)])
    50         self.failUnless(alltrue(data['generatedsegmentlist'].flat == \
    51                                 correct.flat),
     49        correct = num.array([(0, 1), (1, 3), (3, 2), (2, 0)])
     50        self.failUnless(num.alltrue(data['generatedsegmentlist'].flat == \
     51                                    correct.flat),
    5252                        'segmentlist is wrong!')
    5353
    54         correct = array([(0.0, 0.0), (0.0, 10.0),
    55                          (3.0, 0.0), (3.0, 10.0)])
    56         self.failUnless(allclose(data['generatedpointlist'].flat, \
    57                                 correct.flat),
     54        correct = num.array([(0.0, 0.0), (0.0, 10.0),
     55                             (3.0, 0.0), (3.0, 10.0)])
     56        self.failUnless(num.allclose(data['generatedpointlist'].flat, \
     57                                     correct.flat),
    5858                        'Failed')
    5959     
    … …  
    7474        data = generate_mesh(points,seglist,holelist,regionlist,
    7575                              pointattlist,segattlist, mode, points)
    76         correct = array([(1, 0, 2), (2, 3, 1)])
    77         self.failUnless(alltrue(data['generatedtrianglelist'].flat == \
    78                                 correct.flat),
     76        correct = num.array([(1, 0, 2), (2, 3, 1)])
     77        self.failUnless(num.alltrue(data['generatedtrianglelist'].flat == \
     78                                    correct.flat),
    7979                        'trianglelist is wrong!')
    80         correct = array([(0, 1), (1, 3), (3, 2), (2, 0)])
    81         self.failUnless(alltrue(data['generatedsegmentlist'].flat == \
    82                                 correct.flat),
     80        correct = num.array([(0, 1), (1, 3), (3, 2), (2, 0)])
     81        self.failUnless(num.alltrue(data['generatedsegmentlist'].flat == \
     82                                    correct.flat),
    8383                        'segmentlist is wrong!')
    8484
    85         correct = array([(0.0, 0.0), (0.0, 10.0),
    86                          (3.0, 0.0), (3.0, 10.0)])
    87         self.failUnless(allclose(data['generatedpointlist'].flat, \
    88                                 correct.flat),
     85        correct = num.array([(0.0, 0.0), (0.0, 10.0),
     86                             (3.0, 0.0), (3.0, 10.0)])
     87        self.failUnless(num.allclose(data['generatedpointlist'].flat, \
     88                                     correct.flat),
    8989                        'Failed')
    9090
    … …  
    154154                              pointattlist,segattlist, mode, points)
    155155
    156         correct = array([(1, 0, 2), (2, 3, 1)])
    157         self.failUnless(alltrue(data['generatedtrianglelist'].flat == \
    158                                 correct.flat),
     156        correct = num.array([(1, 0, 2), (2, 3, 1)])
     157        self.failUnless(num.alltrue(data['generatedtrianglelist'].flat == \
     158                                    correct.flat),
    159159                        'trianglelist is wrong!')
    160         correct = array([(0, 1), (1, 3), (3, 2), (2, 0)])
    161         self.failUnless(alltrue(data['generatedsegmentlist'].flat == \
    162                                 correct.flat),
     160        correct = num.array([(0, 1), (1, 3), (3, 2), (2, 0)])
     161        self.failUnless(num.alltrue(data['generatedsegmentlist'].flat == \
     162                                    correct.flat),
    163163                        'segmentlist is wrong!')
    164164
    165         correct = array([(0.0, 0.0), (0.0, 10.0),
    166                          (3.0, 0.0), (3.0, 10.0)])
    167         self.failUnless(allclose(data['generatedpointlist'].flat, \
    168                                 correct.flat),
    169                         'Failed')
    170        
    171         self.failUnless(alltrue(data['generatedsegmentmarkerlist'] == \
    172                                 array([1,2,3,4])),
     165        correct = num.array([(0.0, 0.0), (0.0, 10.0),
     166                             (3.0, 0.0), (3.0, 10.0)])
     167        self.failUnless(num.allclose(data['generatedpointlist'].flat, \
     168                                     correct.flat),
     169                        'Failed')
     170       
     171        self.failUnless(num.alltrue(data['generatedsegmentmarkerlist'] == \
     172                                    num.array([1,2,3,4])),
    173173                        'Failed!')
    174174       
    … …  
    338338                              pointattlist,segattlist, mode, points)
    339339
    340         correct = array([[77.0], [77.0], [77.0], [77.0]])
    341         self.failUnless(allclose(data['generatedtriangleattributelist'].flat,
    342                                  correct.flat),
     340        correct = num.array([[77.0], [77.0], [77.0], [77.0]])
     341        self.failUnless(num.allclose(data['generatedtriangleattributelist'].flat,
     342                                     correct.flat),
    343343                        'Failed')
    344344       
    … …  
    362362                              pointattlist,segattlist, mode, points)
    363363
    364         correct = array([[77.0], [77.0], [77.0], [77.0]])
    365         self.failUnless(allclose(data['generatedtriangleattributelist'].flat,
    366                                  correct.flat),
     364        correct = num.array([[77.0], [77.0], [77.0], [77.0]])
     365        self.failUnless(num.allclose(data['generatedtriangleattributelist'].flat,
     366                                     correct.flat),
    367367                        'Failed')
    368368
    … …  
    388388
    389389     
    390         correct = array([(1, 0, 2), (2, 3, 1)])
    391         self.failUnless(alltrue(data['generatedtrianglelist'].flat == \
    392                                 correct.flat),
     390        correct = num.array([(1, 0, 2), (2, 3, 1)])
     391        self.failUnless(num.alltrue(data['generatedtrianglelist'].flat == \
     392                                    correct.flat),
    393393                        'trianglelist is wrong!')
    394         correct = array([(0, 1), (1, 3), (3, 2), (2, 0)])
    395         self.failUnless(alltrue(data['generatedsegmentlist'].flat == \
    396                                 correct.flat),
     394        correct = num.array([(0, 1), (1, 3), (3, 2), (2, 0)])
     395        self.failUnless(num.alltrue(data['generatedsegmentlist'].flat == \
     396                                    correct.flat),
    397397                        'segmentlist is wrong!')
    398398
    399         correct = array([(0.0, 0.0), (0.0, 10.0),
    400                          (3.0, 0.0), (3.0, 10.0)])
    401         self.failUnless(allclose(data['generatedpointlist'].flat, \
    402                                 correct.flat),
    403                         'Failed')
    404        
    405         correct = array([[tri_tag], [tri_tag]])
    406         self.failUnless(allclose(data['generatedtriangleattributelist'].flat, \
    407                                 correct.flat),
    408                         'Failed')
    409         correct = array([(0, 1), (1, 3), (3, 2), (2, 0)])
    410         self.failUnless(alltrue(data['generatedsegmentlist'].flat == \
    411                                 correct.flat),
     399        correct = num.array([(0.0, 0.0), (0.0, 10.0),
     400                             (3.0, 0.0), (3.0, 10.0)])
     401        self.failUnless(num.allclose(data['generatedpointlist'].flat, \
     402                                     correct.flat),
     403                        'Failed')
     404       
     405        correct = num.array([[tri_tag], [tri_tag]])
     406        self.failUnless(num.allclose(data['generatedtriangleattributelist'].flat, \
     407                                     correct.flat),
     408                        'Failed')
     409        correct = num.array([(0, 1), (1, 3), (3, 2), (2, 0)])
     410        self.failUnless(num.alltrue(data['generatedsegmentlist'].flat == \
     411                                    correct.flat),
    412412                        'Failed!')
    413413       
    414         correct = array(segattlist)
    415         self.failUnless(allclose(data['generatedsegmentmarkerlist'].flat,
    416                                  correct.flat),
     414        correct = num.array(segattlist)
     415        self.failUnless(num.allclose(data['generatedsegmentmarkerlist'].flat,
     416                                     correct.flat),
    417417                        'Failed')
    418418       
    419419        # I copied these answers from the output, so bad test..
    420         correct = array([(-1, 1, -1), (-1, 0, -1)])
    421         self.failUnless(alltrue(data['generatedtriangleneighborlist'].flat == \
    422                                 correct.flat),
     420        correct = num.array([(-1, 1, -1), (-1, 0, -1)])
     421        self.failUnless(num.alltrue(data['generatedtriangleneighborlist'].flat == \
     422                                    correct.flat),
    423423                        'Failed!')
    424424       
    … …  
    440440                              pointattlist,segattlist, mode, points)
    441441
    442         correct = array([[0.0],[0.0],[10],[10]])
    443         self.failUnless(allclose(data['generatedpointattributelist'].flat, \
    444                                 correct.flat),
     442        correct = num.array([[0.0],[0.0],[10],[10]])
     443        self.failUnless(num.allclose(data['generatedpointattributelist'].flat, \
     444                                     correct.flat),
    445445                        'Failed')
    446446       
    … …  
    449449        data = generate_mesh(points,seglist,holelist,regionlist,
    450450                              pointattlist,segattlist, mode, points)
    451         correct = array([[0.0],[0.0],[10],[10]])
    452         self.failUnless(allclose(data['generatedpointattributelist'].flat, \
    453                                 correct.flat),
     451        correct = num.array([[0.0],[0.0],[10],[10]])
     452        self.failUnless(num.allclose(data['generatedpointattributelist'].flat, \
     453                                     correct.flat),
    454454                        'Failed')
    455455       
    … …  
    459459                              pointattlist,segattlist, mode, points)
    460460        #print "data", data
    461         correct = array(pointattlist)
    462         self.failUnless(allclose(data['generatedpointattributelist'].flat, \
    463                                 correct.flat),
     461        correct = num.array(pointattlist)
     462        self.failUnless(num.allclose(data['generatedpointattributelist'].flat, \
     463                                     correct.flat),
    464464                        'Failed')
    465465     
Note: See TracChangeset for help on using the changeset viewer.