Add basic unit tests.