add regression prediction tests