add ast tests