Skip to content

Commit d6b8d61

Browse files
committed
simplify wav tests
1 parent e6b7423 commit d6b8d61

1 file changed

Lines changed: 12 additions & 6 deletions

File tree

tests/test_convert_wav.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -67,30 +67,36 @@ def test_wav_to_sigmf_pair(self):
6767
"""test standard wav to sigmf conversion with file pairs"""
6868
sigmf_path = self.tmp_path / "bar.tmp"
6969
meta = wav_to_sigmf(wav_path=str(self.wav_path), out_path=str(sigmf_path))
70+
filenames = sigmf.sigmffile.get_sigmf_filenames(sigmf_path)
71+
self.assertTrue(filenames["data_fn"].exists(), "dataset path missing")
72+
self.assertTrue(filenames["meta_fn"].exists(), "metadata path missing")
73+
# verify data
7074
data = meta.read_samples()
7175
self.assertGreater(len(data), 0, "Should read some samples")
7276
# allow numerical differences due to PCM quantization
7377
self.assertTrue(np.allclose(self.audio_data, data, atol=1e-4))
74-
filenames = sigmf.sigmffile.get_sigmf_filenames(sigmf_path)
75-
self.assertTrue(filenames["data_fn"].exists(), "dataset path missing")
76-
self.assertTrue(filenames["meta_fn"].exists(), "metadata path missing")
7778

7879
def test_wav_to_sigmf_archive(self):
7980
"""test wav to sigmf conversion with archive output"""
8081
sigmf_path = self.tmp_path / "baz.ext"
81-
wav_to_sigmf(wav_path=str(self.wav_path), out_path=str(sigmf_path), create_archive=True)
82+
meta = wav_to_sigmf(wav_path=str(self.wav_path), out_path=str(sigmf_path), create_archive=True)
8283
filenames = sigmf.sigmffile.get_sigmf_filenames(sigmf_path)
8384
self.assertTrue(filenames["archive_fn"].exists(), "archive path missing")
85+
# verify data
86+
data = meta.read_samples()
87+
self.assertGreater(len(data), 0, "Should read some samples")
88+
# allow numerical differences due to PCM quantization
89+
self.assertTrue(np.allclose(self.audio_data, data, atol=1e-4))
8490

8591
def test_wav_to_sigmf_ncd(self):
8692
"""test wav to sigmf conversion as Non-Conforming Dataset"""
8793
meta = wav_to_sigmf(wav_path=str(self.wav_path), create_ncd=True)
8894
_validate_ncd(self, meta, self.wav_path)
8995

90-
# verify data can still be read correctly from NCD
96+
# verify data
9197
data = meta.read_samples()
92-
self.assertGreater(len(data), 0, "Should read some samples")
9398
# allow numerical differences due to PCM quantization
99+
self.assertGreater(len(data), 0, "Should read some samples")
94100
self.assertTrue(np.allclose(self.audio_data, data, atol=1e-4))
95101

96102

0 commit comments

Comments
 (0)