RankProbabilityDataset extends DefaultCategoryDataset { @Override public List<String> getRowKeys() { List<String> keys = new ArrayList<String>(); for (int i = 0; i < d_summary.getTreatments().size(); ++i) { keys.add("Rank " + (i+1)); } return keys; } RankProbabilityDataset(RankProbabilitySummary rankProbabilitySummary, boolean preferDescription);  RankProbabilityDataset(RankProbabilitySummary rankProbabilitySummary); @SuppressWarnings("rawtypes") @Override int getRowIndex(Comparable key); @SuppressWarnings("rawtypes") @Override int getColumnIndex(Comparable key); @Override String getRowKey(int row); @Override String getColumnKey(int column); @Override List<String> getRowKeys(); @Override List<String> getColumnKeys(); @Override int getRowCount(); @Override int getColumnCount(); @Override Number getValue(int row, int column); @SuppressWarnings("rawtypes") @Override Number getValue(Comparable rowKey, Comparable columnKey);  }
@Test public void testGetRowKeys() { List<String> keys = new ArrayList<String>(); for(int i = 0; i < d_summary.getTreatments().size(); ++i) keys.add("Rank " + (i+1)); assertEquals(keys, d_dataSet.getRowKeys()); }