test case format for test_models

apache · Dec 21, 2018 · 98adc8a · 98adc8a
1 parent 9222695
commit 98adc8a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)),