12 lines
1.0 KiB
Python
12 lines
1.0 KiB
Python
from pathlib import Path
|
|
root=Path('C:/kefu/wechat_rpa')
|
|
p=root/'test_visual_review_assistant.py'
|
|
text=p.read_text(encoding='utf-8')
|
|
text=text.replace(' self.assistant.resumed.return_value = True', ' self.assistant.resumed.return_value = True\n self.assistant.has_pending.return_value = False')
|
|
text=text.replace(' self.bot._reply_needs_review = mock.Mock(return_value=(True, \'new high risk\'))', ' self.bot._reply_needs_review = mock.Mock(return_value=(True, \'new high risk\'))')
|
|
text=text.replace("self.assertEqual(self.bot._reply_review_decision(self.state, 'q', 'a'), (True, False, ''))", "self.assertEqual(self.bot._reply_review_decision(self.state, 'q', 'a'), (True, True, 'new high risk'))")
|
|
extra=(Path(__file__).parent/'extra_visual_tests.txt').read_text(encoding='utf-8')
|
|
text=text.replace("\nif __name__ == '__main__':", '\n'+extra+"\nif __name__ == '__main__':")
|
|
for target in (root, Path('C:/wechat_rpa')):
|
|
(target/'test_visual_review_assistant.py').write_text(text,encoding='utf-8',newline='\n')
|