def test_extended_sub1(self):
sys.argv = ["prog", "extended", "sub1"]
with self.assertRaises(ExtendedSub1CmdException):
main()
def test_simple(self):
sys.argv = ["prog", "simple"]
with self.assertRaises(SimpleCmdException):
main()
