def test_apply_postprocessing_2(self):
"""
Unit test apply_postprocessing 2
"""
xpl = SmartExplainer()
xpl.x_pred = pd.DataFrame([[1, 2], [3, 4]],
columns=['Col1', 'Col2'],
index=['Id1', 'Id2'])
xpl.features_dict = {'Col1': 'Column1', 'Col2': 'Column2'}
xpl.columns_dict = {0: 'Col1', 1: 'Col2'}
xpl.inv_features_dict = {'Column1': 'Col1', 'Column2': 'Col2'}
postprocessing = {
'Col1': {
'type': 'suffix',
'rule': ' t'
},
'Col2': {
'type': 'prefix',
'rule': 'test'
}
}
expected_output = pd.DataFrame(data=[['1 t', 'test2'],
['3 t', 'test4']],
columns=['Col1', 'Col2'],
index=['Id1', 'Id2'])
output = xpl.apply_postprocessing(postprocessing)
assert np.array_equal(output, expected_output)
def test_apply_postprocessing_1(self):
"""
Unit test apply_postprocessing 1
"""
xpl = SmartExplainer()
xpl.x_pred = pd.DataFrame([[1, 2], [3, 4]],
columns=['Col1', 'Col2'],
index=['Id1', 'Id2'])
xpl.features_dict = {'Col1': 'Column1', 'Col2': 'Column2'}
xpl.columns_dict = {0: 'Col1', 1: 'Col2'}
xpl.inv_features_dict = {'Column1': 'Col1', 'Column2': 'Col2'}
assert np.array_equal(xpl.x_pred, xpl.apply_postprocessing())
