# Contributor: Iztok Fister, Jr. <iztok@iztok-jr-fister.eu>
# Maintainer: Iztok Fister, Jr. <iztok@iztok-jr-fister.eu>
pkgname=py3-opt_einsum
pkgver=3.4.0
pkgrel=1
pkgdesc="Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization"
url="https://github.com/dgasmith/opt_einsum"
arch="noarch"
license="MIT"
depends="
	py3-numpy
	"
makedepends="
	py3-gpep517
	py3-hatch-fancy-pypi-readme
	py3-hatch-vcs
	py3-hatchling
	py3-wheel
	"
checkdepends="
	py3-pytest-cov
	py3-pytest-xdist
	"
subpackages="$pkgname-doc $pkgname-pyc"
source="https://github.com/dgasmith/opt_einsum/archive/v$pkgver/opt_einsum-$pkgver.tar.gz"
builddir="$srcdir/opt_einsum-$pkgver"

build() {
	export SETUPTOOLS_SCM_PRETEND_VERSION=$pkgver
	gpep517 build-wheel \
		--wheel-dir .dist \
		--output-fd 3 3>&1 >&2
}

check() {
	python3 -m venv --clear --without-pip --system-site-packages .testenv
	.testenv/bin/python3 -m installer .dist/*.whl
	.testenv/bin/python3 -m pytest -n auto
}

package() {
	python3 -m installer -d "$pkgdir" \
		.dist/*.whl

	rm -r "$pkgdir"/usr/lib/python*/site-packages/opt_einsum/tests

	install -Dm644 CITATION.cff -t "$pkgdir"/usr/share/doc/$pkgname
	install -Dm644 README.md -t "$pkgdir"/usr/share/doc/$pkgname

}

sha512sums="
0001a6f95c6101e172eb2ae2492d713517a1e76783b8c9542c0845d97da7181d9fdaa06724d3bd67022835def0d6f0ce68531996c38f18b1f667f67ae1b36957  opt_einsum-3.4.0.tar.gz
"
