Skip to content

Commit 534133d

Browse files
authored
Adding test for checking float16 datatype for vector embedding policy (#43598)
* Adding test for checking float16 datatype for vector embedding policy * Revert unnecessary change * Remove unnecessary print * Revert unnecessary change
1 parent 8b01446 commit 534133d

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

sdk/cosmos/azure-cosmos/tests/test_vector_policy.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,27 @@ def setUpClass(cls):
3535
def tearDownClass(cls):
3636
test_config.TestConfig.try_delete_database_with_id(cls.client, cls.test_db.id)
3737

38+
@unittest.skip
39+
def test_create_valid_vector_embedding_policy(self):
40+
# Using valid data types
41+
data_types = ["float32", "float16", "int8", "uint8"]
42+
for data_type in data_types:
43+
vector_embedding_policy = {
44+
"vectorEmbeddings": [
45+
{
46+
"path": "/vector1",
47+
"dataType": data_type,
48+
"dimensions": 256,
49+
"distanceFunction": "euclidean"
50+
}]}
51+
created_container = self.test_db.create_container(
52+
id='vector_container_' + data_type,
53+
partition_key=PartitionKey(path="/id"),
54+
vector_embedding_policy=vector_embedding_policy)
55+
properties = created_container.read()
56+
assert properties["vectorEmbeddingPolicy"]["vectorEmbeddings"][0]["dataType"] == data_type
57+
self.test_db.delete_container('vector_container_' + data_type)
58+
3859
def test_create_vector_embedding_container(self):
3960
indexing_policy = {
4061
"vectorIndexes": [

0 commit comments

Comments
 (0)