Hackerrank | Sherlock and Squares

from math import *

# Complete the squares function below.

def squares(a, b):

    return floor(sqrt(b) - ceil(sqrt(a)) + 1