From 0101f1ede56c6e84d593d5e4d291c38694e91509 Mon Sep 17 00:00:00 2001
From: zhenwu <zw@zilliz.com>
Date: Sat, 19 Oct 2019 19:34:46 +0800
Subject: [PATCH] update java-tests and classified python-tests by opensource
 and internal case #50

Former-commit-id: 944de289a6fca6d41ee8cd3f4fdc85e4cae52e3d
---
 tests/milvus_python_test/test_delete_vectors.py | 4 ++--
 tests/milvus_python_test/test_index.py          | 4 ++--
 tests/milvus_python_test/test_search_vectors.py | 2 +-
 tests/milvus_python_test/test_table.py          | 7 +++++--
 tests/milvus_python_test/test_table_count.py    | 4 ++--
 5 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/tests/milvus_python_test/test_delete_vectors.py b/tests/milvus_python_test/test_delete_vectors.py
index 70cf086a56..4e1f7d101d 100644
--- a/tests/milvus_python_test/test_delete_vectors.py
+++ b/tests/milvus_python_test/test_delete_vectors.py
@@ -109,7 +109,7 @@ class TestDeleteVectorsBase:
         params=gen_index_params()
     )
     def get_index_params(self, request, args):
-        if "internal" in args:
+        if "internal" not in args:
             if request.param["index_type"] == IndexType.IVF_SQ8H:
                 pytest.skip("sq8h not support in open source")
         return request.param
@@ -295,7 +295,7 @@ class TestDeleteVectorsIP:
         params=gen_index_params()
     )
     def get_index_params(self, request, args):
-        if "internal" in args:
+        if "internal" not in args:
             if request.param["index_type"] == IndexType.IVF_SQ8H:
                 pytest.skip("sq8h not support in open source")
         return request.param
diff --git a/tests/milvus_python_test/test_index.py b/tests/milvus_python_test/test_index.py
index 255e46b2a7..435a547855 100644
--- a/tests/milvus_python_test/test_index.py
+++ b/tests/milvus_python_test/test_index.py
@@ -27,7 +27,7 @@ class TestIndexBase:
         params=gen_index_params()
     )
     def get_index_params(self, request, args):
-        if "internal" in args:
+        if "internal" not in args:
             if request.param["index_type"] == IndexType.IVF_SQ8H:
                 pytest.skip("sq8h not support in open source")
         return request.param
@@ -505,7 +505,7 @@ class TestIndexIP:
         params=gen_index_params()
     )
     def get_index_params(self, request, args):
-        if "internal" in args:
+        if "internal" not in args:
             if request.param["index_type"] == IndexType.IVF_SQ8H:
                 pytest.skip("sq8h not support in open source")
         return request.param
diff --git a/tests/milvus_python_test/test_search_vectors.py b/tests/milvus_python_test/test_search_vectors.py
index b9257f9ce6..e52e0d2d08 100644
--- a/tests/milvus_python_test/test_search_vectors.py
+++ b/tests/milvus_python_test/test_search_vectors.py
@@ -44,7 +44,7 @@ class TestSearchBase:
         params=gen_index_params()
     )
     def get_index_params(self, request, args):
-        if "internal" in args:
+        if "internal" not in args:
             if request.param["index_type"] == IndexType.IVF_SQ8H:
                 pytest.skip("sq8h not support in open source")
         return request.param
diff --git a/tests/milvus_python_test/test_table.py b/tests/milvus_python_test/test_table.py
index c92afb023e..7088f923ff 100644
--- a/tests/milvus_python_test/test_table.py
+++ b/tests/milvus_python_test/test_table.py
@@ -590,8 +590,11 @@ class TestTable:
         scope="function",
         params=gen_index_params()
     )
-    def get_index_params(self, request):
-        yield request.param
+    def get_index_params(self, request, args):
+        if "internal" not in args:
+            if request.param["index_type"] == IndexType.IVF_SQ8H:
+                pytest.skip("sq8h not support in open source")
+        return request.param
 
     @pytest.mark.level(1)
     def test_preload_table(self, connect, table, get_index_params):
diff --git a/tests/milvus_python_test/test_table_count.py b/tests/milvus_python_test/test_table_count.py
index 6d7b56df0c..820fb9d546 100644
--- a/tests/milvus_python_test/test_table_count.py
+++ b/tests/milvus_python_test/test_table_count.py
@@ -39,7 +39,7 @@ class TestTableCount:
         params=gen_index_params()
     )
     def get_index_params(self, request, args):
-        if "internal" in args:
+        if "internal" not in args:
             if request.param["index_type"] == IndexType.IVF_SQ8H:
                 pytest.skip("sq8h not support in open source")
         return request.param
@@ -183,7 +183,7 @@ class TestTableCountIP:
         params=gen_index_params()
     )
     def get_index_params(self, request, args):
-        if "internal" in args:
+        if "internal" not in args:
             if request.param["index_type"] == IndexType.IVF_SQ8H:
                 pytest.skip("sq8h not support in open source")
         return request.param