def test_neighbors_is_none_and_one_when_there_is_a_greater_neighbor(self):
session = Client().session()
album = Album(name='Light me up')
session.add(album)
session.flush()
image = Image(filename='f1acc1d', album_id=album.album_id)
session.add(image)
next = Image(filename='1ace', album_id=album.album_id)
session.add(next)
session.flush()
eq_((None, next), image.neighbors())
def test_neighbors_is_one_and_none_when_the_image_has_a_lesser_neighbor(self):
session = Client().session()
album = Album(name='Light me up')
session.add(album)
session.flush()
prev = Image(filename='1ace', album_id=album.album_id)
session.add(prev)
image = Image(filename='f1acc1d', album_id=album.album_id)
session.add(image)
session.flush()
eq_((prev, None), image.neighbors())
def test_neighbors_skips_over_images_from_other_albums(self):
session = Client().session()
hobbiton = Album(name='Hobbiton')
session.add(hobbiton)
hardbottle = Album(name='Hardbottle')
session.add(hardbottle)
session.flush()
samwise = Image(filename='5411111115e', album_id=hobbiton.album_id)
session.add(samwise)
bilbo = Image(filename='b11b0', album_id=hobbiton.album_id)
session.add(bilbo)
lobelia = Image(filename='10be11a', album_id=hardbottle.album_id)
session.add(lobelia)
frodo = Image(filename='f0d0', album_id=hobbiton.album_id)
session.add(frodo)
session.flush()
eq_((samwise, frodo), bilbo.neighbors())
def test_neighbors_is_nones_when_the_image_has_no_album(self):
image = Image(filename='faceface')
eq_((None, None), image.neighbors())
