wavinfo/tests/test_walk.py

import glob
import unittest

import wavinfo


class TestWalk(unittest.TestCase):
    def test_walk_metadata(self):
        test_file = "tests/test_files/protools/PT A101_4.A1.wav"
        info = wavinfo.WavInfoReader(test_file)

        tested_data, tested_format = False, False
        for scope, key, value in info.walk():
            if scope == "fmt":
                if key == "channel_count":
                    tested_format = True
                    self.assertEqual(value, 2)
            if scope == "data":
                if key == "frame_count":
                    tested_data = True
                    self.assertEqual(value, 144140)

        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 Exception as e:
                self.fail(f"Failed to walk metadata in file {file}: {e}")


if __name__ == "__main__":
    unittest.main()