Coverage for pysource_codegen/__main__.py: 0%

12 statements  

« prev     ^ index     » next       coverage.py v7.5.3, created at 2024-06-13 21:17 +0200

1import argparse 

2 

3from ._codegen import generate 

4 

5 

6def run(): 

7 parser = argparse.ArgumentParser() 

8 parser.add_argument("--seed", type=int, help="seed value") 

9 parser.add_argument( 

10 "--node-limit", type=int, default=400, help="limit number of nodes" 

11 ) 

12 parser.add_argument( 

13 "--depth-limit", type=int, default=5, help="limit for the depth of the ast" 

14 ) 

15 parser.add_argument("--root-node", type=str, default="Module", help="root ast type") 

16 args = parser.parse_args() 

17 

18 print( 

19 generate( 

20 args.seed, 

21 node_limit=args.node_limit, 

22 depth_limit=args.depth_limit, 

23 root_node=args.root_node, 

24 ) 

25 ) 

26 

27 

28if __name__ == "__main__": 

29 run()