from nose.tools import assert_equal
assert_equal(squares(1), [1])
assert_equal(squares(2), [1, 4])
from nose.tools import assert_raises
assert_raises(ValueError, squares, 0)
assert_raises(ValueError, squares, -1)
assert_equal(squares(10), [1, 4, 9, 16, 25, 36, 49, 64, 81, 100])