#!/bin/bash

# Pytorch test suite

HOME_DIR=/test
LOG_FILE=/test/logs.txt
LOG_DIR=`dirname $LOG_FILE`

mkdir -p ${HOME_DIR}/artifacts
mkdir -p ${LOG_DIR}

set -e

cp -r /gaudi-test-suite ${HOME_DIR}/artifacts/gaudi-test-suite
cd ${HOME_DIR}/artifacts/gaudi-test-suite
git checkout r${GIT_BRANCH}
python3 -m pip install -r requirements.txt
python3 -m pip install -r test/framework/tensorflow_test_suite/requirements.txt

python3 gaudi_test_suite.py -tm framework -f tf 2>&1 | tee ${LOG_FILE}
RETURNVAL=${PIPESTATUS[0]}
if [ $RETURNVAL -eq 0 ];
then
    echo "Tensorflow framework test suite executed successfully"
    echo "Kudos!! Habana tests executed successfully" >> "${LOG_FILE}"
    exitcode=0
else
    echo "Tensorflow framework test suite terminated unsuccessfully"
    exitcode=$RETURNVAL
fi

rm -rf ${HOME_DIR}/artifacts/gaudi-test-suite

exit ${exitcode}