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)


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