diff --git a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py index 214896e12a60..7a465654d2db 100644 --- a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py +++ b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py @@ -35,7 +35,6 @@ def setUpClass(cls): def tearDownClass(cls): test_config.TestConfig.try_delete_database_with_id(cls.client, cls.test_db.id) - @unittest.skip def test_create_valid_vector_embedding_policy(self): # Using valid data types data_types = ["float32", "float16", "int8", "uint8"] @@ -54,7 +53,6 @@ def test_create_valid_vector_embedding_policy(self): vector_embedding_policy=vector_embedding_policy) properties = created_container.read() assert properties["vectorEmbeddingPolicy"]["vectorEmbeddings"][0]["dataType"] == data_type - self.test_db.delete_container('vector_container_' + data_type) def test_create_vector_embedding_container(self): indexing_policy = { diff --git a/sdk/cosmos/azure-cosmos/tests/test_vector_policy_async.py b/sdk/cosmos/azure-cosmos/tests/test_vector_policy_async.py index 61c623b79270..1963ff2e26d9 100644 --- a/sdk/cosmos/azure-cosmos/tests/test_vector_policy_async.py +++ b/sdk/cosmos/azure-cosmos/tests/test_vector_policy_async.py @@ -46,6 +46,25 @@ async def asyncSetUp(self): async def asyncTearDown(self): await self.client.close() + async def test_create_valid_vector_embedding_policy_async(self): + # Using valid data types + data_types = ["float32", "float16", "int8", "uint8"] + for data_type in data_types: + vector_embedding_policy = { + "vectorEmbeddings": [ + { + "path": "/vector1", + "dataType": data_type, + "dimensions": 256, + "distanceFunction": "euclidean" + }]} + created_container = await self.test_db.create_container( + id='vector_container_' + data_type, + partition_key=PartitionKey(path="/id"), + vector_embedding_policy=vector_embedding_policy) + properties = await created_container.read() + assert properties["vectorEmbeddingPolicy"]["vectorEmbeddings"][0]["dataType"] == data_type + async def test_create_vector_embedding_container_async(self): indexing_policy = { "vectorIndexes": [