Fix tests

2 files changed, 15 insertions, 3 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 21c0b5867e95..dbd632de321f 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -5,6 +5,12 @@ pkgbase = python-vdf
 	url = https://github.com/ValvePython/vdf
 	arch = any
 	license = MIT
+	checkdepends = python-nose
+	checkdepends = python-coverage
+	checkdepends = python-mock
+	checkdepends = python2-nose
+	checkdepends = python2-coverage
+	checkdepends = python2-mock
 	makedepends = python-setuptools
 	source = https://github.com/ValvePython/vdf/archive/v2.3.tar.gz
 	sha256sums = d129edb0ab0cac55f9cd632959f05b8cbbe0babe017cf9671b8b379ec3186148
diff --git a/PKGBUILD b/PKGBUILD
index 5524dad02d36..43dae8a8a07b 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -10,12 +10,18 @@ url='https://github.com/ValvePython/vdf'
 license=('MIT')
 depends=()
 makedepends=('python-setuptools')
+checkdepends=('python-nose' 'python-coverage' 'python-mock' 'python2-nose' 'python2-coverage' 'python2-mock')
 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() {
+  cd "$srcdir/$_pkgname-$pkgver"
+  if [ $pkgname = python-vdf ]
+  then
+    python setup.py test
+  else
+    python2 setup.py test
+  fi
 }
 
 check_python2-vdf() {