Fancy timeFields.

2026-08-14 12:50:10 +08:00 · 2014-04-26 17:52:11 -07:00
parent 169ee9dd0b
commit df3ba5f8ee
2 changed files with 136 additions and 6 deletions
@@ -95,5 +95,83 @@ class DataAndFieldsTest(unittest.TestCase):
        def fun(): self.F[freq,'notThere']
        self.assertRaises(KeyError, fun)

+
+
+class FieldsTest_Time(unittest.TestCase):
+
+    def setUp(self):
+        mesh = Mesh.TensorMesh([np.ones(n)*5 for n in [10,11,12]],[0,0,-30])
+        x = np.linspace(5,10,3)
+        XYZ = Utils.ndgrid(x,x,np.r_[0.])
+        txLoc = np.r_[0,0,0.]
+        rxList0 = Survey.BaseRx(XYZ, 'exi')
+        Tx0 = Survey.BaseTx(txLoc, 'VMD', [rxList0])
+        rxList1 = Survey.BaseRx(XYZ, 'bxi')
+        Tx1 = Survey.BaseTx(txLoc, 'VMD', [rxList1])
+        rxList2 = Survey.BaseRx(XYZ, 'bxi')
+        Tx2 = Survey.BaseTx(txLoc, 'VMD', [rxList2])
+        rxList3 = Survey.BaseRx(XYZ, 'bxi')
+        Tx3 = Survey.BaseTx(txLoc, 'VMD', [rxList3])
+        Tx4 = Survey.BaseTx(txLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3])
+        txList = [Tx0,Tx1,Tx2,Tx3,Tx4]
+        survey = Survey.BaseSurvey(txList=txList)
+        prob = Problem.BaseTimeProblem(mesh, timeSteps=[(10.,3), (20.,2)])
+        survey.pair(prob)
+        self.F = Survey.TimeFields(mesh, survey, knownFields={'phi':'CC','e':'E','b':'F'})
+        self.Tx0 = Tx0
+        self.Tx1 = Tx1
+        self.mesh = mesh
+        self.XYZ = XYZ
+
+    def test_SetGet(self):
+        F = self.F
+        nTx = F.survey.nTx
+        nT = F.survey.prob.nT
+        e = np.random.rand(F.mesh.nE, nTx, nT)
+        F[:, 'e'] = e
+        b = np.random.rand(F.mesh.nF, nTx, nT)
+        F[:, 'b'] = b
+
+        self.assertTrue(np.all(F[:, 'e'] == e))
+        self.assertTrue(np.all(F[:, 'b'] == b))
+        F[:] = {'b':b,'e':e}
+        self.assertTrue(np.all(F[:, 'e'] == e))
+        self.assertTrue(np.all(F[:, 'b'] == b))
+
+        b = np.random.rand(F.mesh.nF,nT)
+        F[self.Tx0, 'b'] = b
+        self.assertTrue(np.all(F[self.Tx0, 'b'] == b))
+
+        phi = np.random.rand(F.mesh.nC,2,nT)
+        F[[self.Tx0,self.Tx1], 'phi'] = phi
+        self.assertTrue(np.all(F[[self.Tx0,self.Tx1], 'phi'] == phi))
+
+        fdict = F[:]
+        self.assertTrue(type(fdict) is dict)
+        self.assertTrue(sorted([k for k in fdict]) == ['b','e','phi'])
+
+        b = np.random.rand(F.mesh.nF, 2, nT)
+        F[[self.Tx0, self.Tx1],'b'] = b
+        self.assertTrue(F[self.Tx0]['b'].shape == (F.mesh.nF,nT))
+        self.assertTrue(F[self.Tx0,'b'].shape == (F.mesh.nF,nT))
+        self.assertTrue(np.all(F[self.Tx0,'b'] == b[:,0,:]))
+        self.assertTrue(np.all(F[self.Tx1,'b'] == b[:,1,:]))
+        self.assertTrue(np.all(F[self.Tx0,'b',1] == b[:,0,1]))
+        self.assertTrue(np.all(F[self.Tx1,'b',1] == b[:,1,1]))
+        self.assertTrue(np.all(F[self.Tx0,'b',4] == b[:,0,4]))
+        self.assertTrue(np.all(F[self.Tx1,'b',4] == b[:,1,4]))
+
+    def test_assertions(self):
+        freq = [self.Tx0, self.Tx1]
+        bWrongSize = np.random.rand(self.F.mesh.nE, self.F.survey.nTx)
+        def fun(): self.F[freq, 'b'] = bWrongSize
+        self.assertRaises(ValueError, fun)
+        def fun(): self.F[-999.]
+        self.assertRaises(KeyError, fun)
+        def fun(): self.F['notRight']
+        self.assertRaises(KeyError, fun)
+        def fun(): self.F[freq,'notThere']
+        self.assertRaises(KeyError, fun)
+
 if __name__ == '__main__':
    unittest.main()