def test_create_collection_exception(self, connect):
collection_name = collection_name_factory()
collection_param = {
"fields": [{
"name": "v",
"type": DataType.FLOAT_VECTOR,
"params": {
"dim": 128
}
}],
"segment_row_limit":
10000,
"auto_id":
False
}
mock_grpc_timeout = mock.MagicMock(
side_effect=grpc.FutureTimeoutError())
with mock.patch.object(Uum, 'future', mock_grpc_timeout):
with pytest.raises(grpc.FutureTimeoutError):
connect.create_collection(collection_name, collection_param)
mock_grpc_error = mock.MagicMock(side_effect=MockGrpcError())
with mock.patch.object(Uum, 'future', mock_grpc_error):
with pytest.raises(grpc.RpcError):
connect.create_collection(collection_name, collection_param)
mock_exception = mock.MagicMock(side_effect=Exception("error"))
with mock.patch.object(Uum, 'future', mock_exception):
with pytest.raises(Exception):
connect.create_collection(collection_name, collection_param)
def test_create_collection_segment_row_limit_invalid(self, srl, connect):
collection_name = collection_name_factory()
collection_param = {
"fields": [
{"name": "v", "type": DataType.FLOAT_VECTOR, "params": {"dim": 128}}
],
"segment_row_limit": srl
}
with pytest.raises(BaseError):
connect.create_collection(collection_name, collection_param)
def test_create_collection_normal(self, connect):
collection_name = collection_name_factory()
collection_param = {
"fields": [
{"name": "v", "type": DataType.FLOAT_VECTOR, "params": {"dim": 128}}
]
}
try:
connect.create_collection(collection_name, collection_param)
except Exception as e:
pytest.fail("Unexpected MyError: ".format(str(e)))
finally:
connect.drop_collection(collection_name)
def test_create_collection_scalar_vector(self, sd, vd, connect):
collection_name = collection_name_factory()
collection_param = {
"fields": [
{"name": "A", "type": sd},
{"name": "v", "type": vd, "params": {"dim": 128}}
]
}
try:
connect.create_collection(collection_name, collection_param)
except Exception as e:
pytest.fail("Unexpected MyError: ".format(str(e)))
finally:
connect.drop_collection(collection_name)
def test_create_collection_segment_row_limit_outrange(self, autoid, connect):
collection_name = collection_name_factory()
collection_param = {
"fields": [
{"name": "v", "type": DataType.FLOAT_VECTOR, "params": {"dim": 128}}
],
"segment_row_limit": 10000,
"auto_id": autoid
}
try:
connect.create_collection(collection_name, collection_param)
except Exception as e:
pytest.fail("Unexpected MyError: ".format(str(e)))
finally:
connect.drop_collection(collection_name)