remove private testing utils from `CompilationUnit`

[OmarTawfik]

I wonder if we can move the tests to use the AST instead, so that we test the public API directly.

Originally posted by @OmarTawfik in #1721 (comment)