Added axon unit tests, see #2