Пример #1
0
    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)