ok/ChatMastermind
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changes from https://github.com/juk0de/ChatMastermind.git #1

Closed
ok wants to merge 176 commits from main into juk
pull from: main
merge into: ok:juk
Conversation 0 Commits 176 Files Changed 31 +9 -3
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 820d938060 - Show all commits

12
tests/test_main.py
View File

@ -21,9 +21,11 @@ class TestCreateChat(unittest.TestCase):
self.tags = ['test_tag'] self.tags = ['test_tag']
@patch('os.listdir') @patch('os.listdir')
@patch('pathlib.Path.iterdir')
@patch('builtins.open') @patch('builtins.open')
def test_create_chat_with_tags(self, open_mock, listdir_mock): def test_create_chat_with_tags(self, open_mock, iterdir_mock, listdir_mock):
listdir_mock.return_value = ['testfile.txt'] listdir_mock.return_value = ['testfile.txt']
iterdir_mock.return_value = [pathlib.Path(x) for x in listdir_mock.return_value]
open_mock.return_value.__enter__.return_value = io.StringIO(dump_data( open_mock.return_value.__enter__.return_value = io.StringIO(dump_data(
{'question': 'test_content', 'answer': 'some answer', {'question': 'test_content', 'answer': 'some answer',
'tags': ['test_tag']})) 'tags': ['test_tag']}))
@ -41,9 +43,11 @@ class TestCreateChat(unittest.TestCase):
{'role': 'user', 'content': self.question}) {'role': 'user', 'content': self.question})
@patch('os.listdir') @patch('os.listdir')
@patch('pathlib.Path.iterdir')
@patch('builtins.open') @patch('builtins.open')
def test_create_chat_with_other_tags(self, open_mock, listdir_mock): def test_create_chat_with_other_tags(self, open_mock, iterdir_mock, listdir_mock):
listdir_mock.return_value = ['testfile.txt'] listdir_mock.return_value = ['testfile.txt']
iterdir_mock.return_value = [pathlib.Path(x) for x in listdir_mock.return_value]
open_mock.return_value.__enter__.return_value = io.StringIO(dump_data( open_mock.return_value.__enter__.return_value = io.StringIO(dump_data(
{'question': 'test_content', 'answer': 'some answer', {'question': 'test_content', 'answer': 'some answer',
'tags': ['other_tag']})) 'tags': ['other_tag']}))
@ -57,9 +61,11 @@ class TestCreateChat(unittest.TestCase):
{'role': 'user', 'content': self.question}) {'role': 'user', 'content': self.question})
@patch('os.listdir') @patch('os.listdir')
@patch('pathlib.Path.iterdir')
@patch('builtins.open') @patch('builtins.open')
def test_create_chat_without_tags(self, open_mock, listdir_mock): def test_create_chat_without_tags(self, open_mock, iterdir_mock, listdir_mock):
listdir_mock.return_value = ['testfile.txt', 'testfile2.txt'] listdir_mock.return_value = ['testfile.txt', 'testfile2.txt']
iterdir_mock.return_value = [pathlib.Path(x) for x in listdir_mock.return_value]
open_mock.side_effect = ( open_mock.side_effect = (
io.StringIO(dump_data({'question': 'test_content', io.StringIO(dump_data({'question': 'test_content',
'answer': 'some answer', 'answer': 'some answer',