add an assert statement in the test of the h_softmax to checks that fun_output_tg and fun_output are consistent