def k_imagenes_cercanas(x, k, lista_fotos): distancias = [(abs(x - foto), foto) for foto in lista_fotos] distancias.sort() k_imagenes = [distancia[1] for distancia in distancias[:k]] return k_imagenes

import unittest
class TestKImagenesCercanas(unittest.TestCase):
    def test_k_imagenes_cercanas(self):
        lista_fotos = [0, 3, 7, 8, 10]
        self.assertEqual(k_imagenes_cercanas(4, 2, lista_fotos), [3, 7])