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())