test simple grid functions add Nodal Grid

2026-09-12 12:51:28 +08:00 · 2015-02-09 21:48:42 -08:00
parent 310f327b5a
commit 5e5a3221e8
2 changed files with 83 additions and 5 deletions
@@ -0,0 +1,59 @@
+from SimPEG.Mesh import TensorMesh
+from SimPEG.Mesh.NewTreeMesh import TreeMesh
+import numpy as np
+import unittest
+import matplotlib.pyplot as plt
+
+TOL = 1e-10
+
+class TestQuadTreeMesh(unittest.TestCase):
+
+    def setUp(self):
+        M = TreeMesh([np.ones(x) for x in [3,2]])
+        M.refineFace(0)
+        self.M = M
+        M.number()
+        # M.plotGrid(showIt=True)
+
+    def test_MeshSizes(self):
+        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_gridN(self):
+        x = np.r_[0,0.5,1,2,3,0,0.5,1,0,0.5,1,2,3,0,1,2,3]
+        y = np.r_[0,0,0,0,0,.5,.5,.5,1,1,1,1,1,2,2,2,2]
+        self.assertTrue(np.linalg.norm((np.c_[x,y]-self.M.gridN).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)
+
+    def test_gridEx(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.gridEx).flatten()) == 0)
+
+    def test_gridEy(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.gridEy).flatten()) == 0)
+
+
+if __name__ == '__main__':
+    unittest.main()