평균 구하기

test cases 를 위한 함수도 작성해보았다.

def solution(arr):
    n = len(arr) 
    totalSum = sum(arr)  
    answer = totalSum / n
    return answer

def run_tests():
    # Simple test cases
    simple_test_cases = [
        ([1, 2, 3]),
        ([0, 0, 0]),
        ([5]),
        ([-5, -5, -5])
    ]
    for test in simple_test_cases:
        print(solution(test))
        
run_tests()