fix overflow

segasai · Jan 16, 2024 · a8f87c8 · a8f87c8
1 parent d9390c1
commit a8f87c8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/test_fit.py b/tests/test_fit.py
@@ -1,4 +1,5 @@
 import os
+
 os.environ['OMP_NUM_THREADS'] = '1'
 import sys
 import pytest
@@ -10,8 +11,10 @@
 from rvspecfit import vel_fit
 from rvspecfit import spec_fit
 from rvspecfit import fitter_ccf
+
 path = str(pathlib.Path(__file__).parent.absolute())
 
+
 def test_config():
     with pytest.raises(RuntimeError):
         config = utils.read_config('xx.yaml')
@@ -22,14 +25,15 @@ def test_config():
     finally:
         os.chdir(cwd)
 
+
 def test_fit():
     config = utils.read_config(path + '/config.yaml')
 
     # read data
     dat = pyfits.getdata(path + '/data/spec-0266-51602-0031.fits')
     err = dat['ivar']
     err = 1. / err**.5
-    err[~np.isfinite(err)] = 1e40
+    err[~np.isfinite(err)] = 1e30
 
     # construct specdata object
     specdata = [

diff --git a/tests/test_sdss.py b/tests/test_sdss.py
@@ -24,7 +24,7 @@ def test_fits():
     dat = pyfits.getdata(path + '/data/spec-0266-51602-0031.fits')
     err = dat['ivar']
     err = 1. / err**.5
-    err[~np.isfinite(err)] = 1e40
+    err[~np.isfinite(err)] = 1e30
 
     # construct specdata object
     specdata = [