wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-02 03:34:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

removed SrcType

Browse Source
This commit is contained in:
Lindsey
2015-05-06 13:16:56 -07:00
parent 1a5dd2ae99
commit 428e34f631
2 changed files with 21 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Survey.py
+1 -2
View File
@@ -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)
SimPEG/Tests/test_Survey.py
+20 -20
View File
@@ -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)])