Move tests to separate script