Techo del elemento en un array ordenado

Hola, estoy resolviendo problemas de Estructuras de Datos y Algoritmos y encontré un problema llamado “techo del elemento en un arreglo ordenado”. En este problema, se nos da un arreglo ordenado y si el elemento objetivo está presente en el arreglo ordenado, debemos devolver el objetivo. Si el elemento objetivo no se encuentra en el arreglo ordenado, debemos devolver el elemento más pequeño que sea mayor que el objetivo. He escrito el código y también he realizado algunos casos de prueba, pero necesito verificar si todo funciona correctamente. Este problema no está en LeetCode, donde podría ejecutarlo con diferentes casos. Necesito sugerencias/comentarios sobre si el problema está resuelto de la manera correcta y si daría resultados correctos en todos los casos.

class Solution:
    #Mi enfoque
    def smallestNumberGreaterThanTarget(self, nums, target):
        start = 0
        end = len(nums)-1

        if target > nums[end]:
            return -1
        while start <= end:
            mid = start + (end-start)//2

            if nums[mid] == target:
                return nums[mid]
            elif nums[mid] < target:
                if nums[mid+1] >= target:
                    return nums[mid+1]
                start = mid + 1
            else:
                end = mid-1

        return nums[start]