Add tests for world