Fixed broken tests in test_nonlinearities.

- to make this cleaner, I made Neurons directly probable.