Fixed ctests (#2406)
This commit is contained in:
@@ -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}
|
||||
)
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user