Hide keyboard shortcuts

Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

# 

# LSST Data Management System 

# See COPYRIGHT file at the top of the source tree. 

# 

# This product includes software developed by the 

# LSST Project (http://www.lsst.org/). 

# 

# This program is free software: you can redistribute it and/or modify 

# it under the terms of the GNU General Public License as published by 

# the Free Software Foundation, either version 3 of the License, or 

# (at your option) any later version. 

# 

# This program is distributed in the hope that it will be useful, 

# but WITHOUT ANY WARRANTY; without even the implied warranty of 

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

# GNU General Public License for more details. 

# 

# You should have received a copy of the LSST License Statement and 

# the GNU General Public License along with this program. If not, 

# see <https://www.lsstcorp.org/LegalNotices/>. 

# 

 

import pickle 

import unittest 

 

from lsst.sphgeom import (Angle, Circle, Mq3cPixelization, RangeSet, 

UnitVector3d) 

 

 

class Mq3cPixelizationTestCase(unittest.TestCase): 

 

def test_construction(self): 

with self.assertRaises(ValueError): 

Mq3cPixelization(-1) 

with self.assertRaises(ValueError): 

Mq3cPixelization(Mq3cPixelization.MAX_LEVEL + 1) 

m1 = Mq3cPixelization(0) 

self.assertEqual(m1.getLevel(), 0) 

m2 = Mq3cPixelization(1) 

m3 = Mq3cPixelization(m2) 

self.assertNotEqual(m1, m2) 

self.assertEqual(m2, m3) 

 

def test_indexing(self): 

pixelization = Mq3cPixelization(1) 

self.assertEqual(pixelization.index(UnitVector3d(0.5, -0.5, 1.0)), 53) 

 

def test_level(self): 

self.assertEqual(Mq3cPixelization.level(0), -1) 

for level in range(Mq3cPixelization.MAX_LEVEL + 1): 

for root in range(8, 10): 

index = root * 4**level 

self.assertEqual(Mq3cPixelization.level(index), -1) 

for root in range(10, 16): 

index = root * 4**level 

self.assertEqual(Mq3cPixelization.level(index), level) 

 

def test_envelope_and_interior(self): 

pixelization = Mq3cPixelization(1) 

c = Circle(UnitVector3d(1.0, -0.5, -0.5), Angle.fromDegrees(0.1)) 

rs = pixelization.envelope(c) 

self.assertTrue(rs == RangeSet(44)) 

rs = pixelization.envelope(c, 1) 

self.assertTrue(rs == RangeSet(44)) 

self.assertTrue(rs.isWithin(pixelization.universe())) 

rs = pixelization.interior(c) 

self.assertTrue(rs.empty()) 

 

def test_index_to_string(self): 

strings = ['+X', '+Y', '+Z', '-X', '-Y', '-Z'] 

for i in range(6): 

s0 = strings[i] 

components = [0.0]*3 

components[i % 3] = 1.0 if i < 3 else -1.0 

v = UnitVector3d(*components) 

f = Mq3cPixelization(0).index(v) 

self.assertEqual(Mq3cPixelization.asString(f), s0) 

self.assertEqual(Mq3cPixelization(0).toString(f), s0) 

for j in range(4): 

s1 = s0 + str(j) 

self.assertEqual(Mq3cPixelization.asString(f*4 + j), s1) 

self.assertEqual(Mq3cPixelization(1).toString(f*4 + j), s1) 

 

def test_string(self): 

p = Mq3cPixelization(3) 

self.assertEqual(str(p), 'Mq3cPixelization(3)') 

self.assertEqual(str(p), repr(p)) 

self.assertEqual( 

p, eval(repr(p), dict(Mq3cPixelization=Mq3cPixelization))) 

 

def test_pickle(self): 

a = Mq3cPixelization(20) 

b = pickle.loads(pickle.dumps(a)) 

self.assertEqual(a, b) 

 

 

97 ↛ 98line 97 didn't jump to line 98, because the condition on line 97 was never trueif __name__ == '__main__': 

unittest.main()