Range jako generator

def my_range(*args):
    number_of_args = len(args)
    if number_of_args == 1:
        start, stop, step = 0.0, args[0], 1.0
    elif number_of_args == 2:
        start, stop, step = args[0], args[1], 1.0
    elif number_of_args == 3:
        start, stop, step = args[0], args[1], args[2]
    else:
        raise ValueError("Function needs 1 to 3 arguments.")

    if step == 0:
        raise ValueError("Step argument cannot be 0")

    while step > 0 and start < stop:
        yield start
        start += step
    while step < 0 and start < stop:
        yield stop
        stop += step



def main():
    try:
        for val in my_range(1.1, 2.2, 0.5):
            print(val, end=" ")
        print()
        for val in my_range(1.1, 2.1, 0.5):
            print(val, end=" ")
        print()
        for val in my_range(1.1, 2.2):
            print(val, end=" ")
        print()
        for val in my_range(2.2):
            print(val, end=" ")
        print()
        for val in my_range(1.1, 2.2, -0.5):
            print(val, end=" ")
        print()
        for val in my_range():
            print(val, end=" ")

    except ValueError as e:
        print("Error:", e)

main()