From 618f803ffa66acae3874f72ebd1db3924d81415c Mon Sep 17 00:00:00 2001 From: Lindsey Date: Fri, 8 May 2015 10:01:28 -0700 Subject: [PATCH] Fields returns matrices --- SimPEG/Tests/test_Survey.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/SimPEG/Tests/test_Survey.py b/SimPEG/Tests/test_Survey.py index e6a9e507..fd3dbf06 100644 --- a/SimPEG/Tests/test_Survey.py +++ b/SimPEG/Tests/test_Survey.py @@ -98,8 +98,8 @@ class DataAndFieldsTest(unittest.TestCase): F[[self.Src0, self.Src1],'b'] = b self.assertTrue(F[self.Src0]['b'].shape == (F.mesh.nF,1)) self.assertTrue(F[self.Src0,'b'].shape == (F.mesh.nF,1)) - self.assertTrue(np.all(F[self.Src0,'b'] == b[:,0])) - self.assertTrue(np.all(F[self.Src1,'b'] == b[:,1])) + self.assertTrue(np.all(F[self.Src0,'b'] == Utils.mkvc(b[:,0],2))) + self.assertTrue(np.all(F[self.Src1,'b'] == Utils.mkvc(b[:,1],2))) def test_assertions(self): freq = [self.Src0, self.Src1]