mirror of
https://github.com/wassname/simpeg.git
synced 2026-07-25 13:30:06 +08:00
Correct counting on cylinder mesh!
This commit is contained in:
@@ -3,12 +3,11 @@ import sys
|
||||
from SimPEG import *
|
||||
|
||||
|
||||
class TestCylMesh(unittest.TestCase):
|
||||
class TestCyl1DMesh(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
hx = np.ones(10)
|
||||
hz = np.ones(5)
|
||||
|
||||
hx = np.ones(3)
|
||||
hz = np.ones(2)
|
||||
self.mesh = Mesh.CylMesh([hx, 1,hz])
|
||||
|
||||
def test_cylMeshInheritance(self):
|
||||
@@ -17,13 +16,37 @@ class TestCylMesh(unittest.TestCase):
|
||||
def test_cylMeshDimensions(self):
|
||||
self.assertTrue(self.mesh.dim == 3)
|
||||
|
||||
def test_cylMesh_nc(self):
|
||||
self.assertTrue(np.all(self.mesh.vnC == [10, 1, 5]))
|
||||
|
||||
def test_cylMesh_nc_xyz(self):
|
||||
self.assertTrue(self.mesh.nCx == 10)
|
||||
def test_cylMesh_numbers(self):
|
||||
self.assertTrue(self.mesh.nCx == 3)
|
||||
self.assertTrue(self.mesh.nCy == 1)
|
||||
self.assertTrue(self.mesh.nCz == 5)
|
||||
self.assertTrue(self.mesh.nCz == 2)
|
||||
self.assertTrue(np.all(self.mesh.vnC == [3, 1, 2]))
|
||||
|
||||
self.assertTrue(self.mesh.nN == 0)
|
||||
self.assertTrue(self.mesh.nNx == 3)
|
||||
self.assertTrue(self.mesh.nNy == 0)
|
||||
self.assertTrue(self.mesh.nNz == 3)
|
||||
self.assertTrue(np.all(self.mesh.vnN == [3, 0, 3]))
|
||||
|
||||
self.assertTrue(self.mesh.nFx == 6)
|
||||
self.assertTrue(np.all(self.mesh.vnFx == [3, 1, 2]))
|
||||
self.assertTrue(self.mesh.nFy == 0)
|
||||
self.assertTrue(np.all(self.mesh.vnFy == [3, 0, 2]))
|
||||
self.assertTrue(self.mesh.nFz == 9)
|
||||
self.assertTrue(np.all(self.mesh.vnFz == [3, 1, 3]))
|
||||
self.assertTrue(self.mesh.nF == 15)
|
||||
self.assertTrue(np.all(self.mesh.vnF == [6, 0, 9]))
|
||||
|
||||
self.assertTrue(self.mesh.nEx == 0)
|
||||
self.assertTrue(np.all(self.mesh.vnEx == [3, 0, 3]))
|
||||
self.assertTrue(self.mesh.nEy == 9)
|
||||
self.assertTrue(np.all(self.mesh.vnEy == [3, 1, 3]))
|
||||
self.assertTrue(self.mesh.nEz == 0)
|
||||
self.assertTrue(np.all(self.mesh.vnEz == [3, 0, 2]))
|
||||
self.assertTrue(self.mesh.nE == 9)
|
||||
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user