f

Author: Teque5

tests/test_convert_blue.py

@@ -101,52 +101,43 @@ def write_minimal(self, format: bytes = b"CF") -> None:
             handle.seek(512)
             handle.write(iq_converted.tobytes())  # write IQ data
 
+    def check_data_and_metadata(self, meta, form, atol):
+        """ensure blue is sample perfect and metadata is equivalent"""
+        self.assertEqual(meta.sample_rate, self.samp_rate)
+        self.assertEqual(meta.get_capture_info(0)["core:datetime"], self.datetime)
+        if form[0] == "S":
+            np.testing.assert_allclose(self.iq_data.real, meta[:], atol=atol)
+        else:
+            np.testing.assert_allclose(self.iq_data, meta[:], atol=atol)
+
     def test_blue_to_sigmf_pair(self) -> None:
         """test standard blue to sigmf conversion with file pairs"""
-        for format, atol in self.format_tolerance:
+        for form, atol in self.format_tolerance:
             sigmf_path = self.tmp_path / f"bar{format}"
-            self.write_minimal(format.encode())
+            self.write_minimal(form.encode())
             meta = blue_to_sigmf(blue_path=self.blue_path, out_path=sigmf_path)
             filenames = sigmf.sigmffile.get_sigmf_filenames(sigmf_path)
             self.assertTrue(filenames["data_fn"].exists(), "dataset path missing")
             self.assertTrue(filenames["meta_fn"].exists(), "metadata path missing")
-            self.assertEqual(meta.sample_rate, self.samp_rate)
-            self.assertEqual(meta.get_capture_info(0)["core:datetime"], self.datetime)
-            # verify data
-            if format[0] == "S":
-                np.testing.assert_allclose(self.iq_data.real, meta[:], atol=atol)
-            else:
-                np.testing.assert_allclose(self.iq_data, meta[:], atol=atol)
+            self.check_data_and_metadata(meta, form, atol)
 
     def test_blue_to_sigmf_archive(self) -> None:
         """test blue to sigmf conversion with archive output"""
-        for format, atol in self.format_tolerance:
-            self.write_minimal(format.encode())
+        for form, atol in self.format_tolerance:
+            self.write_minimal(form.encode())
             sigmf_path = self.tmp_path / f"baz{format}"
             meta = blue_to_sigmf(blue_path=self.blue_path, out_path=sigmf_path, create_archive=True)
             filenames = sigmf.sigmffile.get_sigmf_filenames(sigmf_path)
             self.assertTrue(filenames["archive_fn"].exists(), "archive path missing")
-            self.assertEqual(meta.sample_rate, self.samp_rate)
-            self.assertEqual(meta.get_capture_info(0)["core:datetime"], self.datetime)
-            # verify data
-            if format[0] == "S":
-                np.testing.assert_allclose(self.iq_data.real, meta[:], atol=atol)
-            else:
-                np.testing.assert_allclose(self.iq_data, meta[:], atol=atol)
+            self.check_data_and_metadata(meta, form, atol)
 
     def test_blue_to_sigmf_ncd(self) -> None:
         """test automatic NCD conversion with fromfile()"""
-        for format, atol in self.format_tolerance:
-            self.write_minimal(format.encode())
+        for form, atol in self.format_tolerance:
+            self.write_minimal(form.encode())
             meta = blue_to_sigmf(self.blue_path)
             _validate_ncd(self, meta, self.blue_path)
-            self.assertEqual(meta.sample_rate, self.samp_rate)
-            self.assertEqual(meta.get_capture_info(0)["core:datetime"], self.datetime)
-            # verify data
-            if format[0] == "S":
-                np.testing.assert_allclose(self.iq_data.real, meta[:], atol=atol)
-            else:
-                np.testing.assert_allclose(self.iq_data, meta[:], atol=atol)
+            self.check_data_and_metadata(meta, form, atol)
 
 
 class TestBlueWithNonSigMFRepo(unittest.TestCase):