diff --git a/srcpkgs/python3-pyinfra/template b/srcpkgs/python3-pyinfra/template
index a1b34afe347..519e5344189 100644
--- a/srcpkgs/python3-pyinfra/template
+++ b/srcpkgs/python3-pyinfra/template
@@ -8,6 +8,7 @@ hostmakedepends="python3-setuptools"
 depends="python3 python3-Jinja2 python3-click python3-colorama python3-dateutil
  python3-distro python3-docopt python3-gevent python3-paramiko python3-pywinrm
  python3-six python3-yaml"
+checkdepends="${depends} python3-pytest python3-mock openssh"
 short_desc="Automate infrastructure super fast at massive scale"
 maintainer="Leah Neukirchen <leah@vuxu.org>"
 license="MIT"
@@ -23,3 +24,7 @@ post_install() {
 	rm -r ${DESTDIR}/${py3_sitelib}/tests
 	vlicense LICENSE.md
 }
+
+do_check() {
+	pytest --disable-warnings
+}