Add some tests to the TextCategorizationDemo