Doctest cosine_nn.py