Test fixes for dsamp .fil output

UCBerkeleySETI · Aug 2, 2022 · 521d21b · 521d21b
1 parent 9fbe0fb
commit 521d21b
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/tests/test_dsamp.py b/tests/test_dsamp.py
@@ -4,7 +4,7 @@
 
 import pytest
 import blimpy as bl
-from tests.data import voyager_fil, voyager_h5, test_h5
+from tests.data import voyager_fil, voyager_h5, test_h5, test_fil
 
 
 GROUP_SIZE = 3
@@ -13,13 +13,25 @@
 def test_dsamp_fil_to_h5():
     """ fil to h5 test.
     """
-    bl.dsamp.make_h5_file(voyager_fil, test_h5, GROUP_SIZE)
+    bl.dsamp.make_output_file(voyager_fil, test_h5, GROUP_SIZE, True)
 
 
 def test_dsamp_h5_to_h5():
     """ h5 to h5 test.
     """
-    bl.dsamp.make_h5_file(voyager_h5, test_h5, GROUP_SIZE)
+    bl.dsamp.make_output_file(voyager_h5, test_h5, GROUP_SIZE, True)
+
+
+def test_dsamp_h5_to_fil():
+    """ h5 to fil test.
+    """
+    bl.dsamp.make_output_file(voyager_h5, test_fil, GROUP_SIZE, False)
+
+
+def test_dsamp_fil_to_fil():
+    """ fil to fil test.
+    """
+    bl.dsamp.make_output_file(voyager_fil, test_fil, GROUP_SIZE, False)
 
 
 def test_cmd_tool():