Forgot to import numpy in test script; corrected this.