//Test 1
EventSet known1 = new EventSet();
EventSet known2 = new EventSet();
EventSet unknown = 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.addEvent(new Event("whose", null));
known1.addEvent(new Event("fleece", null));
known1.addEvent(new Event("was", null));
known1.addEvent(new Event("white", null));
known1.addEvent(new Event("as", null));
known1.addEvent(new Event("snow.", null));
//known1.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("pack", null));
known2.addEvent(new Event("of", null));
known2.addEvent(new Event("pickled", null));
known2.addEvent(new Event("peppers.", null));
//known2.setAuthor("Peter");
unknown.addEvent(new Event("Mary", null));
unknown.addEvent(new Event("had", null));
unknown.addEvent(new Event("a", null));
unknown.addEvent(new Event("little", null));
unknown.addEvent(new Event("lambda", null));
unknown.addEvent(new Event("whose", null));
unknown.addEvent(new Event("syntax", null));
unknown.addEvent(new Event("was", null));
unknown.addEvent(new Event("white", null));
unknown.addEvent(new Event("as", null));
unknown.addEvent(new Event("snow.", null));
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 unknownDocument = new Document();
unknownDocument.addEventSet(null, unknown);
LDA classifier = new LDA();
classifier.train(knowns);
List<Pair<String, Double>> t = classifier.analyze(unknownDocument);
String author1 = t.get(0).getFirst();
String author2 = t.get(1).getFirst();
Double val1 = t.get(0).getSecond();
Double val2 = t.get(1).getSecond();
/*System.out.println("Test 1 Classified");
System.out.println("First : "+author1+" "+t.get(0).getSecond());
System.out.println("Second: "+author2+" "+t.get(1).getSecond());
System.out.println("Expected");
System.out.println("First : Mary");
System.out.println("Second: Peter");*/
assertTrue(author1.equals("Mary"));
//Test 2 - Same classifier
//Testing for persistence
t = classifier.analyze(unknownDocument);
/*System.out.println("Test 2 Classified");
System.out.println("First : "+t.get(0).getFirst()+" "+t.get(0).getSecond());
System.out.println("Second: "+t.get(1).getFirst()+" "+t.get(1).getSecond());
System.out.println("Expected");
System.out.println("First : Mary");
System.out.println("Second: Peter");*/
assertTrue(author1.equals(t.get(0).getFirst()) && Math.abs(val1 - t.get(0).getSecond()) < .000001
&& author2.equals(t.get(1).getFirst()) && Math.abs(val2 - t.get(1).getSecond()) < .000001);
//Test 3 - Different instance of classifier
//Again testing for persistence
LDA lda = new LDA();
lda.train(knowns);
t = lda.analyze(unknownDocument);
//String r = t.get(0).getFirst();
/*System.out.println("Test 3 Classified");
System.out.println("First : "+r+" "+t.get(0).getSecond());
System.out.println("Second: "+t.get(1).getFirst()+" "+t.get(1).getSecond());
System.out.println("Expected");
System.out.println("First : Mary");
System.out.println("Second: Peter");*/
assertTrue(author1.equals(t.get(0).getFirst()) && Math.abs(val1 - t.get(0).getSecond()) < .000001
&& author2.equals(t.get(1).getFirst()) && Math.abs(val2 - t.get(1).getSecond()) < .000001);
//Test 4 - two unknowns
EventSet unknown2 = new EventSet();
unknown2.addEvent(new Event("Peter", null));
unknown2.addEvent(new Event("pumpkin", null));
unknown2.addEvent(new Event("picked", null));
unknown2.addEvent(new Event("a", null));
unknown2.addEvent(new Event("pack", null));
unknown2.addEvent(new Event("of", null));
unknown2.addEvent(new Event("pickled", null));
unknown2.addEvent(new Event("potatoes.", null));
Document unknownDocument2 = new Document();
unknownDocument2.addEventSet(null, unknown2);
List<List<Pair<String, Double>>> t2 = new ArrayList<List<Pair<String,Double>>>();