Simplify the AB code and add tests for it.

Re-write the AB method implementation to use SimpleCodeBuilder. This
reduces the size and complexity of the code in the AB implementation.

Add tests for AB methods. Reuse some of the code from the RK tests for
this purpose.

Bug: #48