Add some unit tests.