Add tests

2 files changed, 12 insertions, 8 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 5caccfdf0fcc..21c0b5867e95 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -11,13 +11,7 @@ pkgbase = python-vdf
 
 pkgname = python-vdf
 	depends = python
-	depends = python-nose
-	depends = python-coverage
-	depends = python-mock
 
 pkgname = python2-vdf
 	depends = python2
-	depends = python2-nose
-	depends = python2-coverage
-	depends = python2-mock
 
diff --git a/PKGBUILD b/PKGBUILD
index 5448e5ce1586..5524dad02d36 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -13,14 +13,24 @@ makedepends=('python-setuptools')
 source=("https://github.com/ValvePython/vdf/archive/v$pkgver.tar.gz")
 sha256sums=('d129edb0ab0cac55f9cd632959f05b8cbbe0babe017cf9671b8b379ec3186148')
 
+check_python-vdf() {
+  checkdepends=('python-nose' 'python-coverage' 'python-mock')
+  python setup.py test
+}
+
+check_python2-vdf() {
+  checkdepends=('python2-nose' 'python2-coverage' 'python2-mock')
+  python2 setup.py test
+}
+
 package_python-vdf() {
-  depends=('python' 'python-nose' 'python-coverage' 'python-mock')
+  depends=('python')
   cd "$srcdir/$_pkgname-$pkgver"
   python setup.py install --optimize=1 --root="${pkgdir}/"
 }
 
 package_python2-vdf() {
-  depends=('python2' 'python2-nose' 'python2-coverage' 'python2-mock')
+  depends=('python2')
   cd "$srcdir/$_pkgname-$pkgver"
   python2 setup.py install --optimize=1 --root="${pkgdir}/"
 }