From 077d5c516d292b9b939d0515b7fb0d3acea27ecd Mon Sep 17 00:00:00 2001
From: jaycedowell <jayce.dowell@gmail.com>
Date: Wed, 29 Nov 2023 19:25:40 -0700
Subject: [PATCH] Fix the extended tests on Github Actions.

---
 tests/test_ldp_extended.py | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/tests/test_ldp_extended.py b/tests/test_ldp_extended.py
index e04db888..aedc9c78 100644
--- a/tests/test_ldp_extended.py
+++ b/tests/test_ldp_extended.py
@@ -24,7 +24,7 @@
 
 
 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
 
 
@@ -32,6 +32,10 @@
 _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,7 +54,7 @@ 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)
@@ -58,7 +62,7 @@ def test_tbn_estimate(self):
     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