Added unit tests for district