def test_replace_null_median(self):
'''This function adds np.nan values to copies of self.df_5 and self.df_30
and then sets the values of df_5 and df_30 in test_handler to the new
copies of df_5 and df_30. Then the replace_null(method='median')
function is called on test_handler whcih should replace the np.nan values
with median values. Finally assertEqual is called to check that the
replace_null function is working correctly.'''
df_5_replace_null_check = self.df_5.copy()
df_5_replace_null_check.loc[self.df_5.first_valid_index(),
'C'] = np.nan
df_30_replace_null_check = self.df_30.copy()
df_30_replace_null_check.loc[self.df_30.last_valid_index(),
'D'] = np.nan
test_handler = DataHandler()
test_handler.df_5 = df_5_replace_null_check
test_handler.df_30 = df_30_replace_null_check
test_handler.replace_null(method='median')
self.assertEqual(
test_handler.df_5.loc[test_handler.df_5.first_valid_index(), 'C'],
30.0)
self.assertEqual(
test_handler.df_30.loc[test_handler.df_30.last_valid_index(), 'D'],
4.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)
