Question: 
Write a python function to count number of cubes of size k in a cube of size n.


Model's Code:
def No_of_cubes(n, k):
    if k > n:
        return 0
    return (n - k + 1) ** 3



Sample Correct Code:
def No_of_cubes(N,K):
    No = 0
    No = (N - K + 1)
    No = pow(No, 3)
    return No


Asserts:
assert No_of_cubes(2,1) == 8
assert No_of_cubes(5,2) == 64
assert No_of_cubes(1,1) == 1


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

Test 646 of 974: 482/646 (74.61300309597523%): success