s434730/RandomSec
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix Recon features computation for empty recons

Browse Source
This commit is contained in:
Antonin Delpeuch 2019-02-22 09:15:42 +00:00
parent 7b2e2b5894
commit deb7335848
2 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main/src/com/google/refine/model/recon/StandardReconConfig.java
View File

@ -566,8 +566,8 @@ public class StandardReconConfig extends ReconConfig {
* @param text
* the cell value to compare the reconciliation data to
*/
public void computeFeatures(Recon recon, String text) {
if (!recon.candidates.isEmpty()) {
public void computeFeatures(Recon recon, String text) {
if (recon.candidates != null && !recon.candidates.isEmpty()) {
ReconCandidate candidate = recon.candidates.get(0);
recon.setFeature(Recon.Feature_nameMatch, text.equalsIgnoreCase(candidate.name));
@ -587,7 +587,7 @@ public class StandardReconConfig extends ReconConfig {
} else {
recon.features = new Object[Recon.Feature_max];
}
}
}
static protected double wordDistance(String s1, String s2) {
Set<String> words1 = breakWords(s1);

12
main/tests/server/src/com/google/refine/tests/model/recon/StandardReconConfigTests.java
View File

@ -27,6 +27,7 @@
package com.google.refine.tests.model.recon;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertNull;
import static org.testng.Assert.assertTrue;
@ -333,4 +334,15 @@ public class StandardReconConfigTests extends RefineTest {
assertEquals(recon.candidates.get(0).score, 0.3);
assertEquals(recon.candidates.get(0).id, "18951129");
}
/**
* computing the features on an empty recon should not fail
*/
@Test
public void testComputeFeatures() {
StandardReconConfigStub stub = new StandardReconConfigStub();
Recon recon = stub.createNewRecon(2384738L);
stub.computeFeatures(recon, "my string");
assertNotNull(recon.features);
}
}