Fix the extended tests on Github Actions.

lwa-project · Nov 30, 2023 · 077d5c5 · 077d5c5
1 parent 9d8677d
commit 077d5c5
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/tests/test_ldp_extended.py b/tests/test_ldp_extended.py
@@ -24,14 +24,18 @@
 
 
 run_extended_tests = False
-if os.getenv('TRAVIS_OS_NAME', None) is not None:
+if os.getenv('GITHUB_ACTIONS', None) is not None:
     run_extended_tests = True
 
 
 _TBN_URL = 'https://lda10g.alliance.unm.edu/tutorial/Meteors/056761_000099453'
 _DRX_URL = 'https://lda10g.alliance.unm.edu/tutorial/UnknownPulsar/056227_000024985_DRX.dat'
 _SPC_URL = 'https://lda10g.alliance.unm.edu/tutorial/B0329+54/056770_000044687'
 
+tbnFile = os.path.join(os.path.dirname(__file__), 'data', 'tbn-extended.dat')
+drxFile = os.path.join(os.path.dirname(__file__), 'data', 'drx-extended.dat')
+drspecFile = os.path.join(os.path.dirname(__file__), 'data', 'drspec-extended.dat')
+
 
 @unittest.skipUnless(run_extended_tests, "requires appropriate environment variable to be set")
 class extended_ldp_tests(unittest.TestCase):
@@ -41,7 +45,7 @@ class extended_ldp_tests(unittest.TestCase):
     def setUp(self):
         """Download the files."""
 
-        for filename,url in zip(('tbn.dat','drx.dat','drspec.dat'), (_TBN_URL,_DRX_URL,_SPC_URL)):
+        for filename,url in zip((tbnFile, drxFile, drspecFile), (_TBN_URL,_DRX_URL,_SPC_URL)):
             if not os.path.exists(filename):
                 subprocess.check_call(['curl', url, 
                                    '--range', '0-%i' % (250*1024*1024), 
@@ -50,15 +54,15 @@ def setUp(self):
     def test_tbn_estimate(self):
         """Test estimating power levels in a TBN file."""
 
-        idf = ldp.TBNFile('tbn.dat')
+        idf = ldp.TBNFile(tbnFile)
         offset = idf.offset(0.1)       # This file needs a skip at the beginning
         levels = idf.estimate_levels()
         self.assertEqual(len(levels), 520)
 
     def test_tbn_read(self):
         """Test more involved reading from a TBN file."""
 
-        idf = ldp.TBNFile('tbn.dat')
+        idf = ldp.TBNFile(tbnFile)
         offset = idf.offset(0.1)       # This file needs a skip at the beginning
 
         # Read some
@@ -70,7 +74,7 @@ def test_tbn_read(self):
     def test_tbn_offset(self):
         """Test offsetting inside a TBN file."""
 
-        idf = ldp.TBNFile('tbn.dat')
+        idf = ldp.TBNFile(tbnFile)
         offset = idf.offset(0.1)       # This file needs a skip at the beginning
 
         # Jump forwards
@@ -102,14 +106,14 @@ def test_tbn_offset(self):
     def test_drx_estimate(self):
         """Test estimating power levels in a DRX file."""
 
-        idf = ldp.DRXFile('drx.dat')
+        idf = ldp.DRXFile(drxFile)
         levels = idf.estimate_levels()
         self.assertEqual(len(levels), 4)
 
     def test_drx_read(self):
         """Test more involved reading from a DRX file."""
 
-        idf = ldp.DRXFile('drx.dat')
+        idf = ldp.DRXFile(drxFile)
 
         # Read some
         for i in range(21):
@@ -120,7 +124,7 @@ def test_drx_read(self):
     def test_drx_offset(self):
         """Test offsetting inside a DRX file."""
 
-        idf = ldp.DRXFile('drx.dat')
+        idf = ldp.DRXFile(drxFile)
 
         # Jump forwards
         fileStart = idf.start_time
@@ -151,7 +155,7 @@ def test_drx_offset(self):
     def test_drspec_read(self):
         """Test more involved reading from a DR Spectrometer file."""
 
-        idf = ldp.DRSpecFile('drspec.dat')
+        idf = ldp.DRSpecFile(drspecFile)
 
         # Read some
         for i in range(21):
@@ -162,7 +166,7 @@ def test_drspec_read(self):
     def test_drspec_offset(self):
         """Test offsetting inside a DR Spectrometer file."""
 
-        idf = ldp.DRSpecFile('drspec.dat')
+        idf = ldp.DRSpecFile(drspecFile)
 
         # Jump forwards
         fileStart = idf.start_time