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

1from __future__ import print_function 

2from builtins import str 

3import os 

4import unittest 

5import shutil 

6import unittest 

7import yaml 

8from soxspipe.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" 

15# settingsFile = home + \ 

16# "/git_repos/_misc_/settings/soxspipe/test_settings.yaml" 

17 

18su = tools( 

19 arguments={"settingsFile": settingsFile}, 

20 docString=__doc__, 

21 logLevel="DEBUG", 

22 options_first=False, 

23 projectName=None, 

24 defaultSettingsFile=False 

25) 

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

27 

28# SETUP PATHS TO COMMON DIRECTORIES FOR TEST DATA 

29moduleDirectory = os.path.dirname(__file__) 

30pathToInputDir = moduleDirectory + "/input/" 

31pathToOutputDir = moduleDirectory + "/output/" 

32 

33try: 

34 shutil.rmtree(pathToOutputDir) 

35except: 

36 pass 

37# COPY INPUT TO OUTPUT DIR 

38shutil.copytree(pathToInputDir, pathToOutputDir) 

39 

40# Recursively create missing directories 

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

42 os.makedirs(pathToOutputDir) 

43 

44 

45# xt-setup-unit-testing-files-and-folders 

46# xt-utkit-refresh-database 

47 

48class test_dispersion_map_to_pixel_arrays(unittest.TestCase): 

49 

50 def test_dispersion_map_to_pixel_arrays_function(self): 

51 

52 dispersionMapPath = "~/xshooter-pipeline-data/unittest_data/xsh/detect_continuum/single_pinhole_NIR_disp_map.csv" 

53 from soxspipe.commonutils import dispersion_map_to_pixel_arrays 

54 import pandas as pd 

55 # CREATE DATA FRAME FROM A DICTIONARY OF LISTS 

56 myDict = { 

57 "order": [11, 11, 11, 11, 11], 

58 "wavelength": [2000., 2100., 2200., 2300., 2400.], 

59 "slit_position": [0, 0, 0, 0, 0] 

60 } 

61 orderPixelTable = pd.DataFrame(myDict) 

62 orderPixelTable = dispersion_map_to_pixel_arrays( 

63 log=log, 

64 dispersionMapPath=dispersionMapPath, 

65 orderPixelTable=orderPixelTable 

66 ) 

67 

68 def test_dispersion_map_to_pixel_arrays_function_exception(self): 

69 

70 from soxspipe.commonutils import dispersion_map_to_pixel_arrays 

71 try: 

72 this = dispersion_map_to_pixel_arrays( 

73 log=log, 

74 settings=settings, 

75 fakeKey="break the code" 

76 ) 

77 this.get() 

78 assert False 

79 except Exception as e: 

80 assert True 

81 print(str(e)) 

82 

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

84 

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