108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
""" Test PlaceInfo """
|
|
import pytest
|
|
|
|
from osxphotos._constants import _UNKNOWN_PERSON
|
|
|
|
|
|
PHOTOS_DB = "./tests/Test-Places-Catalina-10_15_1.photoslibrary/database/photos.db"
|
|
|
|
UUID_DICT = {
|
|
"place_dc": "128FB4C6-0B16-4E7D-9108-FB2E90DA1546",
|
|
"place_maui": "FF7AFE2C-49B0-4C9B-B0D7-7E1F8B8F2F0C",
|
|
"no_place": "A9B73E13-A6F2-4915-8D67-7213B39BAE9F",
|
|
}
|
|
|
|
|
|
def test_place_place_info_1():
|
|
# test valid place info
|
|
import osxphotos
|
|
|
|
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
|
|
photo = photosdb.photos(uuid=[UUID_DICT["place_dc"]])[0]
|
|
assert photo.place is not None
|
|
assert isinstance(photo.place, osxphotos.placeinfo.PlaceInfo)
|
|
assert not photo.place.ishome
|
|
assert photo.place.name == "2038 18th St NW"
|
|
assert photo.place.names == [
|
|
"2038 18th St NW",
|
|
"Adams Morgan",
|
|
"Washington",
|
|
"Washington",
|
|
"Washington",
|
|
"District of Columbia",
|
|
"United States",
|
|
]
|
|
assert photo.place.country_code == "US"
|
|
assert (
|
|
photo.place.address_str
|
|
== "2038 18th St NW, Washington, DC 20009, United States"
|
|
)
|
|
assert photo.place.address.city == "Washington"
|
|
assert photo.place.address.country == "United States"
|
|
assert photo.place.address.postal_code == "20009"
|
|
assert photo.place.address.state == "DC"
|
|
assert photo.place.address.street == "2038 18th St NW"
|
|
assert photo.place.address.sub_administrative_area is None
|
|
assert photo.place.address.sub_locality == "Adams Morgan"
|
|
assert photo.place.address.iso_country_code == "US"
|
|
|
|
|
|
def test_place_place_info_2():
|
|
# test valid place info
|
|
import osxphotos
|
|
|
|
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
|
|
photo = photosdb.photos(uuid=[UUID_DICT["place_maui"]])[0]
|
|
|
|
assert isinstance(photo.place, osxphotos.placeinfo.PlaceInfo)
|
|
assert photo.place is not None
|
|
assert not photo.place.ishome
|
|
assert photo.place.name == "3700 Wailea Alanui Dr"
|
|
assert photo.place.names == [
|
|
"3700 Wailea Alanui Dr",
|
|
"Wailea",
|
|
"Kihei",
|
|
"Kihei",
|
|
"Maui",
|
|
"Maui",
|
|
"Hawai'i",
|
|
"United States",
|
|
]
|
|
assert photo.place.country_code == "US"
|
|
assert (
|
|
photo.place.address_str
|
|
== "3700 Wailea Alanui Dr, Kihei, HI 96753, United States"
|
|
)
|
|
assert type(photo.place.address) == osxphotos.placeinfo.PostalAddress
|
|
assert photo.place.address.city == "Kihei"
|
|
assert photo.place.address.country == "United States"
|
|
assert photo.place.address.postal_code == "96753"
|
|
assert photo.place.address.state == "HI"
|
|
assert photo.place.address.street == "3700 Wailea Alanui Dr"
|
|
assert photo.place.address.sub_administrative_area == "Maui"
|
|
assert photo.place.address.sub_locality is None
|
|
assert photo.place.address.iso_country_code == "US"
|
|
|
|
|
|
def test_place_no_place_info():
|
|
# test valid place info
|
|
import osxphotos
|
|
|
|
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
|
|
photo = photosdb.photos(uuid=[UUID_DICT["no_place"]])[0]
|
|
|
|
assert photo.place is None
|
|
|
|
|
|
def test_place_str():
|
|
# test __str__
|
|
import osxphotos
|
|
|
|
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
|
|
photo = photosdb.photos(uuid=[UUID_DICT["place_dc"]])[0]
|
|
|
|
assert (
|
|
str(photo.place)
|
|
== "PlaceInfo(name='2038 18th St NW', names='['2038 18th St NW', 'Adams Morgan', 'Washington', 'Washington', 'Washington', 'District of Columbia', 'United States']', country_code='US', ishome='False', address_str='2038 18th St NW, Washington, DC 20009, United States', address='PostalAddress(street='2038 18th St NW', sub_locality='Adams Morgan', city='Washington', sub_administrative_area=None, state='DC', postal_code='20009', country='United States', iso_country_code='US')')"
|
|
)
|