wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-09-10 12:37:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Initial fields and data implementation.

Browse Source
This commit is contained in:
rowanc1
2014-03-10 22:47:43 -07:00
parent f10c029161
commit e1010f9eca
4 changed files with 35 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegEM/Tests/test_FieldsObject.py
+8 -2
View File
@@ -22,10 +22,16 @@ class FieldsTest(unittest.TestCase):
def test_SetGet(self):
F = self.F
for freq in F.survey.freqs:
e = np.random.rand(F.mesh.nE, F.survey.nTx[freq])
nFreq = F.survey.nTx[freq]
e = np.random.rand(F.mesh.nE, nFreq)
F[freq, 'e'] = e
b = np.random.rand(F.mesh.nF, F.survey.nTx[freq])
b = np.random.rand(F.mesh.nF, nFreq)
F[freq, 'b'] = b
if nFreq == 1:
F[freq, 'b'] = Utils.mkvc(b)
self.assertTrue(np.all(F[freq, 'e'] == e))
self.assertTrue(np.all(F[freq, 'b'] == b))
F[freq] = {'b':b,'e':e}
self.assertTrue(np.all(F[freq, 'e'] == e))
self.assertTrue(np.all(F[freq, 'b'] == b))