add stem_id test

faroit · Nov 10, 2018 · 406acc6 · 406acc6
1 parent 226e961
commit 406acc6
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_read.py b/tests/test_read.py
@@ -8,6 +8,18 @@ def dtype(request):
     return request.param
 
 
+def test_stem_id():
+    S, rate = stempeg.read_stems(
+        "tests/data/The Easton Ellises - Falcon 69.stem.mp4"
+    )
+    for k in range(S.shape[0]):
+        Sk, rate = stempeg.read_stems(
+            "tests/data/The Easton Ellises - Falcon 69.stem.mp4",
+            stem_id=k
+        )
+        assert Sk.ndim == 2
+
+
 def test_shape():
     S, rate = stempeg.read_stems(
         "tests/data/The Easton Ellises - Falcon 69.stem.mp4"