#!/bin/bash

set -e

rm -rf /tmp/python-build

docker rm jupyter_test || true

mkdir -p /tmp/python-build/tmp

mkdir -p /tmp/python-build/devshm

mkdir -p /tmp/python-build/working

# Check that Jupyter server can run; if it dies on startup, the `docker kill` command will throw an error.

docker run \

-d \

--name=jupyter_test \

--read-only \

--net=none \

-e HOME=/tmp \

-v $PWD:/input:ro \

-v /tmp/python-build/working:/working \

-w=/working \

-v /tmp/python-build/tmp:/tmp \

-v /tmp/python-build/devshm:/dev/shm \

kaggle/python-build \

jupyter notebook --allow-root --ip="*"

sleep 3

docker kill jupyter_test && docker rm jupyter_test

docker run \

--rm \

-t \

--read-only \

--net=none \

-e HOME=/tmp \

-e KAGGLE_DATA_PROXY_TOKEN=test-key \

-e KAGGLE_DATA_PROXY_URL=http://127.0.0.1:8000 \

-e KAGGLE_DATA_PROXY_PROJECT=test \

-e EXPECT_GPU="$EXPECT_GPU" \

-v $PWD:/input:ro \

-v /tmp/python-build/working:/working \

-w=/working \

-v /tmp/python-build/tmp:/tmp \

-v /tmp/python-build/devshm:/dev/shm \

kaggle/python-build \

/bin/bash -c 'python /input/test_build.py'

# The test_build.py script creates a plot called plot1.png; check that it exists.

[ -s /tmp/python-build/working/plot1.png ] || exit 1