[tests] use np.loadtxt in utils
authorPaul Brossier <piem@piem.org>
Thu, 1 Nov 2018 21:13:11 +0000 (22:13 +0100)
committerPaul Brossier <piem@piem.org>
Thu, 1 Nov 2018 21:13:11 +0000 (22:13 +0100)
python/tests/utils.py

index b0963fc..b374fe5 100644 (file)
@@ -8,11 +8,8 @@ from tempfile import mkstemp
 DEFAULT_SOUND = '22050Hz_5s_brownnoise.wav'
 
 def array_from_text_file(filename, dtype = 'float'):
-    filename = os.path.join(os.path.dirname(__file__), filename)
-    with open(filename) as f:
-        lines = f.readlines()
-    return np.array([line.split() for line in lines],
-            dtype = dtype)
+    realpathname = os.path.join(os.path.dirname(__file__), filename)
+    return np.loadtxt(realpathname, dtype = dtype)
 
 def list_all_sounds(rel_dir):
     datadir = os.path.join(os.path.dirname(__file__), rel_dir)