mirror of
https://github.com/mitb-archive/filebot
synced 2024-08-13 17:03:45 -04:00
68 lines
2.4 KiB
Java
68 lines
2.4 KiB
Java
package net.filebot.similarity;
|
|
|
|
import static org.junit.Assert.*;
|
|
|
|
import java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import org.junit.Test;
|
|
|
|
import net.filebot.web.Episode;
|
|
import net.filebot.web.SimpleDate;
|
|
|
|
public class EpisodeMetricsTest {
|
|
|
|
public EpisodeMetrics metrics = new EpisodeMetrics();
|
|
|
|
@Test
|
|
public void substringMetrics() {
|
|
Episode eY1T1 = new Episode("Doctor Who", 1, 1, "Rose");
|
|
|
|
File fY1T1 = new File("Doctor Who (2005)/Doctor Who - 1x01 - Rose");
|
|
File fY2T2 = new File("Doctor Who (1963)/Doctor Who - 1x01 - An Unearthly Child");
|
|
|
|
assertEquals(0.5, metrics.SubstringFields.getSimilarity(eY1T1, fY1T1), 0.1);
|
|
assertEquals(0.5, metrics.SubstringFields.getSimilarity(eY1T1, fY2T2), 0.1);
|
|
}
|
|
|
|
@Test
|
|
public void matcherLevel2() throws Exception {
|
|
List<File> files = new ArrayList<File>();
|
|
List<Episode> episodes = new ArrayList<Episode>();
|
|
|
|
files.add(new File("Greek/Greek - S01E19 - No Campus for Old Rules"));
|
|
files.add(new File("Veronica Mars - Season 1/Veronica Mars [1x19] Hot Dogs"));
|
|
episodes.add(new Episode("Veronica Mars", 1, 19, "Hot Dogs"));
|
|
episodes.add(new Episode("Greek", 1, 19, "No Campus for Old Rules"));
|
|
|
|
List<Match<File, Episode>> m = new Matcher<File, Episode>(files, episodes, false, new SimilarityMetric[] { metrics.EpisodeIdentifier, metrics.SubstringFields }).match();
|
|
|
|
assertEquals("Greek - S01E19 - No Campus for Old Rules", m.get(0).getValue().getName());
|
|
assertEquals("Greek - 1x19 - No Campus for Old Rules", m.get(0).getCandidate().toString());
|
|
assertEquals("Veronica Mars [1x19] Hot Dogs", m.get(1).getValue().getName());
|
|
assertEquals("Veronica Mars - 1x19 - Hot Dogs", m.get(1).getCandidate().toString());
|
|
}
|
|
|
|
@Test
|
|
public void nameIgnoreEmbeddedChecksum() {
|
|
assertEquals(1, metrics.Name.getSimilarity("test", "test [EF62DF13]"), 0);
|
|
}
|
|
|
|
@Test
|
|
public void numericIgnoreEmbeddedChecksum() {
|
|
assertEquals(1, metrics.Numeric.getSimilarity("S01E02", "Season 1, Episode 2 [00A01E02]"), 0);
|
|
}
|
|
|
|
@Test
|
|
public void numericNumbers() {
|
|
String fn = "SEED - 01 - [X 2.0]";
|
|
Episode e1 = new Episode("SEED", null, 1, "Enraged Eyes", 1, null, new SimpleDate(2004, 10, 9), null, null);
|
|
Episode s1 = new Episode("SEED", null, null, "EDITED", null, 1, new SimpleDate(2005, 1, 29), null, null);
|
|
|
|
assertEquals(0.5, metrics.Numeric.getSimilarity(fn, e1), 0.01);
|
|
assertEquals(0.5, metrics.Numeric.getSimilarity(fn, s1), 0.01);
|
|
}
|
|
|
|
}
|