summaryrefslogtreecommitdiff
path: root/lib/test/test_skyline.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/test/test_skyline.py')
-rw-r--r--lib/test/test_skyline.py34
1 files changed, 34 insertions, 0 deletions
diff --git a/lib/test/test_skyline.py b/lib/test/test_skyline.py
new file mode 100644
index 0000000..d450e49
--- /dev/null
+++ b/lib/test/test_skyline.py
@@ -0,0 +1,34 @@
+"""SkyLine tests.
+
+:Author: Pey Lian Lim
+
+:Organization: Space Telescope Science Institute
+
+"""
+from __future__ import absolute_import
+
+import pyfits
+from numpy.testing import assert_almost_equal, assert_array_less
+
+from .. import skyline
+
+from .test_util import ROOT_DIR
+from .test_shared import resolve_imagename
+
+def test_union_simple():
+ # Two similar exposures with slight offset and some rotation.
+ im1 = resolve_imagename(ROOT_DIR, '2chipA.fits')
+ im2 = resolve_imagename(ROOT_DIR, '2chipB.fits')
+
+ skyline1 = skyline.SkyLine(im1)
+ skyline2 = skyline.SkyLine(im2)
+
+ union_1_2 = skyline1.union(skyline2)
+ union_2_1 = skyline2.union(skyline1)
+
+ assert_almost_equal(union_1_2.area(), union_2_1.area())
+
+ for m in union_1_2.members:
+ assert m in union_2_1.members
+
+ assert len(union_1_2.members) == 4