From 618c064571b7de8969d8af81ef13bf921974e4cf Mon Sep 17 00:00:00 2001 From: coleramos425 Date: Wed, 29 May 2024 18:54:43 +0000 Subject: [PATCH] Remove debug print statements from profile_general Signed-off-by: coleramos425 --- tests/test_profile_general.py | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/tests/test_profile_general.py b/tests/test_profile_general.py index 4b4b7f5cdc..210fa1179a 100644 --- a/tests/test_profile_general.py +++ b/tests/test_profile_general.py @@ -697,7 +697,7 @@ def test_block_SQ(): "SQ_INST_LEVEL_VMEM.csv", "sysinfo.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -727,7 +727,7 @@ def test_block_SQC(): ] if soc == "MI200": expected_csvs.append("roofline.csv") - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -761,7 +761,7 @@ def test_block_TA(): ] if soc == "MI200": expected_csvs.insert(9, "roofline.csv") - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -809,7 +809,7 @@ def test_block_TD(): "sysinfo.csv", "timestamps.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -845,7 +845,7 @@ def test_block_TCP(): ] if soc == "MI200": expected_csvs.insert(11, "roofline.csv") - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -911,7 +911,7 @@ def test_block_TCC(): "sysinfo.csv", "timestamps.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -946,7 +946,7 @@ def test_block_SPI(): ] if soc == "MI200": expected_csvs.insert(10, "roofline.csv") - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -978,7 +978,7 @@ def test_block_CPC(): ] if soc == "MI200": expected_csvs.insert(7, "roofline.csv") - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate(inspect.stack()[0][3], workload_dir, file_dict) @@ -1004,7 +1004,7 @@ def test_block_CPF(): ] if soc == "MI200": expected_csvs.insert(5, "roofline.csv") - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -1091,7 +1091,7 @@ def test_block_SQ_CPC(): "SQ_INST_LEVEL_VMEM.csv", "sysinfo.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -1178,7 +1178,7 @@ def test_block_SQ_TA(): "SQ_INST_LEVEL_VMEM.csv", "sysinfo.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate(inspect.stack()[0][3], workload_dir, file_dict) @@ -1261,7 +1261,7 @@ def test_block_SQ_SPI(): "SQ_INST_LEVEL_VMEM.csv", "sysinfo.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate( @@ -1348,7 +1348,7 @@ def test_block_SQ_SQC_TCP_CPC(): "SQ_INST_LEVEL_VMEM.csv", "sysinfo.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate(inspect.stack()[0][3], workload_dir, file_dict) @@ -1433,7 +1433,7 @@ def test_block_SQ_SPI_TA_TCC_CPF(): "SQ_INST_LEVEL_VMEM.csv", "sysinfo.csv", ] - print("Expected CSVs MI300A", "\n", sorted(list(file_dict.keys()))) + assert sorted(list(file_dict.keys())) == sorted(expected_csvs) validate(