remove PIL + add test

2 files changed, 12 insertions, 9 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 0d606a4ed8ba..5ee6201bd358 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,14 +1,14 @@
 pkgbase = python-kornia-git
 	pkgdesc = Open Source Differentiable Computer Vision Library for PyTorch
-	pkgver = v0.2.0.r8.ge110f3b
+	pkgver = 0.2.1.r1.g9797eca
 	pkgrel = 1
 	url = https://github.com/kornia/kornia
 	arch = any
 	license = Apache
+	checkdepends = python-pytest
 	makedepends = git
 	makedepends = python-setuptools
 	depends = python-pytorch
-	depends = python-pillow
 	depends = python-torchvision
 	depends = opencv
 	depends = python-matplotlib
diff --git a/PKGBUILD b/PKGBUILD
index 99a344757919..308c98c4b3ab 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,17 +2,15 @@
 
 pkgname=python-kornia-git
 _name=kornia
-pkgver=v0.2.0.r8.ge110f3b
+pkgver=0.2.1.r1.g9797eca
 pkgrel=1
 arch=(any)
 url='https://github.com/kornia/kornia'
 pkgdesc='Open Source Differentiable Computer Vision Library for PyTorch'
 license=(Apache)
-makedepends=(
-  'git'
-  'python-setuptools'
-)
-depends=('python-pytorch' 'python-pillow' 'python-torchvision' 'opencv' 'python-matplotlib' 'ipython' 'jupyter')
+makedepends=('git' 'python-setuptools')
+depends=('python-pytorch' 'python-torchvision' 'opencv' 'python-matplotlib' 'ipython' 'jupyter')
+checkdepends=('python-pytest')
 provides=('python-kornia')
 conflicts=('python-kornia')
 options=(!emptydirs)
@@ -21,7 +19,12 @@ sha256sums=('SKIP')
 
 pkgver() {
   cd "${_name}"
-    git describe --long --tags | sed 's/\([^-]*-g\)/r\1/;s/-/./g'
+  git describe --long --tags | sed 's/\([^-]*-g\)/r\1/;s/-/./g'
+}
+
+check() {
+  cd "${srcdir}/${_name}-${pkgver}"
+  pytest
 }
 
 package() {