diff --git a/setup.py b/setup.py
index fc6260b..fb70907 100644
--- a/setup.py
+++ b/setup.py
@@ -26,5 +26,8 @@
         'numpy >= 1.7.0',
         'mido >= 1.1.16',
         'six',
-    ]
+    ],
+    extras_require={
+        "test": ["pytest>=6.0"],
+    },
 )