From d1b42bd83659dd9ffec9bf286204be9510a87f73 Mon Sep 17 00:00:00 2001 From: Jamie Hardt Date: Tue, 7 Nov 2023 11:37:36 -0800 Subject: [PATCH] Fixed a bug in the cues to_dict method --- tests/test_walk.py | 12 ++++++++++++ wavinfo/wave_cues_reader.py | 8 ++++---- 2 files changed, 16 insertions(+), 4 deletions(-) diff --git a/tests/test_walk.py b/tests/test_walk.py index 63df1ab..361766a 100644 --- a/tests/test_walk.py +++ b/tests/test_walk.py @@ -1,6 +1,7 @@ import unittest import wavinfo +import glob class TestWalk(unittest.TestCase): def test_walk_metadata(self): @@ -20,6 +21,17 @@ class TestWalk(unittest.TestCase): self.assertTrue(tested_data and tested_format) + def test_walk_all(self): + for file in glob.glob('tests/test_files/**/*.wav'): + info = wavinfo.WavInfoReader(file) + + try: + for _, _, _ in info.walk(): + pass + except: + self.fail(f"Failed to walk metadata in file {file}") + + if __name__ == '__main__': unittest.main() diff --git a/wavinfo/wave_cues_reader.py b/wavinfo/wave_cues_reader.py index 7d578c0..c1fbaeb 100644 --- a/wavinfo/wave_cues_reader.py +++ b/wavinfo/wave_cues_reader.py @@ -247,10 +247,10 @@ class WavCuesReader: if r.name == cue_ident), None) def to_dict(self) -> Dict[str, Any]: - return dict(cues=[c.__dict__ for c in self.cues], - labels=[l.__dict__ for l in self.labels], - ranges=[r.__dict__ for r in self.ranges], - notes=[n.__dict__ for n in self.notes]) + return dict(cues=[c._asdict() for c in self.cues], + labels=[l._asdict() for l in self.labels], + ranges=[r._asdict() for r in self.ranges], + notes=[n._asdict() for n in self.notes])