diff --git a/tests/test_audio.py b/tests/test_audio.py index 26a9529d..80bb957e 100644 --- a/tests/test_audio.py +++ b/tests/test_audio.py @@ -19,8 +19,8 @@ def before_all() -> None: def test_get_audio_frame() -> None: - assert hasattr(get_audio_frame(get_test_example_file('source.mp3'), 25), '__array_interface__') - assert hasattr(get_audio_frame(get_test_example_file('source.wav'), 25), '__array_interface__') + assert get_audio_frame(get_test_example_file('source.mp3'), 25).shape == (80, 16) + assert get_audio_frame(get_test_example_file('source.wav'), 25).shape == (80, 16) assert get_audio_frame('invalid', 25) is None diff --git a/tests/test_common_helper.py b/tests/test_common_helper.py index b886a84e..e96a1d8c 100644 --- a/tests/test_common_helper.py +++ b/tests/test_common_helper.py @@ -16,7 +16,7 @@ def test_create_int_range() -> None: def test_create_float_range() -> None: assert create_float_range(0.0, 1.0, 0.5) == [ 0.0, 0.5, 1.0 ] - assert create_float_range(0.0, 1.0, 0.05) == [ 0.0, 0.05, 0.10, 0.15, 0.20, 0.25, 0.30, 0.35, 0.40, 0.45, 0.50, 0.55, 0.60, 0.65, 0.70, 0.75, 0.80, 0.85, 0.90, 0.95, 1.0 ] + assert create_float_range(0.0, 0.5, 0.05) == [ 0.0, 0.05, 0.10, 0.15, 0.20, 0.25, 0.30, 0.35, 0.40, 0.45, 0.50 ] def test_calc_int_step() -> None: