diff --git a/SimPEG/Survey.py b/SimPEG/Survey.py index 10e4c76f..29bdb452 100644 --- a/SimPEG/Survey.py +++ b/SimPEG/Survey.py @@ -119,14 +119,12 @@ class BaseSrc(object): rxList = None #: SimPEG Receiver List rxPair = BaseRx - def __init__(self, loc, srcType, rxList, **kwargs): + def __init__(self, rxList, **kwargs): assert type(rxList) is list, 'rxList must be a list' for rx in rxList: assert isinstance(rx, self.rxPair), 'rxList must be a %s'%self.rxPair.__name__ assert len(set(rxList)) == len(rxList), 'The rxList must be unique' - self.loc = loc - self.srcType = srcType self.rxList = rxList Utils.setKwargs(self, **kwargs) diff --git a/SimPEG/Tests/test_Survey.py b/SimPEG/Tests/test_Survey.py index f93395ce..f5109b71 100644 --- a/SimPEG/Tests/test_Survey.py +++ b/SimPEG/Tests/test_Survey.py @@ -9,14 +9,14 @@ class DataAndFieldsTest(unittest.TestCase): XYZ = Utils.ndgrid(x,x,np.r_[0.]) srcLoc = np.r_[0,0,0.] rxList0 = Survey.BaseRx(XYZ, 'exi') - Src0 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0]) + Src0 = Survey.BaseSrc([rxList0], loc=srcLoc) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc([rxList1], loc=srcLoc) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc([rxList2], loc=srcLoc) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc([rxList3], loc=srcLoc) + Src4 = Survey.BaseSrc([rxList0, rxList1, rxList2, rxList3], loc=srcLoc) srcList = [Src0,Src1,Src2,Src3,Src4] survey = Survey.BaseSurvey(srcList=srcList) self.D = Survey.Data(survey) @@ -122,14 +122,14 @@ class FieldsTest_Alias(unittest.TestCase): XYZ = Utils.ndgrid(x,x,np.r_[0.]) srcLoc = np.r_[0,0,0.] rxList0 = Survey.BaseRx(XYZ, 'exi') - Src0 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0]) + Src0 = Survey.BaseSrc([rxList0],loc=srcLoc) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc([rxList1],loc=srcLoc) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc([rxList2],loc=srcLoc) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc([rxList3],loc=srcLoc) + Src4 = Survey.BaseSrc([rxList0, rxList1, rxList2, rxList3],loc=srcLoc) srcList = [Src0,Src1,Src2,Src3,Src4] survey = Survey.BaseSurvey(srcList=srcList) self.D = Survey.Data(survey) @@ -193,14 +193,14 @@ class FieldsTest_Time(unittest.TestCase): XYZ = Utils.ndgrid(x,x,np.r_[0.]) srcLoc = np.r_[0,0,0.] rxList0 = Survey.BaseRx(XYZ, 'exi') - Src0 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0]) + Src0 = Survey.BaseSrc([rxList0], loc=srcLoc) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc([rxList1], loc=srcLoc) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc([rxList2], loc=srcLoc) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc([rxList3], loc=srcLoc) + Src4 = Survey.BaseSrc([rxList0, rxList1, rxList2, rxList3], loc=srcLoc) srcList = [Src0,Src1,Src2,Src3,Src4] survey = Survey.BaseSurvey(srcList=srcList) prob = Problem.BaseTimeProblem(mesh, timeSteps=[(10.,3), (20.,2)]) @@ -295,14 +295,14 @@ class FieldsTest_Time_Aliased(unittest.TestCase): XYZ = Utils.ndgrid(x,x,np.r_[0.]) srcLoc = np.r_[0,0,0.] rxList0 = Survey.BaseRx(XYZ, 'exi') - Src0 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0]) + Src0 = Survey.BaseSrc( [rxList0],loc=srcLoc) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc( [rxList1],loc=srcLoc) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc( [rxList2],loc=srcLoc) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc( [rxList3],loc=srcLoc) + Src4 = Survey.BaseSrc( [rxList0, rxList1, rxList2, rxList3],loc=srcLoc) srcList = [Src0,Src1,Src2,Src3,Src4] survey = Survey.BaseSurvey(srcList=srcList) prob = Problem.BaseTimeProblem(mesh, timeSteps=[(10.,3), (20.,2)])