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)