Slight tweak to DiskSCF test to hit Sigma input case

jobovy · Dec 29, 2016 · 66ac0e6 · 66ac0e6
1 parent 9542b11
commit 66ac0e6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/nose/test_potential.py b/nose/test_potential.py
@@ -1817,7 +1817,10 @@ def test_DiskSCFPotential_againstDoubleExp():
     # DoubleExponentialDiskPotential
     dp= potential.DoubleExponentialDiskPotential(amp=13.5,hr=1./3.,hz=1./27.)
     dscfp= potential.DiskSCFPotential(dens=lambda R,z: dp.dens(R,z),
-                                      Sigma={'type':'exp','h':1./3.,'amp':1.},
+                                      Sigma_amp=1.,
+                                      Sigma=lambda R: numpy.exp(-3.*R),
+                                      dSigmadR=lambda R: -3.*numpy.exp(-3.*R),
+                                      d2SigmadR2=lambda R: 9.*numpy.exp(-3.*R),
                                       hz={'type':'exp','h':1./27.},
                                       a=1.,N=10,L=10)
     testRs= numpy.linspace(0.3,1.5,101)