def setUp(self): """Prepare fake corpus with !truth.txt file.""" self.email_dict = create_corpus_dictionary() self.true_class = create_classification_for(self.email_dict.keys()) create_corpus_dir_from_dictionary(self.email_dict) truth_filepath = os.path.join(CORPUS_DIR, TRUTH_FILENAME) save_classification_to_file(self.true_class, fname=truth_filepath) self.tc = TrainingCorpus(CORPUS_DIR)
def setUp(self): """Prepare fake corpus with !truth.txt file.""" self.email_dict = create_corpus_dictionary() self.true_class = create_classification_for(self.email_dict.keys()) create_corpus_dir_from_dictionary(self.email_dict) truth_filepath = os.path.join(CORPUS_DIR, TRUTH_FILENAME) save_classification_to_file(self.true_class, fname=truth_filepath) with replaced_open(): self.tc = TrainingCorpus(CORPUS_DIR)
def create_corpus_without_truth(self): """Create fake directory with text files for testing purposes.""" # Create contents of the fake corpus and store them in a member # variable just for the case it is needed by the add_truth_to_corpus. self.file_dict = create_corpus_dictionary() create_corpus_dir_from_dictionary(self.file_dict, CORPUS_DIR)