Centralized argument parsing in test files.

tests/basic.py

@@ -2,7 +2,8 @@ from src.server import GestureServer
 from src.window import FullscreenWindow
 from src.trackers.basic import BasicTracker
 
-import tests.parse_arguments
+from tests.parse_arguments import create_parser, parse_args
+parse_args(create_parser())
 
 # Create server
 server = GestureServer()

tests/draw.py

@@ -12,17 +12,19 @@ from src.trackers.transform import TransformationTracker
 from src.trackers.tap import TapTracker
 from src.screen import screen_size
 
-import tests.parse_arguments
+from tests.parse_arguments import create_parser, parse_args
+parser = create_parser()
+parser.add_argument('-f', '--fullscreen', action='store_true', default=False,
+                    help='run in fullscreen')
+args = parse_args(parser)
 
 
 pygame.init()
 
-FULLSCREEN = '-f' in sys.argv[1:]
-
 # Config
 FINGER_RADIUS = 20
 CENTROID_RADIUS = 15
-W, H = screen_size if FULLSCREEN else (640, 480)
+W, H = screen_size if args.fullscreen else (640, 480)
 
 BG_COLOR = 0, 0, 0
 LINE_COLOR = 128, 128, 128
@@ -42,7 +44,7 @@ BEAM_INCREMENT = .8
 MAX_SCALE = 10
 
 # Create canvas GUI
-flags = pygame.FULLSCREEN if FULLSCREEN else 0
+flags = pygame.FULLSCREEN if args.fullscreen else 0
 screen = pygame.display.set_mode((W, H), flags)
 
 

tests/parse_arguments.py

@@ -3,22 +3,29 @@ import logging
 
 from src.logger import Logger
 
+
 # Parse arguments
-parser = argparse.ArgumentParser(description='Basic test program for usage '
+def create_parser():
+    parser = argparse.ArgumentParser(description='Basic test program for usage '
                                                  'of multi-touch API.')
-parser.add_argument('--log', metavar='LOG_LEVEL', default='INFO',
+    parser.add_argument('--log', metavar='LOG_LEVEL', default='INFO',
                         choices=['DEBUG', 'INFO', 'WARNING'],
                         help='set log level (defaults to INFO)')
-parser.add_argument('--logfile', metavar='FILENAME',
+    parser.add_argument('--logfile', metavar='FILENAME',
                         help='filename for the log file (the log is printed to '
                              'stdout by default)')
-args = parser.parse_args()
+    return parser
+
 
+def parse_args(parser):
+    args = parser.parse_args()
 
-# Configure logger
-log_config = {'level': getattr(logging, args.log)}
+    # Configure logger
+    log_config = {'level': getattr(logging, args.log)}
 
-if args.logfile:
+    if args.logfile:
         log_config['filename'] = args.logfile
 
-Logger.configure(**log_config)
+    Logger.configure(**log_config)
+
+    return args

tests/tap.py

@@ -2,7 +2,8 @@ from src.server import GestureServer
 from src.window import FullscreenWindow
 from src.trackers.tap import TapTracker
 
-import tests.parse_arguments
+from tests.parse_arguments import create_parser, parse_args
+parse_args(create_parser())
 
 # Create server
 server = GestureServer()

tests/transform.py

@@ -2,7 +2,8 @@ from src.server import GestureServer
 from src.window import FullscreenWindow
 from src.trackers.transform import TransformationTracker
 
-import tests.parse_arguments
+from tests.parse_arguments import create_parser, parse_args
+parse_args(create_parser())
 
 # Create server
 server = GestureServer()