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

import numpy as np 

import unittest 

import healpy as hp 

import lsst.sims.utils as utils 

import lsst.utils.tests 

 

 

def setup_module(module): 

lsst.utils.tests.init() 

 

 

class TestHealUtils(unittest.TestCase): 

 

def testRaDecsRad(self): 

""" 

Test that the Ra Dec conversions round-trip 

""" 

 

nside = 64 

hpids = np.arange(hp.nside2npix(nside)) 

ra, dec = utils._hpid2RaDec(nside, hpids) 

 

hpids_return = utils._raDec2Hpid(nside, ra, dec) 

 

np.testing.assert_array_equal(hpids, hpids_return) 

 

def testRaDecsDeg(self): 

""" 

Test that the Ra Dec conversions round-trip 

""" 

 

nside = 64 

hpids = np.arange(hp.nside2npix(nside)) 

ra, dec = utils.hpid2RaDec(nside, hpids) 

 

hpids_return = utils.raDec2Hpid(nside, ra, dec) 

 

np.testing.assert_array_equal(hpids, hpids_return) 

 

def testBinRad(self): 

""" 

Test that healbin returns correct values and valid healpy maps. 

""" 

 

ra = np.zeros(3) 

dec = np.zeros(3) 

values = ra * 0. + 1. 

 

nside = 128 

hpid = utils._raDec2Hpid(nside, ra[0], dec[0]) 

 

map1 = utils._healbin(ra, dec, values, nside=nside) 

self.assertEqual(map1[hpid], 1.) 

self.assertEqual(hp.maptype(map1), 0) 

map2 = utils._healbin(ra, dec, values, nside=nside, reduceFunc=np.sum) 

self.assertEqual(map2[hpid], 3.) 

self.assertEqual(hp.maptype(map2), 0) 

map3 = utils._healbin(ra, dec, values, nside=nside, reduceFunc=np.std) 

self.assertEqual(map3[hpid], 0.) 

self.assertEqual(hp.maptype(map3), 0) 

 

def testBinDeg(self): 

""" 

Test that healbin returns correct values and valid healpy maps. 

""" 

 

ra = np.zeros(3) 

dec = np.zeros(3) 

values = ra * 0. + 1. 

 

nside = 128 

hpid = utils.raDec2Hpid(nside, ra[0], dec[0]) 

 

map1 = utils.healbin(ra, dec, values, nside=nside) 

self.assertEqual(map1[hpid], 1.) 

self.assertEqual(hp.maptype(map1), 0) 

map2 = utils.healbin(ra, dec, values, nside=nside, reduceFunc=np.sum) 

self.assertEqual(map2[hpid], 3.) 

self.assertEqual(hp.maptype(map2), 0) 

map3 = utils.healbin(ra, dec, values, nside=nside, reduceFunc=np.std) 

self.assertEqual(map3[hpid], 0.) 

self.assertEqual(hp.maptype(map3), 0) 

 

 

class MemoryTestClass(lsst.utils.tests.MemoryTestCase): 

pass 

 

88 ↛ 89line 88 didn't jump to line 89, because the condition on line 88 was never trueif __name__ == "__main__": 

lsst.utils.tests.init() 

unittest.main()