@Test
public void predResultMatrixTest(){

	Row[] rows =
		new Row[] {
			Row.of("prefix0", "prefix2"),
			Row.of("prefix0", "prefix0"),
			Row.of("prefix1", "prefix1"),
			Row.of("prefix2", "prefix0"),
			Row.of("prefix2", "prefix0"),
			Row.of("prefix2", "prefix2"),
			Row.of("prefix2", "prefix0"),
			Row.of("prefix1", "prefix2"),
			Row.of("prefix1", "prefix1"),
			Row.of("prefix1", "prefix1")
		};
	Map<Object, Integer> map = new HashMap<>();
	map.put("prefix0", 2);
	map.put("prefix1", 1);
	map.put("prefix2", 0);
	MultiMetricsSummary metrics = EvaluationUtil.getMultiClassMetrics(Arrays.asList(rows), Tuple2.of(map, new Object[]{"prefix2", "prefix1", "prefix0"}));
	Assert.assertEquals(10, metrics.total);
	Assert.assertArrayEquals(new Object[] {"prefix2", "prefix1", "prefix0"}, metrics.labels);
	Assert.assertArrayEquals(new long[] {1, 1, 1}, metrics.matrix.getMatrix()[0]);
	Assert.assertArrayEquals(new long[] {0, 3, 0}, metrics.matrix.getMatrix()[1]);
	Assert.assertArrayEquals(new long[] {3, 0, 1}, metrics.matrix.getMatrix()[2]);
}