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()