From ccfdf0f92043b1a28b871335c50caf6399e99982 Mon Sep 17 00:00:00 2001 From: kotfu Date: Sun, 6 May 2018 16:03:32 -0600 Subject: =?UTF-8?q?Extract=20AddSubmenu()=20into=20it=E2=80=99s=20own=20mo?= =?UTF-8?q?dule?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_completion.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tests/test_completion.py') diff --git a/tests/test_completion.py b/tests/test_completion.py index bda5bb8a..2d1ee2b7 100644 --- a/tests/test_completion.py +++ b/tests/test_completion.py @@ -12,8 +12,9 @@ import argparse import os import sys -from cmd2 import cmd2 import pytest +from cmd2 import cmd2 +import cmd2.submenu from .conftest import complete_tester, StdOut from examples.subcommands import SubcommandsExample @@ -973,7 +974,7 @@ class SecondLevel(cmd2.Cmd): second_level_cmd = SecondLevel() -@cmd2.AddSubmenu(second_level_cmd, +@cmd2.submenu.AddSubmenu(second_level_cmd, command='second', require_predefined_shares=False) class SubmenuApp(cmd2.Cmd): -- cgit v1.2.1