#!/bin/bash
set -e

pkg=umap-learn

if [ "${AUTOPKGTEST_TMP}" = "" ] ; then
  AUTOPKGTEST_TMP=$(mktemp -d /tmp/${pkg}-test.XXXXXX)
  trap "rm -rf ${AUTOPKGTEST_TMP}" 0 INT QUIT ABRT PIPE TERM
fi

cp /usr/share/doc/${pkg}/examples/* -a "${AUTOPKGTEST_TMP}"

cd "${AUTOPKGTEST_TMP}"
gunzip -r *

# python3 umap_example.py
set -x
export MPLBACKEND=Agg
python3 inverse_transform_example.py
python3 mnist_torus_sphere_example.py
python3 mnist_transform_new_data.py

file torus.png | fgrep -q 'PNG'
file sphere.png | fgrep -q 'PNG'
