chore: import upstream snapshot with attribution

2026-07-13 12:47:42 +08:00
commit be3ef883e1
1214 changed files with 431743 additions and 0 deletions
@@ -0,0 +1,141 @@
+# Copyright 2025-present the zvec project
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from __future__ import annotations
+
+import pytest
+from zvec import (
+    DataType,
+    IndexType,
+    MetricType,
+    QuantizeType,
+    Status,
+    StatusCode,
+)
+
+
+# ----------------------------
+# Enum Test Case
+# ----------------------------
+@pytest.mark.parametrize(
+    "member, name",
+    [
+        (DataType.FLOAT, "FLOAT"),
+        (IndexType.HNSW, "HNSW"),
+        (MetricType.COSINE, "COSINE"),
+        (QuantizeType.INT8, "INT8"),
+        (StatusCode.OK, "OK"),
+    ],
+)
+def test_enum_names(member, name):
+    assert member.name == name
+
+
+@pytest.mark.parametrize(
+    "member, value",
+    [
+        (DataType.FLOAT, 8),
+        (IndexType.HNSW, 1),
+        (MetricType.COSINE, 3),
+        (QuantizeType.INT8, 2),
+        (StatusCode.OK, 0),
+    ],
+)
+def test_enum_values(member, value):
+    assert member.value == value
+
+
+@pytest.mark.parametrize("member", ["L2", "IP", "COSINE"])
+def test_metric_type_has_member(member):
+    assert member in MetricType.__members__
+
+
+@pytest.mark.parametrize(
+    "member",
+    [
+        "STRING",
+        "BOOL",
+        "INT32",
+        "INT64",
+        "FLOAT",
+        "DOUBLE",
+        "UINT32",
+        "UINT64",
+        "VECTOR_FP16",
+        "VECTOR_FP32",
+        "VECTOR_FP64",
+        "VECTOR_INT8",
+        "SPARSE_VECTOR_FP32",
+        "SPARSE_VECTOR_FP16",
+        "ARRAY_STRING",
+        "ARRAY_INT32",
+        "ARRAY_INT64",
+        "ARRAY_FLOAT",
+        "ARRAY_DOUBLE",
+        "ARRAY_BOOL",
+        "ARRAY_UINT32",
+        "ARRAY_UINT64",
+    ],
+)
+def test_data_type_has_member(member):
+    assert member in DataType.__members__
+
+
+@pytest.mark.parametrize("member", ["HNSW", "IVF", "FLAT", "INVERT"])
+def test_index_type_has_member(member):
+    assert member in IndexType.__members__
+
+
+@pytest.mark.parametrize("member", ["FP16", "INT8", "INT4", "UNDEFINED"])
+def test_quantize_type_has_member(member):
+    assert member in QuantizeType.__members__
+
+
+@pytest.mark.parametrize(
+    "member",
+    [
+        "OK",
+        "UNKNOWN",
+        "NOT_FOUND",
+        "ALREADY_EXISTS",
+        "INVALID_ARGUMENT",
+        "PERMISSION_DENIED",
+        "FAILED_PRECONDITION",
+        "RESOURCE_EXHAUSTED",
+        "UNAVAILABLE",
+        "INTERNAL_ERROR",
+        "NOT_SUPPORTED",
+    ],
+)
+def test_status_code_has_member(member):
+    assert member in StatusCode.__members__
+
+
+# ----------------------------
+# Status Test Case
+# ----------------------------
+class TestStatus:
+    def test_status_code(self):
+        status = Status(StatusCode.OK)
+        assert status.code() == StatusCode.OK
+
+    def test_status_message(self):
+        status = Status(StatusCode.OK, "OK")
+        assert status.message() == "OK"
+
+        status = Status(StatusCode.NOT_FOUND, "Not Found")
+        assert status.message() == "Not Found"
+
+    def test_status_ok(self):
+        status = Status(StatusCode.OK)
+        assert status.ok()