from __future__ import absolute_import def square(x): return x * x