146 lines
5.0 KiB
Python
146 lines
5.0 KiB
Python
|
from tiramisu import StrOption, BoolOption, IntOption, ChoiceOption, OptionDescription, SymLinkOption
|
||
|
from py.test import raises, fixture
|
||
|
from io import StringIO
|
||
|
import sys
|
||
|
from os import listdir
|
||
|
from os.path import join, isdir
|
||
|
from contextlib import redirect_stdout, redirect_stderr
|
||
|
from argparse import ArgumentParser
|
||
|
|
||
|
#from pouet import TiramisuParser
|
||
|
from tiramisu_parser import TiramisuParser
|
||
|
|
||
|
|
||
|
DATA_DIR = 'test/data/compare'
|
||
|
TEST_DIRS = []
|
||
|
|
||
|
|
||
|
for test in listdir(DATA_DIR):
|
||
|
test_file = join(DATA_DIR, test)
|
||
|
if isdir(test_file):
|
||
|
TEST_DIRS.append(test_file)
|
||
|
|
||
|
TEST_DIRS.sort()
|
||
|
# TEST_DIRS.remove('test/data/compare/10_positional_list')
|
||
|
# TEST_DIRS = ['test/data/compare/50_conditional_disable']
|
||
|
|
||
|
|
||
|
@fixture(scope="module", params=TEST_DIRS)
|
||
|
def test_list(request):
|
||
|
return request.param
|
||
|
|
||
|
|
||
|
def import_subfile_and_test(filename, parser, arg):
|
||
|
parser_dict = []
|
||
|
parser_system_err = []
|
||
|
f = StringIO()
|
||
|
with redirect_stderr(f):
|
||
|
exec(open(filename).read())
|
||
|
# print('arg', arg)
|
||
|
try:
|
||
|
parser_dict.append(parser.parse_args(arg).__dict__)
|
||
|
except SystemExit as err:
|
||
|
parser_system_err.append(str(err))
|
||
|
else:
|
||
|
parser_system_err.append(None)
|
||
|
parser_error = f.getvalue()
|
||
|
f = StringIO()
|
||
|
with redirect_stdout(f):
|
||
|
parser.print_help()
|
||
|
parser_help = f.getvalue()
|
||
|
return parser_dict, parser_system_err, parser_error, parser_help
|
||
|
|
||
|
|
||
|
def test_files(test_list):
|
||
|
args = [[],
|
||
|
# 10_positional
|
||
|
['bar'], ['foo', 'bar'],
|
||
|
# 10_positional_int
|
||
|
['4'],
|
||
|
# 20_bool
|
||
|
['--verbosity'], ['--verbosity', 'arg'],
|
||
|
# 20_string
|
||
|
['--foo'], ['--foo', '--bar'], ['--foo', 'a'],
|
||
|
['--foo', 'a', '--foo', 'b'],
|
||
|
# 20_int
|
||
|
['--int', '3'], ['--int', 'a'],
|
||
|
# 20 choice
|
||
|
['--door', 'a'], ['--door', '1'],
|
||
|
# 30_string_short
|
||
|
['-f', 'b'], ['--foo', 'c', '-f', 'b'],
|
||
|
# 40 multi_bool
|
||
|
['-v'], ['-v', '-s'], ['-vs'],
|
||
|
# 40_short_long
|
||
|
['-v', '--foo', '1'], ['-vf', '2'], ['-vf'], ['-vf', '-v'],
|
||
|
# 40_positional_optional
|
||
|
['bar', '--verbosity'], ['--verbosity', 'bar'],
|
||
|
]
|
||
|
for arg in args:
|
||
|
tiramparser = TiramisuParser('prog.py')
|
||
|
tiramparser_dict, tiramparser_system_err, tiramparser_error, tiramparser_help = import_subfile_and_test(test_list + '/tiramisu.py',
|
||
|
tiramparser, arg)
|
||
|
#
|
||
|
argparser = ArgumentParser('prog.py')
|
||
|
argparser_dict, argparser_system_err, argparser_error, argparser_help = import_subfile_and_test(test_list + '/argparse.py',
|
||
|
argparser, arg)
|
||
|
#print(tiramparser_dict)
|
||
|
#print(tiramparser_system_err)
|
||
|
#print(tiramparser_error)
|
||
|
#print(tiramparser_help)
|
||
|
#print('-----')
|
||
|
#print(argparser_dict)
|
||
|
#print(argparser_system_err)
|
||
|
#print(argparser_error)
|
||
|
#print(argparser_help)
|
||
|
assert tiramparser_dict == argparser_dict
|
||
|
assert tiramparser_error == argparser_error
|
||
|
assert tiramparser_help == argparser_help
|
||
|
assert tiramparser_system_err == argparser_system_err
|
||
|
|
||
|
|
||
|
#FIXME --verbose sans --quiet
|
||
|
#parser = argparse.ArgumentParser(description="calculate X to the power of Y")
|
||
|
#group = parser.add_mutually_exclusive_group()
|
||
|
#group.add_argument("-v", "--verbose", action="store_true")
|
||
|
#group.add_argument("-q", "--quiet", action="store_true")
|
||
|
#parser.add_argument("x", type=int, help="the base")
|
||
|
#parser.add_argument("y", type=int, help="the exponent")
|
||
|
#args = parser.parse_args()
|
||
|
#answer = args.x**args.y
|
||
|
|
||
|
|
||
|
#FIXME --sum ?
|
||
|
#parser = argparse.ArgumentParser(description='Process some integers.')
|
||
|
#parser.add_argument('integers', metavar='N', type=int, nargs='+',
|
||
|
# help='an integer for the accumulator')
|
||
|
#parser.add_argument('--sum', dest='accumulate', action='store_const',
|
||
|
# const=sum, default=max,
|
||
|
# help='sum the integers (default: find the max)')
|
||
|
#args = parser.parse_args()
|
||
|
#print(args.accumulate(args.integers))
|
||
|
|
||
|
|
||
|
# +++++++++++++++++++++++++++++ nargs
|
||
|
#FIXME longueur fixe
|
||
|
#>>> parser = argparse.ArgumentParser()
|
||
|
#>>> parser.add_argument('--foo', nargs=2)
|
||
|
#>>> parser.add_argument('bar', nargs=1)
|
||
|
#>>> parser.parse_args('c --foo a b'.split())
|
||
|
#Namespace(bar=['c'], foo=['a', 'b'])
|
||
|
|
||
|
|
||
|
#FIXME const
|
||
|
#>>> parser = argparse.ArgumentParser()
|
||
|
#>>> parser.add_argument('--foo', nargs='?', const='c', default='d')
|
||
|
#>>> parser.add_argument('bar', nargs='?', default='d')
|
||
|
#>>> parser.parse_args(['XX', '--foo', 'YY'])
|
||
|
#Namespace(bar='XX', foo='YY')
|
||
|
#>>> parser.parse_args(['XX', '--foo'])
|
||
|
#Namespace(bar='XX', foo='c')
|
||
|
#>>> parser.parse_args([])
|
||
|
#Namespace(bar='d', foo='d')
|
||
|
|
||
|
#FIXME ? | * | +
|
||
|
# * == list
|
||
|
# + == list + mandatory
|