Partial Stream Discharge Calculation

partial.py

L = [.075, .225, .375, .525, .675]
d = [.1, .14, .13, .03, 0]
V = [.88, .75, .35, .07, 0]

i = 0
partials = []


def discharge(L: list[float], d: list[float], V: list[float], i: int, partials: list[float]):
    if len(L) != len(d) != len(V):
        raise ValueError("not same lengths")

    if i == len(L):
        print(partials)
        return sum(partials)

    print(f"i {i+1}")
    Wi = width(L, i)
    print(f"Wi {Wi}")
    Ai = Wi * d[i]
    print(f"Ai {Ai}")
    Qi = V[i] * Ai 
    print(f"Qi {Qi}")
    partials.append(Qi)
    print("\n")
    return discharge(L, d, V, i + 1, partials)

def width(L: list[float], i: int):
    def check_index(to_check: int):
        return to_check if 0 <= to_check < len(L) else i  

    return (L[check_index(i + 1)] - L[check_index(i - 1)]) / 2


discharge(L, d, V, i, partials)