diff --git a/dev-python/cfu/cfu-1.5.0.ebuild b/dev-python/cfu/cfu-1.5.0.ebuild
index 6da9993..c43e129 100644
--- a/dev-python/cfu/cfu-1.5.0.ebuild
+++ b/dev-python/cfu/cfu-1.5.0.ebuild
@@ -14,7 +14,24 @@ SRC_URI="mirror://pypi/${PN:0:1}/${PN}/${P}.tar.gz"
 LICENSE="Apache-2.0"
 SLOT="0"
 KEYWORDS="~amd64 ~x86"
+IUSE="test"
+RESTRICT="!test? ( test )"
 
 DEPEND=""
-RDEPEND="${DEPEND}"
+RDEPEND="${DEPEND}
+	test? ( dev-python/pytest[${PYTHON_USEDEP}] )"
 BDEPEND=""
+
+python_test() {
+	pytest -vv || die "Tests fail with ${EPYTHON}"
+}
+
+src_install() {
+	rm_tests() {
+		use test || rm -r "${S}-${EPYTHON//\./_}/lib/tests"
+	}
+	python_foreach_impl rm_tests
+	use test || rm -r "${S}/tests"
+
+	distutils-r1_src_install
+}