Equilateral numbers

import math
 
k = int(input())
 
def eq(n):
    return math.comb(n,2)
 
def inner(target):
    if target == 0:
        return 0
 
    i = 0
    e = eq(i)
 
    while e < target:
        i += 1
        e = eq(i)
    
    if e == target:
        return 1
    elif e > target:
        one_smaller = eq(i-1)
        diff = target - one_smaller
        return 1 + inner(diff)
 
    elif e < target:
        return 1 + inner(target-e)
 
print(inner(k))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: ACCEPTED

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER