WIP Python GPT4All tests

gpt4all-bindings/python/gpt4all/tests/test_gpt4all.py

@@ -5,6 +5,7 @@
 from typing import Iterable
 
 from gpt4all import GPT4All, Embed4All, pyllmodel
+from gpt4all.gpt4all import empty_chat_session, append_bin_suffix_if_missing
 from . import get_model_folder
 import time
 import pytest
@@ -441,3 +442,24 @@ def test_generate_streaming_with_session(self, model):
         print(response2)
         assert ''.join(response1) == expected1
         assert ''.join(response2) == expected2
+
+
+
+@pytest.mark.parametrize('system_prompt', ['', 'System: Hi.', '### System: Be helpful, damnit.'])
+def test_empty_chat_session(system_prompt):
+    chat_session = empty_chat_session(system_prompt)
+    assert isinstance(chat_session, list)
+    len(chat_session) == 1
+    assert all(k in ('role', 'content') for k in chat_session[0].keys())
+    assert all(isinstance(v, str) for v in chat_session[0].values())
+    assert chat_session[0]['content'] == system_prompt
+
+
+@pytest.mark.parametrize('model_name', [None, '', 'mock-model.bin', 'orca-mini-3b.ggmlv3.q4_0'])
+def test_append_bin_suffix(model_name):
+    if not hasattr(model_name, 'endswith'):  # duck typing test
+        with pytest.raises(Exception):
+            append_bin_suffix_if_missing(model_name)
+    else:
+        new_model_name = append_bin_suffix_if_missing(model_name)
+        assert new_model_name[-4:] == '.bin'