def test_rename_custom_list_too_many_columns_is_error(self):
table = pd.DataFrame({"A": ["x"]})
result = render(table,
P(custom_list=True, list_string="X,Y"),
input_columns={"A": Column()})
self.assertEqual(
result,
"You supplied 2 column names, but the table has 1 columns.")
def test_custom_list(self):
table = pd.DataFrame({"A": ["x"], "B": [1]})
result = render(
table,
P(custom_list=True, list_string="X\nY"),
input_columns={
"A": Column(),
"B": Column("{:,d}")
},
)
assert_frame_equal(result["dataframe"],
pd.DataFrame({
"X": ["x"],
"Y": [1]
}))
self.assertEqual(result["column_formats"], {"Y": "{:,d}", "X": None})
def test_custom_list_disallow_rename_to_null(self):
table = pd.DataFrame({"A": [1], "B": [2], "C": [3]})
result = render(
table,
P(custom_list=True, list_string="D\n\nF"),
input_columns={
"A": Column(),
"B": Column(),
"C": Column()
},
)
assert_frame_equal(result["dataframe"],
pd.DataFrame({
"D": [1],
"B": [2],
"F": [3]
}))
def test_rename_swap_columns(self):
table = pd.DataFrame({"A": ["x"], "B": [1]})
result = render(
table,
P(custom_list=False, renames={
"A": "B",
"B": "A"
}),
input_columns={
"A": Column(),
"B": Column("{:,d}")
},
)
assert_frame_equal(result["dataframe"],
pd.DataFrame({
"B": ["x"],
"A": [1]
}))
self.assertEqual(result["column_formats"], {"A": "{:,d}", "B": None})
def test_dict_disallow_rename_to_null(self):
table = pd.DataFrame({"A": [1]})
result = render(table,
P(renames={"A": ""}),
input_columns={"A": Column()})
assert_frame_equal(result, pd.DataFrame({"A": [1]}))
def test_rename_custom_list_empty_is_no_op(self):
table = pd.DataFrame({"A": ["x"]})
result = render(table,
P(custom_list=True, list_string=""),
input_columns={"A": Column()})
assert_frame_equal(result, pd.DataFrame({"A": ["x"]}))