diff --git a/SimPEG/Survey.py b/SimPEG/Survey.py index 10e4c76f..05a2ad37 100644 --- a/SimPEG/Survey.py +++ b/SimPEG/Survey.py @@ -119,14 +119,13 @@ class BaseSrc(object): rxList = None #: SimPEG Receiver List rxPair = BaseRx - def __init__(self, loc, srcType, rxList, **kwargs): + def __init__(self, loc, 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..7372a686 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(srcLoc, [rxList0]) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc(srcLoc, [rxList1]) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc(srcLoc, [rxList2]) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc(srcLoc, [rxList3]) + Src4 = Survey.BaseSrc(srcLoc, [rxList0, rxList1, rxList2, rxList3]) 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(srcLoc, [rxList0]) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc(srcLoc, [rxList1]) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc(srcLoc, [rxList2]) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc(srcLoc, [rxList3]) + Src4 = Survey.BaseSrc(srcLoc, [rxList0, rxList1, rxList2, rxList3]) 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(srcLoc, [rxList0]) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc(srcLoc, [rxList1]) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc(srcLoc, [rxList2]) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc(srcLoc, [rxList3]) + Src4 = Survey.BaseSrc(srcLoc, [rxList0, rxList1, rxList2, rxList3]) 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(srcLoc, [rxList0]) rxList1 = Survey.BaseRx(XYZ, 'bxi') - Src1 = Survey.BaseSrc(srcLoc, 'VMD', [rxList1]) + Src1 = Survey.BaseSrc(srcLoc, [rxList1]) rxList2 = Survey.BaseRx(XYZ, 'bxi') - Src2 = Survey.BaseSrc(srcLoc, 'VMD', [rxList2]) + Src2 = Survey.BaseSrc(srcLoc, [rxList2]) rxList3 = Survey.BaseRx(XYZ, 'bxi') - Src3 = Survey.BaseSrc(srcLoc, 'VMD', [rxList3]) - Src4 = Survey.BaseSrc(srcLoc, 'VMD', [rxList0, rxList1, rxList2, rxList3]) + Src3 = Survey.BaseSrc(srcLoc, [rxList3]) + Src4 = Survey.BaseSrc(srcLoc, [rxList0, rxList1, rxList2, rxList3]) srcList = [Src0,Src1,Src2,Src3,Src4] survey = Survey.BaseSurvey(srcList=srcList) prob = Problem.BaseTimeProblem(mesh, timeSteps=[(10.,3), (20.,2)])