Ignore:
Timestamp:
Feb 27, 2009, 11:54:09 AM (15 years ago)
Author:
rwilson
Message:

numpy changes

File:
1 edited

Legend:

Unmodified
Added
Removed
  • branches/numpy/anuga/abstract_2d_finite_volumes/test_general_mesh.py

    r6410 r6428  
    5353        e = [2.0, 2.0]
    5454        f = [4.0, 0.0]
    55 
    5655        nodes = num.array([a, b, c, d, e, f])
    5756
    5857        nodes_absolute = geo.get_absolute(nodes)
    5958       
    60         #bac, bce, ecf, dbe, daf, dae
     59        #                        bac,     bce,     ecf,     dbe
    6160        triangles = num.array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]], num.int)
    6261
    63         domain = General_mesh(nodes, triangles,
    64                        geo_reference = geo)
    65 
    66         verts = domain.get_vertex_coordinates(triangle_id=0)       
    67         msg = ("num.array([b,a,c])=\n%s\nshould be the same as 'verts'=\n%s"
     62        domain = General_mesh(nodes, triangles, geo_reference=geo)
     63
     64        verts = domain.get_vertex_coordinates(triangle_id=0)    # bac
     65        msg = ("num.array([b,a,c])=\n%s\nshould be close to 'verts'=\n%s"
    6866               % (str(num.array([b,a,c])), str(verts)))
    69         #self.assert_(num.allclose(num.array([b,a,c]), verts))
    70         assert num.allclose(num.array([b,a,c]), verts), msg
     67        self.failUnless(num.allclose(num.array([b,a,c]), verts), msg)
    7168
    7269        verts = domain.get_vertex_coordinates(triangle_id=0)       
    73         self.assert_(num.allclose(num.array([b,a,c]), verts))
     70        msg = ("num.array([b,a,c])=\n%s\nshould be close to 'verts'=\n%s"
     71               % (str(num.array([b,a,c])), str(verts)))
     72        self.assert_(num.allclose(num.array([b,a,c]), verts), msg)
     73
     74        verts = domain.get_vertex_coordinates(triangle_id=0, absolute=True)
     75        msg = ("num.array([...])=\n%s\nshould be close to 'verts'=\n%s"
     76               % (str(num.array([nodes_absolute[1],
     77                                 nodes_absolute[0],
     78                                 nodes_absolute[2]])),
     79                  str(verts)))
     80        self.assert_(num.allclose(num.array([nodes_absolute[1],
     81                                             nodes_absolute[0],
     82                                             nodes_absolute[2]]),
     83                                  verts), msg)
     84
    7485        verts = domain.get_vertex_coordinates(triangle_id=0,
    7586                                              absolute=True)       
     87        msg = ("num.array([...])=\n%s\nshould be close to 'verts'=\n%s"
     88               % (str(num.array([nodes_absolute[1],
     89                                 nodes_absolute[0],
     90                                 nodes_absolute[2]])),
     91                  str(verts)))
    7692        self.assert_(num.allclose(num.array([nodes_absolute[1],
    77                                      nodes_absolute[0],
    78                                      nodes_absolute[2]]), verts))
    79         verts = domain.get_vertex_coordinates(triangle_id=0,
    80                                               absolute=True)       
    81         self.assert_(num.allclose(num.array([nodes_absolute[1],
    82                                      nodes_absolute[0],
    83                                      nodes_absolute[2]]), verts))
    84        
    85        
     93                                             nodes_absolute[0],
     94                                             nodes_absolute[2]]),
     95                                  verts), msg)
    8696
    8797    def test_get_vertex_coordinates_triangle_id(self):
     
    283293        nodes_absolute = geo.get_absolute(nodes)
    284294       
    285         #bac, bce, ecf, dbe, daf, dae
     295        #                        bac,     bce,     ecf,     dbe
    286296        triangles = num.array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]])
    287297
    288         domain = General_mesh(nodes, triangles,
    289                        geo_reference = geo)
     298        domain = General_mesh(nodes, triangles, geo_reference = geo)
    290299        node = domain.get_node(2)       
    291         #self.assertEqual(c, node)
    292         msg = ('\nc=%s\nmode=%s' % (str(c), str(node)))
     300        msg = ('\nc=%s\nnode=%s' % (str(c), str(node)))
    293301        self.failUnless(num.alltrue(c == node), msg)
     302
     303        # repeat get_node(), see if result same
     304        node = domain.get_node(2)       
     305        msg = ('\nc=%s\nnode=%s' % (str(c), str(node)))
     306        self.failUnless(num.alltrue(c == node), msg)
    294307       
    295308        node = domain.get_node(2, absolute=True)     
    296         #self.assertEqual(nodes_absolute[2], node)
    297         self.failUnless(num.alltrue(nodes_absolute[2] == node))
    298        
     309        msg = ('\nnodes_absolute[2]=%s\nnode=%s'
     310               % (str(nodes_absolute[2]), str(node)))
     311        self.failUnless(num.alltrue(nodes_absolute[2] == node), msg)
     312       
     313        # repeat get_node(absolute=True), see if result same
    299314        node = domain.get_node(2, absolute=True)     
    300         #self.assertEqual(nodes_absolute[2], node)
    301         self.failUnless(num.alltrue(nodes_absolute[2] == node))
     315        msg = ('\nnodes_absolute[2]=%s\nnode=%s'
     316               % (str(nodes_absolute[2]), str(node)))
     317        self.failUnless(num.alltrue(nodes_absolute[2] == node), msg)
    302318       
    303319
     
    337353        self.failUnlessRaises(AssertionError, General_mesh,
    338354                              nodes, triangles, geo_reference=geo)
     355
     356    def test_raw_change_points_geo_ref(self):
     357        x0 = 1000.0
     358        y0 = 2000.0
     359        geo = Geo_reference(56, x0, y0)
     360       
    339361       
    340362
     
    343365
    344366if __name__ == "__main__":
    345     suite = unittest.makeSuite(Test_General_Mesh,'test')
     367    suite = unittest.makeSuite(Test_General_Mesh, 'test')
     368    #suite = unittest.makeSuite(Test_General_Mesh, 'test_get_vertex_coordinates_with_geo_ref')
    346369    runner = unittest.TextTestRunner()
    347370    runner.run(suite)
Note: See TracChangeset for help on using the changeset viewer.