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