Integer Replacement Python Solution with Tests

lcpy gen -n 397   # by problem number
lcpy gen -s integer_replacement   # by problem name

Input: n = 8
Output: 3
Explanation: 8 -> 4 -> 2 -> 1

Input: n = 7
Output: 4
Explanation: 7 -> 8 -> 4 -> 2 -> 1
or 7 -> 6 -> 3 -> 2 -> 1

Input: n = 4
Output: 2

class Solution:
    # Time: O(log n)
    # Space: O(log n)
    def integer_replacement(self, n: int) -> int:
        ops = 0
        while n != 1:
            if n % 2 == 0:
                n //= 2
            elif n == 3 or n % 4 == 1:
                n -= 1
            else:
                n += 1
            ops += 1
        return ops