summaryrefslogtreecommitdiff
path: root/tests/create_test_data.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/create_test_data.py')
-rw-r--r--tests/create_test_data.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/create_test_data.py b/tests/create_test_data.py
index 0bb1567..b44a855 100644
--- a/tests/create_test_data.py
+++ b/tests/create_test_data.py
@@ -333,7 +333,7 @@ def sndobj_partial_tracking():
frames = []
i = 0
- while i < len(audio) - hop_size:
+ while (i < len(audio) - hop_size) and len(frames) < num_frames:
frame = np.asarray(audio[i:i + frame_size], dtype=np.float32)
input.PushIn(frame)
ifgram.DoProcess()