unit test for softmax output