@Test(timeout = 1000)
public void testCrmiCli_precisionCheck() throws Exception {
System.out.println("Testing CrMi(beta=1, gamma=0) equals PrecisionMi from main method.");
File expectedOutput = new File(TEST_OUTPUT_DIR, FRUIT_NAME + ".CrMi-precision-expected");
File crmiOutput = new File(TEST_OUTPUT_DIR, FRUIT_NAME + ".CrMi-precision-actual");
crmiOutput.delete();
expectedOutput.delete();
try {
enableExistTrapping();
Main.main(
new String[] {
"allpairs",
"--charset",
"UTF-8",
"--measure",
"RecallMi",
"--measure-reversed",
"--input",
TEST_FRUIT_ENTRY_FEATURES.toString(),
"--input-features",
TEST_FRUIT_FEATURES.toString(),
"--input-entries",
TEST_FRUIT_ENTRIES.toString(),
"--output",
expectedOutput.toString()
});
Main.main(
new String[] {
"allpairs",
"--charset",
"UTF-8",
"--measure",
"CrMi",
"--crmi-beta",
"1.00",
"--crmi-gamma",
"0.00",
"--input",
TEST_FRUIT_ENTRY_FEATURES.toString(),
"--input-features",
TEST_FRUIT_FEATURES.toString(),
"--input-entries",
TEST_FRUIT_ENTRIES.toString(),
"--output",
crmiOutput.toString()
});
} finally {
disableExitTrapping();
}
assertTrue(WeightedTokenPairSource.equal(expectedOutput, crmiOutput, DEFAULT_CHARSET));
}
private void testCrmiCli(double beta, double gamma) throws Exception {
System.out.println("Testing CrMi (beta=" + beta + ", gamma=" + gamma + ") from main method.");
File output =
new File(TEST_OUTPUT_DIR, FRUIT_NAME + ".CrMi-beta_" + beta + "-gamma_" + gamma + "");
output.delete();
try {
enableExistTrapping();
Main.main(
new String[] {
"allpairs",
"--charset",
"UTF-8",
"--measure",
"CrMi",
"--crmi-beta",
"" + beta,
"--crmi-gamma",
"" + gamma,
"--input",
TEST_FRUIT_ENTRY_FEATURES.toString(),
"--input-features",
TEST_FRUIT_FEATURES.toString(),
"--input-entries",
TEST_FRUIT_ENTRIES.toString(),
"--output",
output.toString()
});
} finally {
disableExitTrapping();
}
assertTrue("Output file " + output + " does not exist.", output.exists());
assertTrue("Output file " + output + " is empty.", output.length() > 0);
}
@Test(timeout = 1000)
public void testCrmiCli_diceCheck() throws Exception {
System.out.println(
"Testing CrMi(beta=NA, gamma=1) \"harmonic mean\" equals DiceMi from main method.");
File expectedOutput = new File(TEST_OUTPUT_DIR, FRUIT_NAME + ".CrMi-dice-expected");
File crmiOutput = new File(TEST_OUTPUT_DIR, FRUIT_NAME + ".CrMi-dice-actual");
crmiOutput.delete();
expectedOutput.delete();
try {
enableExistTrapping();
Main.main(
new String[] {
"allpairs",
"--charset",
"UTF-8",
"--measure",
"DiceMi",
"--input",
TEST_FRUIT_ENTRY_FEATURES.toString(),
"--input-features",
TEST_FRUIT_FEATURES.toString(),
"--input-entries",
TEST_FRUIT_ENTRIES.toString(),
"--output",
expectedOutput.toString()
});
Main.main(
new String[] {
"allpairs",
"--charset",
"UTF-8",
"--measure",
"CrMi",
"--crmi-beta",
"0.00",
"--crmi-gamma",
"1.00",
"--input",
TEST_FRUIT_ENTRY_FEATURES.toString(),
"--input-features",
TEST_FRUIT_FEATURES.toString(),
"--input-entries",
TEST_FRUIT_ENTRIES.toString(),
"--output",
crmiOutput.toString()
});
} finally {
disableExitTrapping();
}
// The ranking should be identical, but the absolute similarity values
// will be different.
WeightedTokenPairSource expected = new WeightedTokenPairSource(expectedOutput, DEFAULT_CHARSET);
WeightedTokenPairSource actual = new WeightedTokenPairSource(crmiOutput, DEFAULT_CHARSET);
while (expected.hasNext() && actual.hasNext()) {
Weighted<TokenPair> e = expected.read();
Weighted<TokenPair> a = actual.read();
assertEquals(e.record().id1(), a.record().id1());
assertEquals(e.record().id2(), a.record().id2());
}
assertEquals(expected.hasNext(), actual.hasNext());
}