Add function to corpus to get number of documents for a test class