Convert to AST-comparison for tests

[whichoneiwonder]

The tests currently compare raw strings.
Perhaps they should be an AST-comparison instead