Create argmax_test.sv