def test_data_stats(self): '''This function sets the values of test_handler.df_5 and test_handler.df_30 to the values self.df_5 and self.df_30 and then calls the data_stats function on the test_handler. Finally assertEqual is called to see if the correct values are calculated.''' test_handler = DataHandler() test_handler.df_5 = self.df_5 test_handler.df_30 = self.df_30 test_handler.data_stats(print_op=False) self.assertEqual(test_handler.df_stats['Mean']['A'], 29.5) self.assertEqual(test_handler.df_stats['Median']['C'], 29.5) self.assertEqual(test_handler.df_stats['Min']['D'], 0) self.assertEqual(test_handler.df_stats['Max']['E'], 9) self.assertEqual(test_handler.df_stats['Count']['F'], 10)