diff --git a/tests/python-pytest/onnx/test_models.py b/tests/python-pytest/onnx/test_models.py
index 46080c6daecc..12bc2711190d 100644
--- a/tests/python-pytest/onnx/test_models.py
+++ b/tests/python-pytest/onnx/test_models.py
@@ -153,6 +153,7 @@ def get_model_results(modelpath):
                 logging.info(model_name + " conversion successful")
 
 
+# test_case = ("model name", input shape, output shape)
 test_cases = [
     ("bvlc_googlenet", (1, 3, 224, 224), (1, 1000)),
     ("bvlc_reference_caffenet", (1, 3, 224, 224), (1, 1000)),