Coverage for soxspipe/commonutils/tests/test_dispersion_map_to_pixel_arrays.py : 88%

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("~")
13packageDirectory = utKit("").get_project_root()
14settingsFile = packageDirectory + "/test_settings.yaml"
15# settingsFile = home + \
16# "/git_repos/_misc_/settings/soxspipe/test_settings.yaml"
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()
28# SETUP PATHS TO COMMON DIRECTORIES FOR TEST DATA
29moduleDirectory = os.path.dirname(__file__)
30pathToInputDir = moduleDirectory + "/input/"
31pathToOutputDir = moduleDirectory + "/output/"
33try:
34 shutil.rmtree(pathToOutputDir)
35except:
36 pass
37# COPY INPUT TO OUTPUT DIR
38shutil.copytree(pathToInputDir, pathToOutputDir)
40# Recursively create missing directories
41if not os.path.exists(pathToOutputDir):
42 os.makedirs(pathToOutputDir)
45# xt-setup-unit-testing-files-and-folders
46# xt-utkit-refresh-database
48class test_dispersion_map_to_pixel_arrays(unittest.TestCase):
50 def test_dispersion_map_to_pixel_arrays_function(self):
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 )
68 def test_dispersion_map_to_pixel_arrays_function_exception(self):
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))
83 # x-print-testpage-for-pessto-marshall-web-object
85 # x-class-to-test-named-worker-function