Add tests for location helper