diff --git a/projects/rocprofiler-compute/CMakeLists.txt b/projects/rocprofiler-compute/CMakeLists.txt index 2a3512a674..bedea3c56f 100644 --- a/projects/rocprofiler-compute/CMakeLists.txt +++ b/projects/rocprofiler-compute/CMakeLists.txt @@ -383,8 +383,7 @@ add_test( add_test( NAME test_profile_iteration_multiplexing_stochastic COMMAND - ${PYTHON_TEST_COMMAND} -m pytest -s -m - test_profile_iteration_multiplexing_stochastic + ${PYTHON_TEST_COMMAND} -m pytest -s -m iteration_multiplexing_stochastic --junitxml=tests/test_profile_iteration_multiplexing_stochastic.xml ${COV_OPTION} tests/test_profile_general.py ${WORKING_DIR_OPTION} ) diff --git a/projects/rocprofiler-compute/tests/test_utils.py b/projects/rocprofiler-compute/tests/test_utils.py index 3dafe53621..41947aa801 100644 --- a/projects/rocprofiler-compute/tests/test_utils.py +++ b/projects/rocprofiler-compute/tests/test_utils.py @@ -7883,20 +7883,12 @@ def test_merge_counters_iteration_multiplex(): # For "kernel" policy result = utils_mod.merge_counters_iteration_multiplex(df, "kernel") - column_headers = [headers[1] for headers in result.columns.tolist()] - assert isinstance(result, pd.DataFrame) - assert "Mean_Time" in column_headers - assert "Median_Time" in column_headers assert len(result) == 1 # Only one unique kernel_name 'kernel_a' # For "kernel_launch_params" policy result = utils_mod.merge_counters_iteration_multiplex(df, "kernel_launch_params") - column_headers = [headers[1] for headers in result.columns.tolist()] - assert isinstance(result, pd.DataFrame) - assert "Mean_Time" in column_headers - assert "Median_Time" in column_headers assert len(result) == 2 data = { @@ -7921,11 +7913,7 @@ def test_merge_counters_iteration_multiplex(): df.columns = pd.MultiIndex.from_tuples(df.columns) result = utils_mod.merge_counters_iteration_multiplex(df, "kernel_launch_params") - column_headers = [headers[1] for headers in result.columns.tolist()] - assert isinstance(result, pd.DataFrame) - assert "Mean_Time" in column_headers - assert "Median_Time" in column_headers assert len(result) == 3 # Test multi_kernel @@ -7952,20 +7940,12 @@ def test_merge_counters_iteration_multiplex(): # For "kernel" policy result = utils_mod.merge_counters_iteration_multiplex(df, "kernel") - column_headers = [headers[1] for headers in result.columns.tolist()] - assert isinstance(result, pd.DataFrame) - assert "Mean_Time" in column_headers - assert "Median_Time" in column_headers assert len(result) == 2 # For "kernel_launch_params" policy result = utils_mod.merge_counters_iteration_multiplex(df, "kernel_launch_params") - column_headers = [headers[1] for headers in result.columns.tolist()] - assert isinstance(result, pd.DataFrame) - assert "Mean_Time" in column_headers - assert "Median_Time" in column_headers assert len(result) == 3