pass test for simple cases with new ast api