@@ -501,7 +501,7 @@ def test_multisrc_multirec(self):
501501 assert_allclose (out0f , out1f )
502502 assert_allclose (out0t , out1t )
503503
504- def test_cole_cole (self ):
504+ def test_cole_cole (self , capsys ):
505505 # Check user-hook for eta/zeta
506506
507507 def func_eta (inp , pdict ):
@@ -518,25 +518,35 @@ def func_zeta(inp, pdict):
518518
519519 return etaH , etaV
520520
521- model = {'src' : [0 , 0 , 500 , 0 , 0 ], 'rec' : [500 , 0 , 600 , 0 , 0 ],
522- 'depth' : [0 , 550 ], 'freqtime' : [0.1 , 1 , 10 ]}
523- res = np .array ([2 , 10 , 5 ])
524- fact = np .array ([2 , 2 , 2 ])
525- eta = {'res' : fact * res , 'fact' : fact , 'func_eta' : func_eta }
526- zeta = {'res' : res , 'fact' : fact , 'func_zeta' : func_zeta }
527-
528- # Frequency domain
529- standard = bipole (res = res , ** model )
530- outeta = bipole (res = eta , ** model )
531- assert_allclose (standard , outeta )
532- outzeta = bipole (res = zeta , mpermH = fact , mpermV = fact , ** model )
533- assert_allclose (standard , outzeta )
534- # Time domain
535- standard = bipole (res = res , signal = 0 , ** model )
536- outeta = bipole (res = eta , signal = 0 , ** model )
537- assert_allclose (standard , outeta )
538- outzeta = bipole (res = zeta , signal = 0 , mpermH = fact , mpermV = fact , ** model )
539- assert_allclose (standard , outzeta )
521+ for rev in [1 , - 1 ]:
522+ model = {
523+ 'src' : [0 , 0 , 500 , 0 , 0 ], 'rec' : [500 , 0 , 600 , 0 , 0 ],
524+ 'depth' : [0 , rev * 550 ], 'freqtime' : [0.1 , 1 , 10 ], 'verb' : 3 ,
525+ }
526+ res = np .array ([2 , 10 , 5 ])
527+ fact = np .array ([1.5 , 2 , 2.5 ])
528+ eta = {'res' : fact * res , 'fact' : fact , 'func_eta' : func_eta }
529+ zeta = {'res' : res , 'fact' : fact , 'func_zeta' : func_zeta }
530+
531+ # Frequency domain
532+ standard = bipole (res = res , ** model )
533+ _ , _ = capsys .readouterr ()
534+ outeta = bipole (res = eta , ** model )
535+ out , _ = capsys .readouterr ()
536+ if rev == 1 :
537+ assert "fact : 1.5 2 2.5" in out
538+ else :
539+ assert "fact : 2.5 2 1.5" in out
540+ assert_allclose (standard , outeta )
541+ outzeta = bipole (res = zeta , mpermH = fact , mpermV = fact , ** model )
542+ assert_allclose (standard , outzeta )
543+ # Time domain
544+ standard = bipole (res = res , signal = 0 , ** model )
545+ outeta = bipole (res = eta , signal = 0 , ** model )
546+ assert_allclose (standard , outeta )
547+ outzeta = bipole (
548+ res = zeta , signal = 0 , mpermH = fact , mpermV = fact , ** model )
549+ assert_allclose (standard , outzeta )
540550
541551 def test_src_rec_definitions (self ):
542552 inp = {'depth' : [0 , - 250 ], 'res' : [1e20 , 0.3 , 5 ], 'freqtime' : 1.23456 }
0 commit comments