wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-08-09 12:30:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Grid updates for TreeMesh, initial generalizations for OcTree

Browse Source
This commit is contained in:
rowanc1
2014-02-07 16:28:03 -08:00
parent 3d21f690cc
commit 2f52e15e41
3 changed files with 159 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/tests/test_QuadTree.py
+27 -5
View File
@@ -1,19 +1,41 @@
from SimPEG import mesh, np
from SimPEG import Mesh, np
import unittest
class TestCheckDerivative(unittest.TestCase):
class TestQuadTreeMesh(unittest.TestCase):
def setUp(self):
M = mesh.QuadTreeMesh([3,2],[1,2])
M = Mesh.TreeMesh([np.ones(x) for x in [3,2]])
for ii in range(1):
M.children[ii,ii].refine()
self.M = M
M.number()
# M.plotGrid(showIt=True)
def test_MeshSizes(self):
self.assertTrue(len(self.M.faces)==25)
self.assertTrue(len(self.M.cells)==9)
self.assertTrue(self.M.nC==9)
self.assertTrue(self.M.nF==25)
self.assertTrue(self.M.nFx==12)
self.assertTrue(self.M.nFy==13)
self.assertTrue(self.M.nE==25)
self.assertTrue(self.M.nEx==13)
self.assertTrue(self.M.nEy==12)
def test_gridCC(self):
x = np.r_[0.25,0.75,1.5,2.5,0.25,0.75,0.5,1.5,2.5]
y = np.r_[0.25,0.25,0.5,0.5,0.75,0.75,1.5,1.5,1.5]
self.assertTrue(np.linalg.norm((np.c_[x,y]-self.M.gridCC).flatten()) == 0)
def test_gridFx(self):
x = np.r_[0.0,0.5,1.0,2.0,3.0,0.0,0.5,1.0,0.0,1.0,2.0,3.0]
y = np.r_[0.25,0.25,0.25,0.5,0.5,0.75,0.75,0.75,1.5,1.5,1.5,1.5]
self.assertTrue(np.linalg.norm((np.c_[x,y]-self.M.gridFx).flatten()) == 0)
def test_gridFy(self):
x = np.r_[0.25,0.75,1.5,2.5,0.25,0.75,0.25,0.75,1.5,2.5,0.5,1.5,2.5]
y = np.r_[0,0,0,0,0.5,0.5,1,1,1,1,2,2,2]
self.assertTrue(np.linalg.norm((np.c_[x,y]-self.M.gridFy).flatten()) == 0)