Added unit tests from original repo.