Add WIP unit tests.