Adjust test assertion

pinecone-io · Mar 21, 2024 · 4d19d19 · 4d19d19
1 parent 52795ce
commit 4d19d19
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/test_control.py b/tests/unit/test_control.py
@@ -48,8 +48,8 @@ def test_set_source_tag_in_useragent(self):
 
     def test_set_source_tag_in_useragent_via_config(self):
         config = ConfigBuilder.build(api_key='YOUR_API_KEY', host='https://my-host', source_tag='my_source_tag')
-        pc = Pinecone(config=config)
-        assert pc.config.source_tag == 'my_source_tag'
+        p = Pinecone(config=config)
+        assert re.search(r"source_tag=my_source_tag", p.index_api.api_client.user_agent) is not None
 
     @pytest.mark.parametrize("timeout_value, describe_index_responses, expected_describe_index_calls, expected_sleep_calls", [
         # When timeout=None, describe_index is called until ready