@Test
public void testTfidfVectorizer() {

    TfidfVectorizer vectorizer = new TfidfVectorizer();
    vectorizer.initialize(new Configuration());
    RecordReader reader = new CollectionRecordReader(Writables.writables(Arrays.asList("Testing one.", "Testing 2.")));
    INDArray n = vectorizer.fitTransform(reader);
    //number of vocab words is 3
    assertEquals(3,n.columns());
    //number of records is 2
    assertEquals(2,n.rows());
}