def test_FmtMultiplyCellValue_cell_values(): # Check that factor is applied to numerical value factor = 1e6 fmt = pbtf.FmtMultiplyCellValue(1 / factor, '', rows=['a']) data = FormatterData(factor, 'a', 'aa', df) res = fmt._modify_cell_content(data) assert res == 1. # No change to string value data = FormatterData(TEST_STRING, 'a', 'aa', df) res = fmt._modify_cell_content(data) assert res == TEST_STRING
def test_FmtMultiplyCellValue_header_modification(): factor = 1. suffix = 'suffix' # All columns in header are modified fmt = pbtf.FmtMultiplyCellValue(factor, suffix, rows=['a']) data = FormatterData(TEST_STRING, pbtf.HEADER_ROW_NAME, 'aa', df) res = fmt._modify_cell_content(data) assert res == TEST_STRING + suffix # Only specified columns in header are modified fmt = pbtf.FmtMultiplyCellValue(factor, suffix, rows=['a'], columns=['bb']) data = FormatterData(TEST_STRING, pbtf.HEADER_ROW_NAME, 'bb', df) res = fmt._modify_cell_content(data) assert res == TEST_STRING + suffix data = FormatterData(TEST_STRING, pbtf.HEADER_ROW_NAME, 'aa', df) res = fmt._modify_cell_content(data) assert res == TEST_STRING # No modification is column name is not string fmt = pbtf.FmtMultiplyCellValue(factor, suffix, rows=['a'], columns=['bb']) data = FormatterData(0, pbtf.HEADER_ROW_NAME, 'bb', df) res = fmt._modify_cell_content(data) assert res == 0