1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import yaml 

7from transientNamer.utKit import utKit 

8from fundamentals import tools 

9from os.path import expanduser 

10home = expanduser("~") 

11 

12packageDirectory = utKit("").get_project_root() 

13settingsFile = packageDirectory + "/test_settings.yaml" 

14 

15su = tools( 

16 arguments={"settingsFile": settingsFile}, 

17 docString=__doc__, 

18 logLevel="DEBUG", 

19 options_first=False, 

20 projectName=None, 

21 defaultSettingsFile=False 

22) 

23arguments, settings, log, dbConn = su.setup() 

24 

25# SETUP PATHS TO COMMON DIRECTORIES FOR TEST DATA 

26moduleDirectory = os.path.dirname(__file__) 

27pathToInputDir = moduleDirectory + "/input/" 

28pathToOutputDir = moduleDirectory + "/output/" 

29 

30try: 

31 shutil.rmtree(pathToOutputDir) 

32except: 

33 pass 

34# COPY INPUT TO OUTPUT DIR 

35shutil.copytree(pathToInputDir, pathToOutputDir) 

36 

37# Recursively create missing directories 

38if not os.path.exists(pathToOutputDir): 

39 os.makedirs(pathToOutputDir) 

40 

41class test_search(unittest.TestCase): 

42 

43 def test_search_function01(self): 

44 kwargs = {} 

45 kwargs["log"] = log 

46 kwargs["settings"] = settings 

47 kwargs["ra"] = "06:50:36.74" 

48 kwargs["dec"] = "+31:06:44.7" 

49 kwargs["radiusArcsec"] = 5.0 

50 # xt-kwarg_key_and_value 

51 from transientNamer import search 

52 tns = search(**kwargs) 

53 print(tns.sources) 

54 print(tns.spectra) 

55 print(tns.photometry) 

56 print(tns.files) 

57 print(tns.url) 

58 csvSources, csvPhot, csvSpec, csvFiles = tns.csv() 

59 print(csvSources) 

60 jsonSources, jsonPhot, jsonSpec, jsonFiles = tns.json() 

61 print(jsonSources) 

62 yamlSources, yamlPhot, yamlSpec, yamlFiles = tns.yaml() 

63 print(yamlSources) 

64 markdownSources, markdownPhot, markdownSpec, markdownFiles = tns.markdown() 

65 print(markdownSources) 

66 tableSources, tablePhot, tableSpec, tableFiles = tns.table() 

67 print(tableSources) 

68 mysqlSources, mysqlPhot, mysqlSpec, mysqlFiles = tns.mysql("TNS") 

69 print(mysqlSources) 

70 print(tablePhot) 

71 print(tableFiles) 

72 

73 tns.csv(dirPath=pathToOutputDir) 

74 tns.table(dirPath=pathToOutputDir) 

75 tns.mysql(tableNamePrefix="TNS", dirPath=pathToOutputDir) 

76 tns.json(dirPath=pathToOutputDir) 

77 tns.yaml(dirPath=pathToOutputDir) 

78 tns.markdown(dirPath=pathToOutputDir) 

79 

80 def test_search_function02(self): 

81 kwargs = {} 

82 kwargs["log"] = log 

83 kwargs["settings"] = settings 

84 kwargs["ra"] = "06:45:03.36" 

85 kwargs["dec"] = "+35:44:29.8" 

86 kwargs["radiusArcsec"] = 5.0 

87 

88 from transientNamer import search 

89 tns = search(**kwargs) 

90 print(tns.url) 

91 tns.csv(dirPath=pathToOutputDir) 

92 tns.table(dirPath=pathToOutputDir) 

93 tns.mysql(tableNamePrefix="fs_tns", dirPath=pathToOutputDir) 

94 tns.json(dirPath=pathToOutputDir) 

95 tns.yaml(dirPath=pathToOutputDir) 

96 tns.markdown(dirPath=pathToOutputDir) 

97 

98 def test_search_function03(self): 

99 # A TEST FOR MUPTIPLE RESULTS 

100 kwargs = {} 

101 kwargs["log"] = log 

102 kwargs["settings"] = settings 

103 kwargs["ra"] = "00:00:00.00" 

104 kwargs["dec"] = "+00:00:00.00" 

105 kwargs["radiusArcsec"] = 5.0 

106 # xt-kwarg_key_and_value 

107 from transientNamer import search 

108 tns = search(**kwargs) 

109 

110 print(tns.url) 

111 tns.csv(dirPath=pathToOutputDir) 

112 tns.table(dirPath=pathToOutputDir) 

113 tns.mysql(tableNamePrefix="fs_tns", dirPath=pathToOutputDir) 

114 tns.json(dirPath=pathToOutputDir) 

115 tns.yaml(dirPath=pathToOutputDir) 

116 tns.markdown(dirPath=pathToOutputDir) 

117 

118 def test_search_function04(self): 

119 # A TEST FOR MUPTIPLE RESULTS 

120 kwargs = {} 

121 kwargs["log"] = log 

122 kwargs["settings"] = settings 

123 kwargs["name"] = "2016fbz" 

124 

125 # xt-kwarg_key_and_value 

126 from transientNamer import search 

127 tns = search(**kwargs) 

128 print(tns.url) 

129 tns.csv(dirPath=pathToOutputDir) 

130 tns.table(dirPath=pathToOutputDir) 

131 tns.mysql(tableNamePrefix="fs_tns", dirPath=pathToOutputDir) 

132 tns.json(dirPath=pathToOutputDir) 

133 tns.yaml(dirPath=pathToOutputDir) 

134 tns.markdown(dirPath=pathToOutputDir) 

135 

136 def test_search_function05(self): 

137 # A TEST FOR MUPTIPLE RESULTS 

138 kwargs = {} 

139 kwargs["log"] = log 

140 kwargs["settings"] = settings 

141 kwargs["name"] = "Gaia16bbi" 

142 

143 # xt-kwarg_key_and_value 

144 from transientNamer import search 

145 tns = search(**kwargs) 

146 

147 print(tns.url) 

148 tns.csv(dirPath=pathToOutputDir) 

149 tns.table(dirPath=pathToOutputDir) 

150 tns.mysql(tableNamePrefix="fs_tns", dirPath=pathToOutputDir) 

151 tns.json(dirPath=pathToOutputDir) 

152 tns.yaml(dirPath=pathToOutputDir) 

153 tns.markdown(dirPath=pathToOutputDir) 

154 

155 def test_search_function06(self): 

156 # A TEST FOR MUPTIPLE RESULTS 

157 kwargs = {} 

158 kwargs["log"] = log 

159 kwargs["settings"] = settings 

160 kwargs["discInLastDays"] = 3 

161 

162 # xt-kwarg_key_and_value 

163 from transientNamer import search 

164 tns = search(**kwargs) 

165 

166 print(tns.url) 

167 tns.csv(dirPath=pathToOutputDir) 

168 tns.table(dirPath=pathToOutputDir) 

169 tns.mysql(tableNamePrefix="fs_tns", dirPath=pathToOutputDir) 

170 tns.json(dirPath=pathToOutputDir) 

171 tns.yaml(dirPath=pathToOutputDir) 

172 tns.markdown(dirPath=pathToOutputDir) 

173 

174 # x-print-testpage-for-pessto-marshall-web-object 

175 

176 # x-class-to-test-named-worker-function