Question

This must be a very simple solution that has eluded me this last hour. I’ve tried to build this test function where the return value of the test_cases list should match the values in the test_case_answers list but for some reason, test case 1 and test case 2 fail. When I print the return values for the test cases they return the correct answers, but for some reason test case 1 and test case 2 return False.

import math

test_cases = [1, 9, -3]  
test_case_answers = [1, 3, 0]

def custom_sqrt(num):  
    for i in range(len(test_cases)):  
        if test_cases[i] >= 0:  
            return math.sqrt(test_cases[i])  
        else:  
            return 0

for i in range(len(test_cases)):  
    if custom_sqrt(test_cases[i]) != test_case_answers[i]:  
        print "Test Case #", i, "failed!"


custom_sqrt(test_cases)

Answer 1

You are looping too many times. In custom_sqrt, you try to loop over all test cases, but you return the value for the first one, so you never get to the rest. Remove your loop from custom_sqrt so it’s just:

def custom_sqrt(num):
    if num >= 0:
        return math.sqrt(num)
    else: 
        return 0