Add tests for current implementation.