summaryrefslogtreecommitdiff
path: root/stwcs/tests
diff options
context:
space:
mode:
Diffstat (limited to 'stwcs/tests')
-rw-r--r--stwcs/tests/test_altwcs.py51
-rw-r--r--stwcs/tests/test_headerlet.py8
-rw-r--r--stwcs/tests/test_updatewcs.py12
3 files changed, 40 insertions, 31 deletions
diff --git a/stwcs/tests/test_altwcs.py b/stwcs/tests/test_altwcs.py
index 86d100f..d646106 100644
--- a/stwcs/tests/test_altwcs.py
+++ b/stwcs/tests/test_altwcs.py
@@ -1,11 +1,14 @@
+from __future__ import absolute_import, division, print_function
+
import shutil
import os
from astropy.io import fits as pyfits
-from stwcs.wcsutil import altwcs
-from stwcs import updatewcs
-from stwcs.wcsutil import HSTWCS
+from ..wcsutil import altwcs
+from .. import updatewcs
+from .. import wcsutil
import numpy as np
from numpy.testing import utils
+import pytest
from . import data
data_path = os.path.split(os.path.abspath(data.__file__))[0]
@@ -70,25 +73,25 @@ class TestAltWCS(object):
updatewcs.updatewcs(acs_file)
self.acs_file = acs_file
self.simplefits = simple_file
- self.ww = HSTWCS(self.acs_file, ext=1)
+ self.ww = wcsutil.HSTWCS(self.acs_file, ext=1)
def test_archive(self):
altwcs.archiveWCS(self.acs_file, ext=1, wcskey='Z', wcsname='ZTEST', reusekey=False)
- w1 = HSTWCS(self.acs_file, ext=1)
- w1z = HSTWCS(self.acs_file, ext=1, wcskey='Z')
+ w1 = wcsutil.HSTWCS(self.acs_file, ext=1)
+ w1z = wcsutil.HSTWCS(self.acs_file, ext=1, wcskey='Z')
compare_wcs(w1, w1z)
def test_archive_clobber(self):
altwcs.archiveWCS(self.acs_file, ext=1, wcskey='Z', wcsname='ZTEST', reusekey=True)
- w1 = HSTWCS(self.acs_file, ext=1)
- w1z = HSTWCS(self.acs_file, ext=1, wcskey='Z')
+ w1 = wcsutil.HSTWCS(self.acs_file, ext=1)
+ w1z = wcsutil.HSTWCS(self.acs_file, ext=1, wcskey='Z')
compare_wcs(w1, w1z)
def test_restore_wcs(self):
# test restore on a file
altwcs.restoreWCS(self.acs_file, ext=1, wcskey='O')
- w1o = HSTWCS(self.acs_file, ext=1, wcskey='O')
- w1 = HSTWCS(self.acs_file, ext=1)
+ w1o = wcsutil.HSTWCS(self.acs_file, ext=1, wcskey='O')
+ w1 = wcsutil.HSTWCS(self.acs_file, ext=1)
compare_wcs(w1, w1o, exclude_keywords=['ctype'])
def test_restore_wcs_mem(self):
@@ -99,8 +102,8 @@ class TestAltWCS(object):
f = pyfits.open(self.acs_file, mode='update')
altwcs.restoreWCS(f, ext=1, wcskey='T')
f.close()
- w1o = HSTWCS(self.acs_file, ext=1, wcskey='T')
- w1 = HSTWCS(self.acs_file, ext=1)
+ w1o = wcsutil.HSTWCS(self.acs_file, ext=1, wcskey='T')
+ w1 = wcsutil.HSTWCS(self.acs_file, ext=1)
compare_wcs(w1, w1o)
def test_restore_simple(self):
@@ -108,8 +111,8 @@ class TestAltWCS(object):
altwcs.archiveWCS(self.simplefits, ext=0, wcskey='R')
pyfits.setval(self.simplefits, ext=0, keyword='CRVAL1R', value=1)
altwcs.restoreWCS(self.simplefits, ext=0, wcskey='R')
- wo = HSTWCS(self.simplefits, ext=0, wcskey='R')
- ws = HSTWCS(self.simplefits, ext=0)
+ wo = wcsutil.HSTWCS(self.simplefits, ext=0, wcskey='R')
+ ws = wcsutil.HSTWCS(self.simplefits, ext=0)
compare_wcs(ws, wo)
def test_restore_wcs_from_to(self):
@@ -121,25 +124,27 @@ class TestAltWCS(object):
altwcs.restore_from_to(f, fromext='SCI', toext=['SCI', 'ERR', 'DQ'],
wcskey='T')
f.close()
- w1o = HSTWCS(self.acs_file, ext=('SCI', 1), wcskey='T')
- w1 = HSTWCS(self.acs_file, ext=('SCI', 1))
+ w1o = wcsutil.HSTWCS(self.acs_file, ext=('SCI', 1), wcskey='T')
+ w1 = wcsutil.HSTWCS(self.acs_file, ext=('SCI', 1))
compare_wcs(w1, w1o)
- w2 = HSTWCS(self.acs_file, ext=('ERR', 1))
+ w2 = wcsutil.HSTWCS(self.acs_file, ext=('ERR', 1))
compare_wcs(w2, w1o, exclude_keywords=['ctype'])
- w3 = HSTWCS(self.acs_file, ext=('DQ', 1))
+ w3 = wcsutil.HSTWCS(self.acs_file, ext=('DQ', 1))
compare_wcs(w3, w1o, exclude_keywords=['ctype'])
- w4o = HSTWCS(self.acs_file, ext=4, wcskey='T')
- w4 = HSTWCS(self.acs_file, ext=('SCI', 2))
+ w4o = wcsutil.HSTWCS(self.acs_file, ext=4, wcskey='T')
+ w4 = wcsutil.HSTWCS(self.acs_file, ext=('SCI', 2))
compare_wcs(w4, w4o)
- w5 = HSTWCS(self.acs_file, ext=('ERR', 2))
+ w5 = wcsutil.HSTWCS(self.acs_file, ext=('ERR', 2))
compare_wcs(w5, w4o, exclude_keywords=['ctype'])
- w6 = HSTWCS(self.acs_file, ext=('DQ', 2))
+ w6 = wcsutil.HSTWCS(self.acs_file, ext=('DQ', 2))
compare_wcs(w3, w1o, exclude_keywords=['ctype'])
def test_delete_wcs(self):
#altwcs.archiveWCS(self.acs_file, ext=1, wcskey='Z')
altwcs.deleteWCS(self.acs_file, ext=1, wcskey='Z')
- utils.assert_raises(KeyError, HSTWCS, self.acs_file, ext=1, wcskey='Z')
+ #utils.assert_raises(KeyError, wcsutil.HSTWCS, self.acs_file, ext=1, wcskey='Z')
+ with pytest.raises(KeyError):
+ wcsutil.HSTWCS(self.acs_file, ext=1, wcskey='Z')
def test_pars_file_mode1(self):
assert(not altwcs._parpasscheck(self.acs_file, ext=1, wcskey='Z'))
diff --git a/stwcs/tests/test_headerlet.py b/stwcs/tests/test_headerlet.py
index 48fb470..f4fa050 100644
--- a/stwcs/tests/test_headerlet.py
+++ b/stwcs/tests/test_headerlet.py
@@ -1,9 +1,11 @@
+from __future__ import absolute_import, division, print_function
+
import shutil
import os
from astropy.io import fits
-from stwcs import updatewcs
-from stwcs.wcsutil import headerlet, wcsdiff
-from stwcs.wcsutil import HSTWCS
+from .. import updatewcs
+from ..wcsutil import headerlet, wcsdiff
+from ..wcsutil import HSTWCS
import numpy as np
from numpy.testing import utils
from nose.tools import *
diff --git a/stwcs/tests/test_updatewcs.py b/stwcs/tests/test_updatewcs.py
index dc5a34f..b7e75c8 100644
--- a/stwcs/tests/test_updatewcs.py
+++ b/stwcs/tests/test_updatewcs.py
@@ -1,12 +1,14 @@
+from __future__ import absolute_import, division, print_function
+
import shutil
import os
from astropy import wcs
from astropy.io import fits
-from stwcs import updatewcs
-from stwcs.updatewcs import apply_corrections
-from stwcs.distortion import utils as dutils
-from stwcs.wcsutil import HSTWCS
+from .. import updatewcs
+from ..updatewcs import apply_corrections
+from ..distortion import utils as dutils
+from ..wcsutil import HSTWCS
import numpy as np
from numpy.testing import utils
import pytest
@@ -111,7 +113,7 @@ class TestStwcs(object):
#print('outwcs.wcs.crval = {0}'.format(outwcs.wcs.crval))
utils.assert_allclose(
- outwcs.wcs.crval, np.array([5.65109952, -72.0674181]), rtol=1e-7)
+ outwcs.wcs.crval, np.array([5.65109952, -72.0674181]), atol=1e-7)
utils.assert_almost_equal(outwcs.wcs.crpix, np.array([2107.0, 2118.5]))
utils.assert_almost_equal(
generated by cgit (git 2.34.1) at 2025-05-08 10:25:57 -0400