cp2k/tools/docker/scripts/test_i-pi.sh

117 lines
3.1 KiB
Bash
Executable file

#!/bin/bash -e
# author: Ole Schuett
# shellcheck disable=SC1091
source /opt/cp2k-toolchain/install/setup
echo -e "\n========== Installing Dependencies =========="
apt-get update -qq
apt-get install -qq --no-install-recommends \
git \
python3 \
python3-venv \
python3-pip \
python3-wheel \
python3-setuptools
rm -rf /var/lib/apt/lists/*
# Create and activate a virtual environment for Python packages.
python3 -m venv /opt/venv
export PATH="/opt/venv/bin:$PATH"
echo -e "\n========== Installing i-Pi =========="
git clone --quiet --depth=1 --single-branch -b main https://github.com/i-pi/i-pi.git /opt/i-pi
cd /opt/i-pi
pip3 install --quiet .
pip3 install --quiet scipy # missing from i-pi requirements
echo -e "\n========== Running i-Pi Tests =========="
cd /opt/i-pi/examples/clients/cp2k/nvt_cl
set +e # disable error trapping for remainder of script
TIMEOUT_SEC="300"
ulimit -t ${TIMEOUT_SEC} # Limit cpu time.
# launch cp2k
(
mkdir -p run_1
cd run_1
echo 42 > cp2k_exit_code
sleep 10 # give i-pi some time to startup
OMP_NUM_THREADS=2 timeout ${TIMEOUT_SEC} /opt/cp2k/build/bin/cp2k.ssmp ../in.cp2k
echo $? > cp2k_exit_code
) &
# launch i-pi
sed -i "s/total_steps>1000/total_steps>10/" input.xml
# Limit walltime too, because waiting for a connection consumes no cpu time.
timeout ${TIMEOUT_SEC} i-pi input.xml
IPI_EXIT_CODE=$?
wait # for cp2k to shutdown
CP2K_EXIT_CODE=$(cat ./run_1/cp2k_exit_code)
echo ""
echo "CP2K exit code: ${CP2K_EXIT_CODE}"
echo "i-Pi exit code: ${IPI_EXIT_CODE}"
IPI_REVISION=$(git rev-parse --short HEAD)
if ((IPI_EXIT_CODE)) || ((CP2K_EXIT_CODE)); then
echo -e "\nSomething is wrong with i-Pi commit ${IPI_REVISION}."
IPI_TEST_SUCCESS=false
else
echo -e "\ni-Pi commit ${IPI_REVISION} works fine."
IPI_TEST_SUCCESS=true
fi
echo -e "\n========== Running i-Pi Protocol Tests =========="
cd /opt
export CP2K_DATA_DIR="/opt/cp2k/data"
TIMEOUT_SEC="300"
ulimit -t ${TIMEOUT_SEC} # Limit cpu time.
export OMP_NUM_THREADS=2
# launch cp2k in client mode
(
mkdir -p run_client
cd run_client
echo 42 > cp2k_client_exit_code
sleep 10 # give server some time to startup
timeout ${TIMEOUT_SEC} /opt/cp2k/build/bin/cp2k.ssmp /opt/cp2k/tests/i-PI/ipi_client.inp
echo $? > cp2k_client_exit_code
) &
# launch cp2k in server mode
mkdir -p run_server
cd run_server
timeout ${TIMEOUT_SEC} /opt/cp2k/build/bin/cp2k.ssmp /opt/cp2k/tests/i-PI/ipi_server.inp
SERVER_EXIT_CODE=$?
wait # for cp2k client to shutdown
cd /opt
CLIENT_EXIT_CODE=$(cat ./run_client/cp2k_client_exit_code)
echo ""
echo "Client CP2K exit code: ${CLIENT_EXIT_CODE}"
echo "Server CP2K exit code: ${SERVER_EXIT_CODE}"
if ((SERVER_EXIT_CODE)) || ((CLIENT_EXIT_CODE)); then
echo -e "\nSomething is wrong with i-Pi functionality."
COMM_TEST_SUCCESS=false
else
echo -e "\ni-Pi communication works fine."
COMM_TEST_SUCCESS=true
fi
if [ "$IPI_TEST_SUCCESS" = true ] && [ "$COMM_TEST_SUCCESS" = true ]; then
echo -e "\nSummary: i-Pi commit ${IPI_REVISION} works fine."
echo -e "Status: OK\n"
else
echo -e "\nSummary: Something is wrong with i-Pi commit ${IPI_REVISION}."
echo -e "Status: FAILED\n"
fi
#EOF