//[[[Susie:0.0], [Peter:0.0], [Mary:0.0]]]
//Test 1
//Create known texts
EventSet known1 = new EventSet();
EventSet known2 = new EventSet();
EventSet known3 = new EventSet();
EventSet known4 = new EventSet();
known1.addEvent(new Event("mary", null));
known1.addEvent(new Event("had", null));
known1.addEvent(new Event("a", null));
known1.addEvent(new Event("little", null));
known1.addEvent(new Event("lamb", null));
//known1.setAuthor("Mary");
known3.addEvent(new Event("mary", null));
known3.addEvent(new Event("had", null));
known3.addEvent(new Event("a", null));
known3.addEvent(new Event("small", null));
known3.addEvent(new Event("lamb", null));
//known3.setAuthor("Mary");
known2.addEvent(new Event("peter", null));
known2.addEvent(new Event("piper", null));
known2.addEvent(new Event("picked", null));
known2.addEvent(new Event("a", null));
known2.addEvent(new Event("peck", null));
//known2.setAuthor("Peter");
known4.addEvent(new Event("peter", null));
known4.addEvent(new Event("piper", null));
known4.addEvent(new Event("collected", null));
known4.addEvent(new Event("a", null));
known4.addEvent(new Event("peck", null));
//known4.setAuthor("Peter");
List<Document> knowns = new ArrayList<Document>();
Document knownDocument1 = new Document();
knownDocument1.setAuthor("Mary");
knownDocument1.addEventSet(null, known1);
knowns.add(knownDocument1);
Document knownDocument2 = new Document();
knownDocument2.setAuthor("Peter");
knownDocument2.addEventSet(null, known2);
knowns.add(knownDocument2);
Document knownDocument3 = new Document();
knownDocument3.setAuthor("Mary");
knownDocument3.addEventSet(null, known3);
knowns.add(knownDocument3);
Document knownDocument4 = new Document();
knownDocument4.setAuthor("Peter");
knownDocument4.addEventSet(null, known4);
knowns.add(knownDocument4);
//Create unknown text
EventSet unknown1 = new EventSet();
unknown1.addEvent(new Event("mary", null));
unknown1.addEvent(new Event("had", null));
unknown1.addEvent(new Event("a", null));
unknown1.addEvent(new Event("little", null));
unknown1.addEvent(new Event("beta", null));
Document unknownDocument = new Document();
unknownDocument.addEventSet(null, unknown1);
//Classify unknown based on the knowns
WEKALeastMedSq classifier = new WEKALeastMedSq();
List<List<Pair<String, Double>>> t = new ArrayList<List<Pair<String,Double>>>();
classifier.train(knowns);
t.add(classifier.analyze(unknownDocument));
System.out.println(t.toString());
//Assert that the authors match
assertTrue(t.get(0).get(0).getFirst().equals("Mary"));
// Test 1b - Test equal likelihood
EventSet unknown2 = new EventSet();
unknown2.addEvent(new Event("mary", null));
unknown2.addEvent(new Event("had", null));
unknown2.addEvent(new Event("a", null));
unknown2.addEvent(new Event("peter", null));
unknown2.addEvent(new Event("piper", null));
Document unknownDocument2 = new Document();
unknownDocument2.addEventSet(null, unknown2);
t = new ArrayList<List<Pair<String,Double>>>();
classifier.train(knowns);
t.add(classifier.analyze(unknownDocument2));
System.out.println(t.toString());
assertTrue(Math.abs(t.get(0).get(0).getSecond()-0.5)<.0001);
assertTrue(Math.abs(t.get(0).get(1).getSecond()-0.5)<.0001);
classifier = new WEKALeastMedSq();
t = new ArrayList<List<Pair<String,Double>>>();
classifier.train(knowns);
t.add(classifier.analyze(unknownDocument));
System.out.println(t.toString());
assertTrue(t.get(0).get(0).getFirst().equals("Mary"));
//Test 3 - Add in another unknown
EventSet unknown3 = new EventSet();
unknown3.addEvent(new Event("peter", null));
unknown3.addEvent(new Event("piper", null));
unknown3.addEvent(new Event("picked", null));
unknown3.addEvent(new Event("a", null));
unknown3.addEvent(new Event("shells", null));
Document unknownDocument3 = new Document();
classifier = new WEKALeastMedSq();
t = new ArrayList<List<Pair<String,Double>>>();