def test_render_rename_custom_list_empty_is_no_op(): result = render( make_table(make_column("A", ["x"])), P(custom_list=True, list_string=""), settings=Settings(), ) assert_result_equals( result, ArrowRenderResult(make_table(make_column("A", ["x"]))))
def test_render_dict_disallow_rename_to_null(): result = render( make_table(make_column("A", ["x"])), P(renames={"A": ""}), settings=Settings(), ) assert_result_equals( result, ArrowRenderResult(make_table(make_column("A", ["x"]))))
def test_dict_disallow_rename_to_null(self): table = pd.DataFrame({"A": [1]}) result = render( table, P(renames={"A": ""}), settings=Settings(), 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=""), settings=Settings(), input_columns={"A": Column()}, ) assert_frame_equal(result, pd.DataFrame({"A": ["x"]}))
def test_render_custom_list(): result = render( make_table(make_column("A", ["x"]), make_column("B", [1])), P(custom_list=True, list_string="X\nY"), settings=Settings(), ) assert_result_equals( result, ArrowRenderResult( make_table(make_column("X", ["x"]), make_column("Y", [1]))), )
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"), settings=Settings(), 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"), settings=Settings(), 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}) self.assertEqual(result["errors"], [])
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"}), settings=Settings(), 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}) self.assertEqual(result["errors"], [])
def test_render_custom_list_disallow_rename_to_null(): result = render( make_table(make_column("A", ["a"]), make_column("B", ["b"]), make_column("C", ["c"])), P(custom_list=True, list_string="D\n\nF"), settings=Settings(), ) assert_result_equals( result, ArrowRenderResult( make_table( make_column("D", ["a"]), make_column("B", ["b"]), make_column("F", ["c"]), )), )
def test_render_rename_swap_columns(): result = render( make_table(make_column("A", ["x"]), make_column("B", [1], format="{:d}")), P(custom_list=False, renames={ "A": "B", "B": "A" }), settings=Settings(), ) assert_result_equals( result, ArrowRenderResult( make_table(make_column("B", ["x"]), make_column("A", [1], format="{:d}"))), )
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"), settings=Settings(), input_columns={"A": Column()}, ) self.assertEqual( result, [ i18n_message( "badParam.custom_list.wrongNumberOfNames", {"n_names": 2, "n_columns": 1}, ) ], )
def test_render_rename_custom_list_too_many_columns_is_error(): result = render( make_table(make_column("A", ["x"])), P(custom_list=True, list_string="X,Y"), settings=Settings(), ) assert_result_equals( result, ArrowRenderResult( make_table(), [ RenderError( i18n_message( "badParam.custom_list.wrongNumberOfNames", { "n_names": 2, "n_columns": 1 }, )) ], ), )