test.py

import numpy as np

def test():
    print("Hello, World!")
    return np.array([1, 2, 3])