Organize data object by rx as well as tx

2026-08-19 12:40:13 +08:00 · 2014-03-22 12:52:30 -07:00
parent aacbc03cf6
commit b7d47bcf38
3 changed files with 44 additions and 19 deletions
@@ -79,10 +79,11 @@ class FieldsTest(unittest.TestCase):
    def test_dataFDEM(self):
        V = []
        for tx in self.D.survey.txList:
-            v = np.random.rand(tx.nD)
-            V += [v]
-            self.D[tx] = v
-            self.assertTrue(np.all(v == self.D[tx]))
+            for rx in tx.rxList:
+                v = np.random.rand(rx.nD)
+                V += [v]
+                self.D[tx, rx] = v
+                self.assertTrue(np.all(v == self.D[tx, rx]))
        V = np.concatenate(V)
        self.assertTrue(np.all(V == Utils.mkvc(self.D)))