github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-10 13:37:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix linting

Browse Source
This commit is contained in:
LeonOstrez
2024-01-29 16:56:47 -08:00
parent 77962ec651
commit 1b27ea2460
2 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pilot/helpers/agents/test_Developer.py
View File

@@ -119,7 +119,6 @@ class TestDeveloper:
mock_execute_command):
# mock_ask_user, mock_get_saved_command_run):
# Given
monkey = None
convo = AgentConvo(self.developer)
convo.save_branch = lambda branch_name=None: branch_name
@@ -137,12 +136,11 @@ class TestDeveloper:
@patch('helpers.Project.ask_user', return_value='continue')
def test_code_changes_manual_test_continue(self, mock_save, mock_chat_completion, mock_ask_user):
# Given
monkey = None
convo = AgentConvo(self.developer)
convo.save_branch = lambda branch_name=None: branch_name
# When
result = self.developer.test_code_changes(monkey, convo)
result = self.developer.test_code_changes(convo)
# Then
assert result == {'success': True}
@@ -154,7 +152,6 @@ class TestDeveloper:
# https://github.com/Pythagora-io/gpt-pilot/issues/35
def test_code_changes_manual_test_no(self, mock_get_saved_user_input, mock_chat_completion, mock_save):
# Given
monkey = None
convo = AgentConvo(self.developer)
convo.save_branch = lambda branch_name=None: branch_name
convo.load_branch = lambda function_uuid=None: function_uuid
@@ -170,7 +167,7 @@ class TestDeveloper:
with patch('utils.questionary.questionary', mock_questionary):
# When
result = self.developer.test_code_changes(monkey, convo)
result = self.developer.test_code_changes(convo)
# Then
assert result == {'success': True, 'user_input': 'no'}
@@ -184,7 +181,6 @@ class TestDeveloper:
mock_execute,
monkeypatch):
# Given
monkey = None
convo = AgentConvo(self.developer)
convo.save_branch = lambda branch_name=None: branch_name
convo.load_branch = lambda function_uuid=None: function_uuid
@@ -222,7 +218,7 @@ class TestDeveloper:
# with patch('utils.questionary.questionary', mock_questionary):
# When
result = self.developer.test_code_changes(monkey, convo)
result = self.developer.test_code_changes(convo)
# Then
assert result == {'success': True}

2
pilot/main.py
View File

@@ -115,7 +115,7 @@ if __name__ == "__main__":
telemetry.send()
run_exit_fn = False
except GracefulExit as err:
except GracefulExit:
# can't call project.finish_loading() here because project can be None
run_exit_fn = False
print('', type='loadingFinished')