add tests

2 files changed, 14 insertions, 6 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 3fa7cb167dad..0802671afb17 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -5,6 +5,8 @@ pkgbase = python-croniter
 	url = https://github.com/kiorky/croniter
 	arch = any
 	license = MIT
+	checkdepends = python-pytest
+	checkdepends = python-pytz
 	makedepends = python-setuptools
 	depends = python-dateutil
 	depends = python-natsort
diff --git a/PKGBUILD b/PKGBUILD
index 065499b7d53f..f65757b03a0d 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -10,18 +10,24 @@ url="https://github.com/kiorky/croniter"
 license=('MIT')
 depends=('python-dateutil' 'python-natsort')
 makedepends=('python-setuptools')
+checkdepends=('python-pytest' 'python-pytz')
 source=("https://pypi.org/packages/source/${_name:0:1}/$_name/$_name-$pkgver.tar.gz")
 sha256sums=('a70dfc9d52de9fc1a886128b9148c89dd9e76b67d55f46516ca94d2d73d58219')
 
 build() {
-	cd "$_name-$pkgver"
-	python setup.py build
+  cd "$_name-$pkgver"
+  python setup.py build
+}
+
+check() {
+  cd "$_name-$pkgver"
+  pytest src
 }
 
 package() {
-	cd "$_name-$pkgver"
-	export PYTHONHASHSEED=0
-	python setup.py install --root="$pkgdir" --optimize=1 --skip-build
+  cd "$_name-$pkgver"
+  export PYTHONHASHSEED=0
+  python setup.py install --root="$pkgdir" --optimize=1 --skip-build
 
-	install -Dm644 docs/LICENSE -t "$pkgdir/usr/share/licenses/$pkgname"
+  install -Dm644 docs/LICENSE -t "$pkgdir/usr/share/licenses/$pkgname"
 }