Changeset 6155


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

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

Location:
anuga_core/source/anuga/mesh_engine
Files:
2 edited

Legend:

Unmodified
Added
Removed
  • anuga_core/source/anuga/mesh_engine/mesh_engine.py

    r4916 r6155  
    1111#import anuga.mesh_engine.list_dic as triang
    1212
    13 from Numeric import array, Float, Int32, reshape
     13import Numeric as num
    1414
    1515from anuga.utilities.numerical_tools import ensure_numeric
     
    4646       
    4747    try:
    48         points =  ensure_numeric(points, Float)
     48        points =  ensure_numeric(points, num.Float)
    4949    except ValueError:
    5050        msg = 'ERROR: Inconsistent points array.'
     
    6161    try:
    6262        # If Int is used, instead of Int32, it fails in Linux
    63         segments = ensure_numeric(segments, Int32)
     63        segments = ensure_numeric(segments, num.Int32)
    6464       
    6565    except ValueError:
     
    7272       
    7373    try:
    74         holes = ensure_numeric(holes, Float)
     74        holes = ensure_numeric(holes, num.Float)
    7575    except ValueError:
    7676        msg = 'ERROR: Inconsistent holess array.'
     
    8080    regions = add_area_tag(regions)
    8181    try:
    82         regions = ensure_numeric(regions, Float)
     82        regions = ensure_numeric(regions, num.Float)
    8383    except  (ValueError, TypeError):
    8484        msg = 'ERROR: Inconsistent regions array.'
     
    9090   
    9191    try:
    92         pointatts = ensure_numeric(pointatts, Float)
     92        pointatts = ensure_numeric(pointatts, num.Float)
    9393    except (ValueError, TypeError):
    9494        msg = 'ERROR: Inconsistent point attributes array.'
     
    100100        raise ANUGAError, msg
    101101    if len(pointatts.shape) == 1:
    102         pointatts = reshape(pointatts,(pointatts.shape[0],1))
    103    
    104     try:
    105         segatts = ensure_numeric(segatts, Int32)
     102        pointatts = num.reshape(pointatts,(pointatts.shape[0],1))
     103   
     104    try:
     105        segatts = ensure_numeric(segatts, num.Int32)
    106106    except ValueError:
    107107        msg = 'ERROR: Inconsistent point attributes array.'
  • 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.