From 5caf2371212f27adea43b65a44db400bf6d8c27d Mon Sep 17 00:00:00 2001 From: Rowan Cockett Date: Mon, 1 Jun 2015 16:27:31 -0700 Subject: [PATCH] updates to the time fields side of things as well --- SimPEG/Fields.py | 7 ++----- SimPEG/Tests/test_Fields.py | 2 +- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/SimPEG/Fields.py b/SimPEG/Fields.py index d9da317d..9baa32f8 100644 --- a/SimPEG/Fields.py +++ b/SimPEG/Fields.py @@ -235,11 +235,8 @@ class TimeFields(Fields): pointerFields = pointerFields.reshape(pointerShape, order='F') timeII = np.arange(self.survey.prob.nT + 1)[timeInd] - srcII = np.array(self.survey.srcList)[srcInd] - if isinstance(srcII, np.ndarray): - srcII = srcII.tolist() - if len(srcII) == 1: - srcII = srcII[0] + srcII = np.array(self.survey.srcList)[srcInd] + srcII = srcII.tolist() if timeII.size == 1: pointerShapeDeflated = self._correctShape(alias, ind, deflate=True) diff --git a/SimPEG/Tests/test_Fields.py b/SimPEG/Tests/test_Fields.py index bd75d712..629189b8 100644 --- a/SimPEG/Tests/test_Fields.py +++ b/SimPEG/Tests/test_Fields.py @@ -343,7 +343,7 @@ class FieldsTest_Time_Aliased(unittest.TestCase): count = [0] def alias(e, srcInd, timeInd): count[0] += 1 - self.assertTrue(srcInd is self.Src0) + self.assertTrue(srcInd[0] is self.Src0) return self.F.mesh.edgeCurl * e F = Problem.TimeFields(self.F.mesh, self.F.survey, knownFields={'e':'E'}, aliasFields={'b':['e','F',alias]}) e = np.random.rand(F.mesh.nE,1,nT)