def setUp(self) -> None: self.corpus = Corpus(download('subreddit-hey')) self.utt_df = self.corpus.get_utterances_dataframe() self.convo_df = self.corpus.get_conversations_dataframe() self.speaker_df = self.corpus.get_speakers_dataframe() self.new_corpus = Corpus.from_pandas(self.utt_df, self.speaker_df, self.convo_df)
def test_no_speaker_convo_dfs(self): test_corpus = Corpus.from_pandas(utterances_df=self.utt_df) self._test_convo_reconstruction(test_corpus)
def test_no_convos_df(self): test_corpus = Corpus.from_pandas(utterances_df=self.utt_df, speakers_df=self.speaker_df, conversations_df=None) self._test_convo_reconstruction(test_corpus)