def square(x):
    """
Calculate the square of a number

Parameters
----------

x : a number (int, float, complex...) or numpy array

Outputs
-------

The number squared,

.. math::
    x^2

Example
-------
>>> square(5)
25
"""
    return x**2