What is the correct way to define a function that calculates the square of a number in

Python?

function square$(x)$: return $x*x$

def square $(x)=x^{**}x$

def square$(x)$: return $x^{****}2$

define square$($x$)$: $x^{****}2$