def test_sub_menu_notes(self, add_entry_mock):
with patch('builtins.input', return_value='f', side_effect='g'):
work_log.sub_menu()
add_entry_mock.called_once()
def test_sub_menu_quit(self, mock):
with patch('builtins.input', return_value='g'):
work_log.sub_menu()
mock.called_once()