Индекс столбца в матрице

  public static int index(int m[][]) {
        int max = 0,
                min = 0,
                s = 0,
                indexmax = 0;
 
        for (int i = 0; i < m.length; i++) {
            s = 0;
            for (int j = 0; j < m.length; j++) {
                s = s + m[j][i];
            }
            if (max <= s || max == 0) {
                max = s;
                indexmax=i;
                
 
            }
            
 
        }
        return indexmax;
    }
 
    public static void main(String[] args) {
        int m[][] = ArrayUtils.readIntArray2FromConsole();
        index(m);
        System.out.printf("%d\n", index(m));
    }
 
}

import org.hamcrest.core.Is;
import org.junit.Test;
 
import static org.junit.Assert.assertThat;
 
public class StolbarayTest {
 
    @Test
    public void index() {
        for (int i = 0; i < 500; i++) {
            int[][] arr = new int[][]{
                    {1, 1, 1, 1},
                    {1, 2, 2, 2},
                    {1, 2, 2, 1},
                    {1, 1, 1, 1}};
            Stolbaray res = new Stolbaray();
            assertThat(res.index(arr), Is.is(2));
        }
    }
 
}