From d77a16db5af0ac07ac826d05df8dd655c0ec38a7 Mon Sep 17 00:00:00 2001 From: Christophe Bedard Date: Tue, 30 Mar 2021 18:03:07 -0400 Subject: [PATCH] Update test Signed-off-by: Christophe Bedard --- .../tracetools_analysis/test_profile_handler.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/tracetools_analysis/test/tracetools_analysis/test_profile_handler.py b/tracetools_analysis/test/tracetools_analysis/test_profile_handler.py index 1fba862..9c17650 100644 --- a/tracetools_analysis/test/tracetools_analysis/test_profile_handler.py +++ b/tracetools_analysis/test/tracetools_analysis/test_profile_handler.py @@ -286,17 +286,8 @@ class TestProfileHandler(unittest.TestCase): @staticmethod def build_expected_df(expected_data: List[Dict[str, Any]]) -> DataFrame: - # Make sure the columns are in the same order - expected_df = DataFrame(columns=[ - 'tid', - 'depth', - 'function_name', - 'parent_name', - 'start_timestamp', - 'duration', - 'actual_duration', - ]) - return expected_df.append(expected_data, ignore_index=True) + # Columns should be in the same order + return DataFrame.from_dict(expected_data) @staticmethod def transform_fake_fields(events: List[DictEvent]) -> None: