From aad3e849e5db42b30e474c8f886cc34e6d58a341 Mon Sep 17 00:00:00 2001
From: Vandana Kannan <vandana268@gmail.com>
Date: Wed, 26 Dec 2018 13:36:02 -0800
Subject: [PATCH] Add test for TopK

---
 tests/python-pytest/onnx/test_cases.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/python-pytest/onnx/test_cases.py b/tests/python-pytest/onnx/test_cases.py
index 6a189b62492d..24a3966ef00e 100644
--- a/tests/python-pytest/onnx/test_cases.py
+++ b/tests/python-pytest/onnx/test_cases.py
@@ -76,7 +76,8 @@
              'test_selu_default',
              'test_elu',
              'test_max_',
-             'test_softplus'
+             'test_softplus',
+             'test_top_k'
              ],
     'import': ['test_gather',
                'test_global_lppooling',