1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import yaml 

7from sloancone.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 

41 

42class test_check_coverage(unittest.TestCase): 

43 

44 def test_check_coverage_function(self): 

45 

46 from sloancone import check_coverage 

47 covered = check_coverage( 

48 log=log, 

49 ra="122.3343", 

50 dec="45.34343" 

51 ).get() 

52 

53 print(covered) 

54 

55 def test_check_coverage_function2(self): 

56 

57 from sloancone import check_coverage 

58 covered = check_coverage( 

59 log=log, 

60 ra=122.3343, 

61 dec=-45.34343 

62 ).get() 

63 

64 print(covered) 

65 

66 def test_check_coverage_function3(self): 

67 

68 from sloancone import check_coverage 

69 covered = check_coverage( 

70 log=log, 

71 ra="12:45:4.45466", 

72 dec="-25:22:34.3434" 

73 ).get() 

74 

75 print(covered) 

76 

77 def test_check_coverage_function_exception(self): 

78 from sloancone import check_coverage 

79 try: 

80 this = check_coverage( 

81 log=log, 

82 settings=settings, 

83 fakeKey="break the code" 

84 ) 

85 this.get() 

86 assert False 

87 except Exception as e: 

88 assert True 

89 print(str(e)) 

90 

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

92 

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