diff options
author | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-01-23 12:17:16 +0100 |
---|---|---|
committer | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-01-23 12:17:16 +0100 |
commit | e4f743881c915ba250b859d01ef78c7b90a96bf5 (patch) | |
tree | 22e1897967212a209751880cd6f9e3bac82bb9d9 /etc/unittest/model.py | |
parent | Check for comments in copilot (diff) | |
download | gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.gz gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.bz2 gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.lz gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.xz gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.zst gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.zip |
Diffstat (limited to 'etc/unittest/model.py')
-rw-r--r-- | etc/unittest/model.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/etc/unittest/model.py b/etc/unittest/model.py new file mode 100644 index 00000000..0d318e7a --- /dev/null +++ b/etc/unittest/model.py @@ -0,0 +1,27 @@ +import unittest +import g4f +from g4f import ChatCompletion +from .mocks import ModelProviderMock + +DEFAULT_MESSAGES = [{'role': 'user', 'content': 'Hello'}] + +test_model = g4f.models.Model( + name = "test/test_model", + base_provider = "", + best_provider = ModelProviderMock +) +g4f.models.ModelUtils.convert["test_model"] = test_model + +class TestPassModel(unittest.TestCase): + + def test_model_instance(self): + response = ChatCompletion.create(test_model, DEFAULT_MESSAGES) + self.assertEqual(test_model.name, response) + + def test_model_name(self): + response = ChatCompletion.create("test_model", DEFAULT_MESSAGES) + self.assertEqual(test_model.name, response) + + def test_model_pass(self): + response = ChatCompletion.create("test/test_model", DEFAULT_MESSAGES, ModelProviderMock) + self.assertEqual(test_model.name, response)
\ No newline at end of file |