#!/bin/bash

# Continue on error
set -e

# Build with progress output and capture the tag
TAG=$( \
docker build \
    --target test \
    . \
    2>&1 | tee /dev/tty | grep "writing image" | cut -d' ' -f4 \
)

# Exit if tag is empty
[ -z "$TAG" ] && exit 1

# Run tests
docker run \
    --rm \
    --gpus=all \
    -v /$(pwd)/model/:/root/model/ \
    $TAG

# Clean up image
[ ! -z "$TAG" ] && docker rmi $TAG