test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-11-30 10:59:32 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Merge branch 'linxj' into 'develop'

Browse Source 
Linxj

See merge request jinhai/vecwise_engine!13
This commit is contained in:
xiaojun.lin 2019-03-28 10:25:39 +08:00
commit 77cbac9b98
2 changed files with 51 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyengine/engine/storage/storage_manager.py
View File

@ -2,7 +2,7 @@ import os
import faiss
class StorageManager(object):
def __init__():
def __init__(self):
pass
def put(vector, directory, index_type):

51
pyengine/tests/test_function.py
View File

@ -7,7 +7,6 @@ import json
url = "http://127.0.0.1:5000"
# TODO: LOG and Assert
class TestEngineFunction():
def test_1m_add(self):
d = 4
@ -40,6 +39,56 @@ class TestEngineFunction():
r = requests.get(vector_search_route, json=data)
print(r.json())
def test_restful_interface(self):
d = 4
nb = 100
nq = 1
k = 10
_, xb, xq = get_dataset(d, nb, 1, nq)
groupid_1 = "Group_1"
groupid_2 = "Group_2"
vector_add_route = url + "/vector/add/"
vector_search_route = url + "/vector/search/"
group_route = url + "/vector/group/"
group_list_route = url + "/vector/group"
# Add groupid
r = requests.post(group_route + groupid_1, json={"dimension": d})
print(r.json())
r = requests.post(group_route + groupid_2, json={"dimension": d})
print(r.json())
# Get groupid list
r = requests.get(group_list_route)
print(r.json())
# delete groupid
r = requests.delete(group_route + groupid_2)
print(r.json())
# get groupid
r = requests.get(group_route + groupid_1)
print(r.json())
# add vector
for i in xb:
data = dict()
data['vector'] = i.tolist()
# print(data)
r = requests.post(vector_add_route + groupid_1, json=data)
print(r.json())
# search dataset
data = dict()
for i in xq:
data['vector'] = i.tolist()
data['limit'] = k
# print(data)
r = requests.get(vector_search_route + groupid_1, json=data)
print(r.json())
def get_dataset(d, nb, nt, nq):
d1 = 10 # intrinsic dimension (more or less)