Question: 
Write a function to count number of lists in a given list of lists and square the count.


Model's Code:
def count_list(lst):
    return len(lst) ** 2



Sample Correct Code:
def count_list(input_list): 
    return (len(input_list))**2


Asserts:
assert count_list([[0], [1, 3], [5, 7], [9, 11], [13, 15, 17]])==25
assert count_list([[1, 3], [5, 7], [9, 11], [13, 15, 17]] )==16
assert count_list([[2, 4], [[6,8], [4,5,8]], [10, 12, 14]])==9


Tests:
Test 1 of 3: PASS
Test 2 of 3: PASS
Test 3 of 3: PASS

Test 858 of 974: 660/858 (76.92307692307693%): success