Daniel Lubarov

Mystery code

Can you tell what this does without running it?

from itertools import *
from operator import *

def fs():
  return filter(f, chain(range(2), islice(count(), 1, None)))

def f(n):
  return [1 for o in [4, -4] if p(5*n*n + o)]

def p(n):
  return [d for d in range(n) if d*d == n]

for n in fs():
  print(n)

(Run with Python 3.)