Add check function

2 files changed, 9 insertions, 1 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 31a99009bb3e..bd1161fb3314 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -5,6 +5,8 @@ pkgbase = python-binance
 	url = https://github.com/sammchardy/python-binance
 	arch = any
 	license = MIT
+	checkdepends = python-pytest
+	checkdepends = python-requests-mock
 	makedepends = python-setuptools
 	depends = python
 	depends = python-aiohttp
diff --git a/PKGBUILD b/PKGBUILD
index 2aa98907d055..f50aa1be07a0 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -10,8 +10,9 @@ license=('MIT')
 depends=('python' 'python-aiohttp' 'python-dateparser'
          'python-requests' 'python-ujson' 'python-websockets=9.1')
 makedepends=('python-setuptools')
+checkdepends=('python-pytest' 'python-requests-mock')
 provides=("$pkgname")
-source=("$pkgname-$pkgver.tar.gz::https://github.com/sammchardy/$pkgname/archive/v$pkgver.tar.gz")
+source=("$pkgname-$pkgver.tar.gz::$url/archive/v$pkgver.tar.gz")
 sha512sums=('2b059b763a380e9fe2529a6a9621acbbd180594ea54f1310ba0a8e266373ce1b13a9138bddd8232a77f1265216e0c3d5cb45e4a0c13840929a4e99c41a83829a')
 
 build() {
@@ -19,6 +20,11 @@ build() {
     python setup.py build
 }
 
+check() {
+    cd $pkgname-$pkgver
+    pytest
+}
+
 package() {
     cd $pkgname-$pkgver
     python setup.py install --root="$pkgdir" --optimize=1 --skip-build