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)