Add neuron tests