>>> C = lambda n: [round(math.sqrt(n) * math.cos(math.pi/12)), round(math.sqrt(n) * math.sin(math.pi/12))]
>>> C(17)
[4.0, 1.0]
>>> foo = lambda n: abs(C(n)[0]**2 + C(n)[1]**2 - 2*(C(n)[0] - C(n)[1])**2) == 1
>>> foo(17)
True
>>> filter(foo, range(200))
[1, 2, 14, 15, 16, 17, 18, 19, 20, 21]
>>> C(21)
[4.0, 1.0]
>>> filter(foo, range(2000))
[1, 2, 14, 15, 16, 17, 18, 19, 20, 21, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257]