leetcode-grind/905/main.py

class Solution:
    
    def makeNegIfEven(self, n):
        if n % 2 == 0:
            return -n
        else:
            return n
        
    def sortArrayByParity(self, A):
        return sorted(A, key = self.makeNegIfEven)

s = Solution()
print("Expected: [6, 4, 2, 1, 3]")
print("Got:", s.sortArrayByParity([1, 2, 3, 4, 6]))