# your code goes here
import numpy
def mySquare(x):
if isinstance(x, int):
return x**2
elif isinstance(x, range):
return [i ** 2 for i in x]
print(mySquare(2));
print(mySquare(range(10)));
print(mySquare(numpy.zeros(10)));
IyB5b3VyIGNvZGUgZ29lcyBoZXJlCmltcG9ydCBudW1weQoKZGVmIG15U3F1YXJlKHgpOgoJaWYgaXNpbnN0YW5jZSh4LCBpbnQpOgoJCXJldHVybiB4KioyCgllbGlmIGlzaW5zdGFuY2UoeCwgcmFuZ2UpOgoJICAgIHJldHVybiBbaSAqKiAyIGZvciBpIGluIHhdCgoJIAoJICAgIApwcmludChteVNxdWFyZSgyKSk7CnByaW50KG15U3F1YXJlKHJhbmdlKDEwKSkpOwpwcmludChteVNxdWFyZShudW1weS56ZXJvcygxMCkpKTs=