intsranges.py 文件源码

def test1():
        """
        IntsRanges micro-test
        """
        def new_parser():
            parser = argparse.ArgumentParser(
                formatter_class=argparse.ArgumentDefaultsHelpFormatter)
            parser.add_argument("-r", "--ranges", default=[1, 3, 5],
                                action=IntsRanges,
                                help="specify inclusive integer ranges")
            return parser

        a = new_parser().parse_args([])
        assert a.ranges == [1, 3, 5]
        a = new_parser().parse_args(['-r', '1'])
        assert a.ranges == [1]
        a = new_parser().parse_args(['-r', '1', '-r', '3-5'])
        assert a.ranges == [1, 3, 4, 5]
        a = new_parser().parse_args(['-r', '1-5-2', '-r', '5-7'])
        assert a.ranges == [1, 3, 5, 6, 7]
        a = new_parser().parse_args(['-r', '0-100-5'])
        assert a.ranges == list(range(0, 101, 5))