Changeset b610be5

Timestamp:

Nov 29, 2018, 4:08:24 PM (6 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master

Children:

da67922

Parents:

60384e9

Message:

[tests] pure python create_sine_wave

File:

• tests/create_tests_source.py (modified) (2 diffs)

tests/create_tests_source.py

@@ -2 +2 @@
 
 """ Create a simple stereo file containing a sine tone at 441 Hz, using only
-numpy and python's native wave module. """
+python's built-in modules. """
 
 import wave
-import numpy as np
+import math
+import struct
 
 
 def create_sine_wave(freq, samplerate, nframes, nchannels):
-    """ create a pure tone """
-    # samples indices
-    _t = np.tile(np.arange(nframes), (nchannels, 1))
-    # sine wave generation
-    _x = 0.7 * np.sin(2. * np.pi * freq * _t / float(samplerate))
-    # conversion to int and channel interleaving
-    return (_x * 32767.).astype(np.int16).T.flatten()
+    """ create a pure tone (without numpy) """
+    _x = [0.7 * math.sin(2. * math.pi * freq * t / float(samplerate))
+          for t in range(nframes)]
+    _x = [int(a * 32767) for a in _x]
+    _x = b''.join([b''.join([struct.pack('h', v)
+                             for _ in range(nchannels)])
+                   for v in _x])
+    return _x
 
 
 def create_test_sound(pathname, freq=441, duration=None,
-                      sampwidth=2, framerate=44100, nchannels=2):
+                      framerate=44100, nchannels=2):
     """ create a sound file at pathname, overwriting exiting file """
+    sampwidth = 2
     nframes = duration or framerate  # defaults to 1 second duration
     fid = wave.open(pathname, 'w')
@@ -28 +31 @@
     fid.setnframes(nframes)
     frames = create_sine_wave(freq, framerate, nframes, nchannels)
-    fid.writeframes(frames.tobytes())
+    fid.writeframes(frames)
     fid.close()
     return 0