wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-06-29 08:32:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

test counting, put dummies in for grid3D calcs

Browse Source
This commit is contained in:
Rowan Cockett
2015-02-10 12:42:35 -08:00
parent 72c8dfb3ef
commit 5c85f13cb6
2 changed files with 46 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Mesh/NewTreeMesh.py
+8
View File
@@ -375,6 +375,10 @@ class TreeMesh(BaseMesh):
return np.c_[Ex, Ey][P, :]
@property
def gridEz(self):
pass
@property
def gridFx(self):
if self.dim == 2:
@@ -385,6 +389,10 @@ class TreeMesh(BaseMesh):
if self.dim == 2:
return self.gridEx
@property
def gridFz(self):
pass
def _push(self, attr, rows):
self.isNumbered = False
rows = np.atleast_2d(rows)
SimPEG/Tests/test_NewTreeMesh.py
+38
View File
@@ -102,6 +102,44 @@ class SimpleOctreeOperatorTests(unittest.TestCase):
class TestOcTreeObjects(unittest.TestCase):
def setUp(self):
self.M = TreeMesh([2,1,1])
self.M.number()
# self.Mr = TreeMesh([2,1,1])
# self.Mr.children[0,0,0].refine()
# self.Mr.number()
def test_counts(self):
self.assertTrue(self.M.nC == 2)
self.assertTrue(self.M.nFx == 3)
self.assertTrue(self.M.nFy == 4)
self.assertTrue(self.M.nFz == 4)
self.assertTrue(self.M.nF == 11)
self.assertTrue(self.M.nEx == 8)
self.assertTrue(self.M.nEy == 6)
self.assertTrue(self.M.nEz == 6)
self.assertTrue(self.M.nE == 20)
self.assertTrue(self.M.nN == 12)
# self.assertTrue(self.Mr.nC == 9)
# self.assertTrue(self.Mr.nFx == 13)
# self.assertTrue(self.Mr.nFy == 14)
# self.assertTrue(self.Mr.nFz == 14)
# self.assertTrue(self.Mr.nF == 41)
# for cell in self.Mr.sortedCells:
# for e in cell.edgeDict:
# self.assertTrue(cell.edgeDict[e].edgeType==e[1].lower())
# self.assertTrue(self.Mr.nN == 31)
# self.assertTrue(self.Mr.nEx == 22)
# self.assertTrue(self.Mr.nEy == 20)
# self.assertTrue(self.Mr.nEz == 20)
if __name__ == '__main__':
unittest.main()