Add tests for ABAAB tree construction