Coverage for tests/test_RangeSet.py: 16%
66 statements
« prev ^ index » next coverage.py v7.2.7, created at 2023-07-12 10:50 -0700
« prev ^ index » next coverage.py v7.2.7, created at 2023-07-12 10:50 -0700
1#
2# LSST Data Management System
3# See COPYRIGHT file at the top of the source tree.
4#
5# This product includes software developed by the
6# LSST Project (http://www.lsst.org/).
7#
8# This program is free software: you can redistribute it and/or modify
9# it under the terms of the GNU General Public License as published by
10# the Free Software Foundation, either version 3 of the License, or
11# (at your option) any later version.
12#
13# This program is distributed in the hope that it will be useful,
14# but WITHOUT ANY WARRANTY; without even the implied warranty of
15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16# GNU General Public License for more details.
17#
18# You should have received a copy of the LSST License Statement and
19# the GNU General Public License along with this program. If not,
20# see <https://www.lsstcorp.org/LegalNotices/>.
21#
23import pickle
24import sys
25import unittest
27from lsst.sphgeom import RangeSet
30class RangeSetTestCase(unittest.TestCase):
31 """Test RangeSet."""
33 def testConstruction(self):
34 s1 = RangeSet(1)
35 s2 = RangeSet()
36 s3 = RangeSet(2, 1)
37 s4 = RangeSet(s3)
38 self.assertTrue(s2.empty())
39 self.assertEqual(s3, s4)
40 self.assertEqual(s1, s3.complement())
42 def testComparisonOperators(self):
43 s1 = RangeSet(1)
44 s2 = RangeSet(2)
45 self.assertNotEqual(s1, s2)
46 s1.insert(2)
47 s2.insert(1)
48 self.assertEqual(s1, s2)
49 self.assertTrue(RangeSet(2, 1).contains(RangeSet(3, 4)))
50 self.assertTrue(RangeSet(2, 1).contains(3, 4))
51 self.assertTrue(RangeSet(2, 1).contains(3))
52 self.assertTrue(RangeSet(2, 4).isWithin(RangeSet(1, 5)))
53 self.assertTrue(RangeSet(2, 4).isWithin(1, 5))
54 self.assertFalse(RangeSet(2, 4).isWithin(3))
55 self.assertTrue(RangeSet(2, 4).intersects(RangeSet(3, 5)))
56 self.assertTrue(RangeSet(2, 4).intersects(3, 5))
57 self.assertTrue(RangeSet(2, 4).intersects(3))
58 self.assertTrue(RangeSet(2, 4).isDisjointFrom(RangeSet(6, 8)))
59 self.assertTrue(RangeSet(2, 4).isDisjointFrom(6, 8))
60 self.assertTrue(RangeSet(2, 4).isDisjointFrom(6))
62 def testSetOperators(self):
63 a = RangeSet(1)
64 b = ~a
65 self.assertTrue((a | b).full())
66 self.assertTrue((a & b).empty())
67 self.assertEqual(a - b, a)
68 self.assertEqual(b - a, b)
69 a &= a
70 b &= b
71 c = (a ^ b) - RangeSet(2, 4)
72 self.assertEqual(c, RangeSet(4, 2))
73 c |= b
74 self.assertTrue(c.full())
75 c ^= c
76 self.assertTrue(c.empty())
78 def testRanges(self):
79 s = RangeSet()
80 s.insert(0, 1)
81 s.insert(2, 3)
82 self.assertEqual(s.ranges(), [(0, 1), (2, 3)])
83 s = RangeSet(4, 2)
84 self.assertEqual(list(s), [(0, 2), (4, 0)])
86 def testString(self):
87 s = RangeSet(1, 10)
88 if sys.version_info[0] >= 3:
89 self.assertEqual(str(s), "[(1, 10)]")
90 self.assertEqual(repr(s), "RangeSet([(1, 10)])")
91 else:
92 # pybind11 maps C++ integers to Python long instances in Python 2.
93 self.assertEqual(str(s), "[(1L, 10L)]")
94 self.assertEqual(repr(s), "RangeSet([(1L, 10L)])")
95 self.assertEqual(s, eval(repr(s), {"RangeSet": RangeSet}))
97 def testPickle(self):
98 r = RangeSet([2, 3, 5, 7, 11, 13, 17, 19])
99 s = pickle.loads(pickle.dumps(r))
100 self.assertEqual(r, s)
103if __name__ == "__main__":
104 unittest.main()