mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
from engine.controller.vector_engine import VectorEngine
|
|
from engine.settings import DATABASE_DIRECTORY
|
|
from engine import app
|
|
from flask import jsonify
|
|
import pytest
|
|
import os
|
|
import logging
|
|
import json
|
|
|
|
logging.basicConfig(level = logging.INFO,format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class TestViews:
|
|
|
|
def loads(self, resp):
|
|
return json.loads(resp.data.decode())
|
|
|
|
def test_group(self, test_client):
|
|
data = {"dimension": 10}
|
|
|
|
resp = test_client.get('/vector/group/6')
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 1
|
|
|
|
resp = test_client.post('/vector/group/6', data=json.dumps(data))
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
|
|
resp = test_client.get('/vector/group/6')
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
|
|
# GroupList
|
|
resp = test_client.get('/vector/group')
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
assert self.loads(resp)['group_list'] == [{'file_number': 0, 'group_name': '6'}]
|
|
|
|
resp = test_client.delete('/vector/group/6')
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
|
|
|
|
def test_vector(self, test_client):
|
|
dimension = {"dimension": 10}
|
|
resp = test_client.post('/vector/group/6', data=json.dumps(dimension))
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
|
|
vector = {"vector": [1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8]}
|
|
resp = test_client.post('/vector/add/6', data=json.dumps(vector))
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
|
|
resp = test_client.post('/vector/index/6')
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
|
|
limit = {"vector": [1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8], "limit": 3}
|
|
resp = test_client.get('/vector/search/6', data=json.dumps(limit))
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
assert self.loads(resp)['vector_id'] == 0
|
|
|
|
resp = test_client.delete('/vector/group/6')
|
|
assert resp.status_code == 200
|
|
assert self.loads(resp)['code'] == 0
|
|
|
|
|
|
|
|
|