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_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_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_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 render(table, params): result = renamecolumns.render(table, params) return ProcessResult.coerce(result)
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"]}))
def test_rename_empty_is_no_op(self): table = pd.DataFrame({'A': ['x']}) result = render(table, P(custom_list=False, renames={}), input_columns={'A': Column()}) assert_frame_equal(result, pd.DataFrame({'A': ['x']}))