Add tests for Location