1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import unittest 

7import yaml 

8from astrocalc.utKit import utKit 

9from fundamentals import tools 

10from os.path import expanduser 

11home = expanduser("~") 

12 

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

14settingsFile = packageDirectory + "/test_settings.yaml" 

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_converter(unittest.TestCase): 

43 

44 def test_converter_function01(self): 

45 

46 from astrocalc.distances import converter 

47 c = converter(log=log) 

48 

49 dists = c.redshift_to_distance( 

50 z=0.108, 

51 WM=0.3, 

52 WV=0.7, 

53 H0=70.0 

54 ) 

55 

56 print(dists) 

57 

58 def test_converter_function02(self): 

59 

60 from astrocalc.distances import converter 

61 c = converter(log=log) 

62 

63 dists = c.distance_to_redshift( 

64 mpc=500 

65 ) 

66 

67 print(dists) 

68 

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

70 

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