def test_get_place_names_has_places(self): tag = TagFactory() place = PlaceFactory() place.tags.add(tag) place = PlaceFactory() place.tags.add(tag) place = PlaceFactory() place.tags.add(tag) places = tag.get_place_names() self.assertEqual(3, places.count()) places_list = list(places) self.assertEqual(['Some Place 1', 'Some Place 2', 'Some Place 3'], places_list)
def test_get_place_names_no_places(self): tag = TagFactory() places = tag.get_place_names() self.assertEqual(0, places.count())