mirror of
https://github.com/wassname/simpeg.git
synced 2026-07-02 00:29:13 +08:00
rowanc1
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import unittest
|
|
import sys
|
|
from SimPEG import *
|
|
|
|
|
|
class TestCyl1DMesh(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
hx = np.ones(3)
|
|
hz = np.ones(2)
|
|
self.mesh = Mesh.CylMesh([hx, 1,hz])
|
|
|
|
def test_cylMeshInheritance(self):
|
|
self.assertTrue(isinstance(self.mesh, Mesh.BaseMesh))
|
|
|
|
def test_cylMeshDimensions(self):
|
|
self.assertTrue(self.mesh.dim == 3)
|
|
|
|
def test_cylMesh_numbers(self):
|
|
self.assertTrue(self.mesh.nCx == 3)
|
|
self.assertTrue(self.mesh.nCy == 1)
|
|
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()
|