def test_backward_fill(self):
_dataset = FillMissingValues(self.dataset).backward_fill()
self.assertTrue(
pd.DataFrame({
'a': [1.0, 2.0, 2.0, 3.0, 4.0],
'b': [3.0, 6.0, 8.0, 9.0, 9.0]
}).equals(_dataset))
def test_median(self):
_dataset = FillMissingValues(self.dataset).median()
self.assertTrue(
pd.DataFrame({
'a': [1.0, 2.5, 2.0, 3.0, 4.0],
'b': [3.0, 6.0, 8.0, 7.0, 9.0]
}).equals(_dataset))
def test_backward_fill_nan(self):
_dataset = FillMissingValues(self.dataset).backward_fill_nan()
self.assertTrue(
pd.DataFrame({
'a': [1.0, 2.0, 2.0, 3.0, 4.0],
'b': [3, 6, 8, '', 9]
}).equals(_dataset))
def test_default(self):
_dataset = FillMissingValues(self.dataset).default(6)
self.assertTrue(
pd.DataFrame({
'a': [1.0, 6.0, 2.0, 3.0, 4.0],
'b': [3.0, 6.0, 8.0, 6.0, 9.0]
}).equals(_dataset))