#!/bin/sh

EXP_VER="$2"

python3 - << EOF
import sys
import numpy as np

if (np.__version__ != "$EXP_VER"):
    print("Got incorrect version: " + np.__version__)
    sys.exit(1)

arr = np.array([1, 2, 3, 4, 5])

print(arr)

EOF