def check_final_dataframes(self): # Note: this function is used twice in this file # generate dataframes dfa, dfq, dfm = get_dfs() # check pandas DataFrame class type self.assertTrue(isinstance(dfa, DataFrame)) self.assertTrue(isinstance(dfq, DataFrame)) self.assertTrue(isinstance(dfm, DataFrame)) # check dataframe contents self.assertEqual(dfa.to_csv(), self.REF_DFA) self.assertEqual(dfq.to_csv(), self.REF_DFQ) self.assertEqual(dfm.to_csv(), self.REF_DFM)
def check_final_dataframes(): # Note: this function is used twice in this file # generate dataframes dfa, dfq, dfm = get_dfs() # check pandas DataFrame class type assert isinstance(dfa, DataFrame) assert isinstance(dfq, DataFrame) assert isinstance(dfm, DataFrame) # check dataframe contents assert dfa.to_csv() == REF_DFA assert dfq.to_csv() == REF_DFQ assert dfm.to_csv() == REF_DFM