Sum of ones

def sum_of_ones():
    n = int(input())
    binary = bin(n)[2:]
    length = len(binary)
    result = 0
    
    for i in range(length):
        if binary[i] == '1':
            remaining = length - i - 1
            if remaining > 0:
                result += remaining * (2 ** (remaining - 1))
            if remaining > 0:
                result += int(binary[i+1:], 2) + 1
            else:
                result += 1 
    print(result)

if __name__ == "__main__":
    sum_of_ones()

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED