mirror of
https://github.com/wassname/simpeg.git
synced 2026-07-01 10:53:21 +08:00
Rowan Cockett
124 lines
4.0 KiB
Python
124 lines
4.0 KiB
Python
from SimPEG import Mesh
|
|
from SimPEG.Mesh.PointerTree import Tree
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
import unittest
|
|
|
|
TOL = 1e-10
|
|
|
|
|
|
|
|
class TestSimpleQuadTree(unittest.TestCase):
|
|
|
|
def test_counts(self):
|
|
|
|
M = Tree([8,8])
|
|
M._refineCell([0,0,0])
|
|
M._refineCell([0,0,1])
|
|
M.number()
|
|
# M.plotGrid(showIt=True)
|
|
# assert sorted(M._cells) == [2, 34, 66, 99, 107, 115, 123, 129, 257, 386, 418, 450, 482]
|
|
assert M.nhFx == 2
|
|
assert M.nFx == 9
|
|
assert M.vol.sum() == 1.0
|
|
|
|
|
|
# def test_connectivity(self):
|
|
# T = Tree([8,8])
|
|
# T._refineCell([0,0,0])
|
|
# T._refineCell([4,4,1])
|
|
# T._refineCell([0,0,1])
|
|
# T._refineCell([2,2,2])
|
|
# T.number()
|
|
# assert T._getNextCell([4,0,1]) is None
|
|
# assert T._getNextCell([0,4,1]) == [T._index([4,4,2]), T._index([4,6,2])]
|
|
# assert T._getNextCell([0,2,2]) == [T._index([2,2,3]), T._index([2,3,3])]
|
|
# assert T._getNextCell([4,4,2]) == T._index([6,4,2])
|
|
# assert T._getNextCell([6,4,2]) is None
|
|
# assert T._getNextCell([2,0,2]) == T._index([4,0,1])
|
|
# assert T._getNextCell([4,0,1], positive=False) == [T._index([2,0,2]), [T._index([3,2,3]), T._index([3,3,3])]]
|
|
# assert T._getNextCell([3,3,3]) == T._index([4,0,1])
|
|
# assert T._getNextCell([3,2,3]) == T._index([4,0,1])
|
|
# assert T._getNextCell([2,2,3]) == T._index([3,2,3])
|
|
# assert T._getNextCell([3,2,3], positive=False) == T._index([2,2,3])
|
|
|
|
|
|
# assert T._getNextCell([0,0,2], direction=1) == T._index([0,2,2])
|
|
# assert T._getNextCell([0,2,2], direction=1, positive=False) == T._index([0,0,2])
|
|
# assert T._getNextCell([0,2,2], direction=1) == T._index([0,4,1])
|
|
# assert T._getNextCell([0,4,1], direction=1, positive=False) == [T._index([0,2,2]), [T._index([2,3,3]), T._index([3,3,3])]]
|
|
|
|
|
|
class TestOperatorsQuadTree(unittest.TestCase):
|
|
|
|
def test_counts(self):
|
|
|
|
hx, hy = np.r_[1.,2,3,4], np.r_[5.,6,7,8]
|
|
T = Tree([hx, hy], levels=2)
|
|
T.refine(lambda xc:2)
|
|
# T.plotGrid(showIt=True)
|
|
M = Mesh.TensorMesh([hx, hy])
|
|
assert M.nC == T.nC
|
|
assert M.nF == T.nF
|
|
assert M.nFx == T.nFx
|
|
assert M.nFy == T.nFy
|
|
assert M.nE == T.nE
|
|
assert M.nEx == T.nEx
|
|
assert M.nEy == T.nEy
|
|
assert np.allclose(M.area, T.permuteF*T.area)
|
|
assert np.allclose(M.edge, T.permuteE*T.edge)
|
|
assert np.allclose(M.vol, T.permuteCC*T.vol)
|
|
|
|
# plt.subplot(211).spy(M.faceDiv)
|
|
# plt.subplot(212).spy(T.permuteCC.T*T.faceDiv*T.permuteF)
|
|
# plt.show()
|
|
|
|
assert (M.faceDiv - T.permuteCC*T.faceDiv*T.permuteF.T).nnz == 0
|
|
|
|
|
|
class TestOperatorsOcTree(unittest.TestCase):
|
|
|
|
def test_faceDiv(self):
|
|
|
|
hx, hy, hz = np.r_[1.,2,3,4], np.r_[5.,6,7,8], np.r_[9.,10,11,12]
|
|
M = Tree([hx, hy, hz], levels=2)
|
|
M.refine(lambda xc:2)
|
|
# M.plotGrid(showIt=True)
|
|
Mr = Mesh.TensorMesh([hx, hy, hz])
|
|
assert M.nC == Mr.nC
|
|
assert M.nF == Mr.nF
|
|
assert M.nFx == Mr.nFx
|
|
assert M.nFy == Mr.nFy
|
|
assert M.nE == Mr.nE
|
|
assert M.nEx == Mr.nEx
|
|
assert M.nEy == Mr.nEy
|
|
assert np.allclose(Mr.area, M.permuteF*M.area)
|
|
assert np.allclose(Mr.edge, M.permuteE*M.edge)
|
|
assert np.allclose(Mr.vol, M.permuteCC*M.vol)
|
|
|
|
# plt.subplot(211).spy(Mr.faceDiv)
|
|
# plt.subplot(212).spy(M.permuteCC.T*M.faceDiv*M.permuteF)
|
|
# plt.show()
|
|
|
|
assert (Mr.faceDiv - M.permuteCC*M.faceDiv*M.permuteF.T).nnz == 0
|
|
|
|
|
|
def test_edgeCurl(self):
|
|
|
|
hx, hy, hz = np.r_[1.,2,3,4], np.r_[5.,6,7,8], np.r_[9.,10,11,12]
|
|
M = Tree([hx, hy, hz], levels=2)
|
|
M.refine(lambda xc:2)
|
|
# M.plotGrid(showIt=True)
|
|
Mr = Mesh.TensorMesh([hx, hy, hz])
|
|
|
|
# plt.subplot(211).spy(Mr.faceDiv)
|
|
# plt.subplot(212).spy(M.permuteCC.T*M.faceDiv*M.permuteF)
|
|
# plt.show()
|
|
|
|
assert (Mr.edgeCurl - M.permuteF*M.edgeCurl*M.permuteE.T).nnz == 0
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|