Merge pull request #2064 from confident-ai/prompt-test-version-number

Prompt test version number

Author: penguine-ip

tests/test_confident/test_prompt.py

@@ -22,6 +22,7 @@ def test_push(self):
 
         prompt.pull()
 
+        assert prompt.version[0] == "0"
         assert prompt._text_template == f"Hello, world! {UUID}"
         assert prompt._messages_template is None
         assert prompt._prompt_version_id is not None
@@ -40,6 +41,7 @@ def test_push_with_interpolation_type(self):
 
         prompt.pull()
 
+        assert prompt.version[0] == "0"
         assert prompt._text_template == f"Hello, world! {UUID}"
         assert prompt._messages_template is None
         assert prompt._prompt_version_id is not None
@@ -62,9 +64,9 @@ def test_push(self):
         ]
 
         prompt.push(messages=messages)
-
         prompt.pull()
 
+        assert prompt.version[0] == "0"
         assert prompt._text_template is None
         assert prompt._messages_template == messages
         assert prompt._prompt_version_id is not None
@@ -88,6 +90,7 @@ def test_push_with_interpolation_type(self):
 
         prompt.pull()
 
+        assert prompt.version[0] == "0"
         assert prompt._text_template is None
         assert prompt._messages_template == messages
         assert prompt._prompt_version_id is not None