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

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

""" 

This unit test essentially copies the unit test of the same name that 

exists in sims_utils. However, in this iteration, the test to verify 

that a mapped filename exists is never skipped. The idea being that we need 

to make sure that we have loaded the correct version of sims_sed_library. 

 

In the future, we will replace this test with a version requirement on 

sims_sed_library in sims_catUtils' ups table. Right now, we are keeping 

this test as an extra-paranoid verification that the correct version of 

sims_sed_library was loaded. 

""" 

 

import os 

import unittest 

import lsst.utils.tests 

 

from lsst.utils import getPackageDir 

from lsst.sims.utils import defaultSpecMap 

 

 

def setup_module(module): 

lsst.utils.tests.init() 

 

 

class FileMapTest(unittest.TestCase): 

 

def setUp(self): 

self.root_dir = getPackageDir('sims_sed_library') 

 

def verifyFile(self, file_name, dir_name): 

""" 

Verify that specMape[file_name] results in a file in dir dir_name 

""" 

test_name = defaultSpecMap[file_name] 

control_name = os.path.join(dir_name, file_name+'.gz') 

msg = '%s should map to %s; it actually maps to %s' % (file_name, control_name, test_name) 

self.assertEqual(test_name, control_name, msg=msg) 

 

add_space = file_name+' ' 

self.assertNotEqual(add_space, file_name) 

test_name = defaultSpecMap[add_space] 

msg = '%s should map to %s; it actually maps to %s' % (add_space, control_name, test_name) 

self.assertEqual(test_name, control_name, msg=msg) 

 

add_space = ' '+file_name 

self.assertNotEqual(add_space, file_name) 

test_name = defaultSpecMap[add_space] 

msg = '%s should map to %s; it actually maps to %s' % (add_space, control_name, test_name) 

self.assertEqual(test_name, control_name, msg=msg) 

 

add_gz = file_name+'.gz' 

self.assertNotEqual(add_gz, file_name) 

test_name = defaultSpecMap[add_gz] 

msg = '%s should map to %s; it actually maps to %s' % (add_gz, control_name, test_name) 

self.assertEqual(test_name, control_name, msg=msg) 

 

full_path = os.path.join(self.root_dir, test_name) 

msg = '%s does not exist; it should' % full_path 

self.assertTrue(os.path.exists(full_path), msg=msg) 

 

def testMLT(self): 

""" 

Test that defaultSpecMap correctly locates MLT dwarf spectra 

""" 

self.verifyFile('lte004-3.5-0.0a+0.0.BT-Settl.spec', 'starSED/mlt') 

 

def test_m_spec(self): 

""" 

Test that defaultSpecMap correctly finds old MLT dwarf spectra 

that begin with 'm' 

""" 

self.verifyFile('m5.1Full.dat', 'starSED/old_mlt') 

 

def test_l4_spec(self): 

""" 

Test that defaultSpecMap correctly finds l4Full.dat 

""" 

self.verifyFile('l4Full.dat', 'starSED/old_mlt') 

 

def test_L_spec(self): 

""" 

Test that defaultSpecMap correctly find the L#_# spectra 

""" 

self.verifyFile('L2_0Full.dat', 'starSED/old_mlt') 

 

def test_burrows_spec(self): 

""" 

Test that defaultSpecMap correctly find the burrows spectra 

""" 

self.verifyFile('burrows+2006c91.21_T1400_g5.5_cf_0.3X', 'starSED/old_mlt') 

 

def testBergeron(self): 

""" 

Test that defaultSpecMap correctly locates the bergeron spectra 

""" 

self.verifyFile('bergeron_4750_85.dat_4900', 'starSED/wDs') 

 

def testKurucz(self): 

""" 

Test that defaultSpecMap correctly locates the kurucz spectra 

""" 

self.verifyFile('km30_5000.fits_g10_5040', 'starSED/kurucz') 

self.verifyFile('kp10_9000.fits_g40_9100', 'starSED/kurucz') 

 

def testGalaxy(self): 

""" 

Test that defaultSpecMap correctly locates the galaxy SEDs 

""" 

self.verifyFile('Const.79E06.002Z.spec', 'galaxySED') 

self.verifyFile('Inst.79E06.02Z.spec', 'galaxySED') 

self.verifyFile('Exp.40E08.02Z.spec', 'galaxySED') 

self.verifyFile('Burst.40E08.002Z.spec', 'galaxySED') 

 

 

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

pass 

 

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

lsst.utils.tests.init() 

unittest.main()