### Question:
TableIndex { public boolean isPrimaryKey(String fieldName) { if(fieldName == null) return false; return fieldName.equals(partitionKey) || fieldName.equals(sortKey); } TableIndex(String name, String partitionKey);  TableIndex(String name, String partitionKey, String sortKey); String getName(); String getPartitionKey(); String getSortKey(); boolean isPrimaryKey(String fieldName);  }### Answer:
@Test(dataProvider = "isPrimaryKeyProvider") public void testIsPrimaryKey(TableIndex tableIndex, String field, final boolean EXPECTED_RESULT) { Assert.assertEquals(tableIndex.isPrimaryKey(field), EXPECTED_RESULT); }