Improve tests and add AST assertions with nightly features

[doonv]

Objective

There isn't an easy way to assert the AST in unit tests.

Solution

Use the nightly box_pattern feature in tests (This doesn't make the library itself nightly, only the tests)