Coverage for neddy/tests/test_conesearch.py: 94%

1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import yaml 

7from neddy.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_conesearch(unittest.TestCase): 

42 

43 def test_conesearch_function(self): 

44 from neddy import conesearch 

45 kwargs = {} 

46 kwargs["log"] = log 

47 kwargs["ra"] = 204.253833 

48 kwargs["dec"] = -29.865750 

49 kwargs["radiusArcsec"] = 15 

50 kwargs["nearestOnly"] = False 

51 kwargs["unclassified"] = True 

52 testObject = conesearch(**kwargs) 

53 testObject.get() 

54 

55 def test_conesearch_function_02(self): 

56 from neddy import conesearch 

57 import codecs 

58 pathToReadFile = pathToInputDir + "/coordinates.txt" 

59 readFile = codecs.open(pathToReadFile, encoding='utf-8', mode='r') 

60 

61 listOfCoordinates = [] 

62 for line in readFile.readlines(): 

63 line = line.strip() 

64 [ra, dec] = line.split() 

65 listOfCoordinates.append(str(ra) + " " + str(dec)) 

66 

67 kwargs = {} 

68 kwargs["log"] = log 

69 kwargs["nearestOnly"] = False 

70 kwargs["unclassified"] = True 

71 kwargs["radiusArcsec"] = 29 

72 kwargs["listOfCoordinates"] = listOfCoordinates 

73 testObject = conesearch(**kwargs) 

74 testObject.get() 

75 

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

77 

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