Improved neuron data gen tests