diff --git a/wetb/prepost/tests/test_hawcstab2.py b/wetb/prepost/tests/test_hawcstab2.py index a260b8a973b9fd287e02c04fb7ae5104e24312cf..249df2097c7a4167640d420eb3af1a1064981151 100644 --- a/wetb/prepost/tests/test_hawcstab2.py +++ b/wetb/prepost/tests/test_hawcstab2.py @@ -11,25 +11,26 @@ from future import standard_library standard_library.install_aliases() import unittest +import os import numpy as np from wetb.prepost.hawcstab2 import results, hs2_control_tuning -# path for test data files -#fpath = os.path.join(os.path.dirname(__file__), 'data/') - class Tests(unittest.TestCase): """ """ def setUp(self): - self.fpath_linear = 'data/controller_input_linear.txt' - self.fpath_quadratic = 'data/controller_input_quadratic.txt' + self.fpath_linear = os.path.join(os.path.dirname(__file__), + 'data/controller_input_linear.txt') + self.fpath_quad = os.path.join(os.path.dirname(__file__), + 'data/controller_input_quadratic.txt') def test_cmb_df(self): - fname1 = 'data/campbell_diagram.cmb' + fname1 = os.path.join(os.path.dirname(__file__), + 'data/campbell_diagram.cmb') speed, freq, damp = results().load_cmb(fname1) df = results().load_cmb_df(fname1) @@ -69,7 +70,7 @@ class Tests(unittest.TestCase): def test_quadratic_file(self): hs2 = hs2_control_tuning() - hs2.read_parameters(self.fpath_quadratic) + hs2.read_parameters(self.fpath_quad) self.assertEqual(hs2.pi_gen_reg1.K, 0.108313E+07) diff --git a/wetb/prepost/tests/test_windIO.py b/wetb/prepost/tests/test_windIO.py index 67239c76e00e8dbf4684830bcc34f1e5ab8ff089..e6951ab7d2a908446cbb7ff3b4df2601a3136375 100644 --- a/wetb/prepost/tests/test_windIO.py +++ b/wetb/prepost/tests/test_windIO.py @@ -11,6 +11,7 @@ from future import standard_library standard_library.install_aliases() import unittest +import os import numpy as np @@ -20,7 +21,8 @@ from wetb.prepost.windIO import LoadResults class TestsLoadResults(unittest.TestCase): def setUp(self): - self.respath = '../../hawc2/tests/test_files/hawc2io/' + self.respath = os.path.join(os.path.dirname(__file__), + '../../hawc2/tests/test_files/hawc2io/') self.fascii = 'Hawc2ascii' self.fbin = 'Hawc2bin'