# Contributor: Iztok Fister, Jr. <iztok@iztok-jr-fister.eu>
# Maintainer: Iztok Fister, Jr. <iztok@iztok-jr-fister.eu>
pkgname=py3-imbalanced-learn
_pkgorig=imbalanced-learn
pkgver=0.10.1
pkgrel=1
pkgdesc="Toolbox for imbalanced dataset in machine learning"
url="https://imbalanced-learn.org"
# py3-scikit-learn is missing on s390x
arch="noarch !s390x"
license="MIT"
depends="
	python3
	py3-numpy
	py3-pandas
	py3-scikit-learn
	py3-scipy
	"
makedepends="py3-setuptools"
checkdepends="py3-pytest"
source="$pkgname-$pkgver.tar.gz::https://files.pythonhosted.org/packages/source/i/imbalanced-learn/imbalanced-learn-$pkgver.tar.gz
	exclude-tests.patch"
builddir="$srcdir/$_pkgorig-$pkgver"

build() {
	python3 setup.py build
}

check() {
	local _skipped_tests="
		test_all_estimators
		and not test_classification_report_imbalanced_multiclass_with_unicode_label
		and not test_rusboost
		and not test_cluster_centroids_n_jobs
		and not test_fit_docstring
		and not keras
		and not test_function_sampler_validate
		and not test_fetch
		and not check_samplers_preserve_dtype
		and not test_pipeline_fit_then_sample_with_sampler_last_estimator
		and not test_pipeline_fit_then_sample_3_samplers_with_sampler_last_estimator"

	# shellcheck disable=SC2116
	pytest -v -k "not $(echo $_skipped_tests)"
}

package() {
	python3 setup.py install --skip-build --root="$pkgdir"
}

sha512sums="
8e5cdf4dc751faa949fc250eee9c78f5188c991491cef2abf6efa1f9e510fc6d1cc8b2b385b35066979104baddb682e6365dd349a3d47ddf68239d4bd1034fd2  py3-imbalanced-learn-0.10.1.tar.gz
57a181476c728451b4c54ae535d0f236ac7228b525c8dcbe118bfc8837f1f270fd8185beef59dcbf8fe9c12a67af9c4064339de5df1caf18743c1a4c475db9b0  exclude-tests.patch
"