Update test

Backport of d77a16db5a

Signed-off-by: Christophe Bedard <bedard.christophe@gmail.com>
This commit is contained in:
Christophe Bedard 2021-03-30 18:03:07 -04:00
parent 61a2783a4d
commit ceb6715bae

View file

@ -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: