def test_close_by_5(self):
hotel_1 = Hotel.query.filter_by(hotel_id=1).first()
places_data = close_places(hotel_1, "restaurant", 0)
self.assertEqual(len(places_data), 0)
def test_close_by_3(self):
restaurant_1 = Restaurant.query.filter_by(restaurant_id=1).first()
places_data = close_places(restaurant_1, "attraction", 5)
self.assertEqual(len(places_data), 5)
def test_close_by_2(self):
restaurant_1 = Restaurant.query.filter_by(restaurant_id=1).first()
places_data = close_places(restaurant_1, "hotel", 0)
self.assertEqual(len(places_data), 0)
def test_close_by_9(self):
attraction_1 = Attraction.query.filter_by(attraction_id=10).first()
places_data = close_places(attraction_1, "hotel", 10)
self.assertEqual(len(places_data), 10)
def test_close_by_8(self):
attraction_1 = Attraction.query.filter_by(attraction_id=1).first()
places_data = close_places(attraction_1, "restaurant", 0)
self.assertEqual(len(places_data), 0)
def test_close_by_6(self):
hotel_1 = Hotel.query.filter_by(hotel_id=10).first()
places_data = close_places(hotel_1, "attraction", 10)
self.assertEqual(len(places_data), 10)