arithematic_subarray.py

def arithematic_subarray(array):
    if len(array) < 3:
        return 0
    iteration_count = 1
    difference = array[1] - array[0]
    res = 0
    for i in range(2, len(array)):
        diff = array[i] - array[i-1]
        if diff == difference:
            iteration_count += 1
        else:
            difference = diff
            iteration_count = 1

        if iteration_count is 2:
            res += 1
            print(array[i-2:i+1])
        elif iteration_count > 2:
            res += 1 + (iteration_count - 2)

    return res