@@ -8,6 +8,7 @@
 from subprocess import getstatusoutput
 
 prg = './wod.py'
+tab = 'inputs/bad-delimiter.tab'
 input1 = 'inputs/exercises.csv'
 input2 = 'inputs/silly-exercises.csv'
 
@@ -49,6 +50,41 @@ def test_bad_file():
     assert re.search(f"No such file or directory: '{bad}'", out)
 
 
+# --------------------------------------------------
+def test_tab_delimiter():
+    """tab delimiter"""
+
+    expected = """
+Exercise      Reps
+----------  ------
+Pushups         56
+Situps          88
+Crunches        27
+Burpees         35
+"""
+    tab_flag = '-t' if random.choice([0, 1]) else '--tab'
+    rv, out = getstatusoutput(f'{prg} -f {tab} -s 1 {tab_flag}')
+    assert rv == 0
+    assert out.strip() == expected.strip()
+
+
+# --------------------------------------------------
+def test_tab_delimiter_extension():
+    """tab delimiter based on file extension"""
+
+    expected = """
+Exercise      Reps
+----------  ------
+Pushups         56
+Situps          88
+Crunches        27
+Burpees         35
+"""
+    rv, out = getstatusoutput(f'{prg} -f {tab} -s 1')
+    assert rv == 0
+    assert out.strip() == expected.strip()
+
+
 # --------------------------------------------------
 def test_seed1():
     """Runs OK"""
 
@@ -6,6 +6,7 @@
 """
 
 import sys
+import pathlib
 import argparse
 import csv
 import random
@@ -44,17 +45,27 @@ def get_args():
                         default=False,
                         action='store_true',
                         help='Halve the reps')
+    
+    parser.add_argument('-t',
+                        '--tab',
+                        default=False,
+                        action='store_true',
+                        help='Tab delimiter in exercises')
 
     args = parser.parse_args()
     if(args.num <= 0):
         parser.error(f'--num "{args.num}" must be greater than 0')
+    args.delimiter = '\t' if args.tab else ','
+    p = pathlib.Path(args.file.name)
+    if (p.suffix == '.tab'):
+        args.delimiter = '\t'
     return args
 
 
-def read_csv(fh):
+def read_csv(fh, delimiter=','):
     """Read the CSV input"""
     exercises = []
-    for rec in csv.DictReader(fh, delimiter=','):
+    for rec in csv.DictReader(fh, delimiter=delimiter):
         if 'exercise' not in rec or 'reps' not in rec:
             continue
         name, reps = rec['exercise'], rec['reps']
@@ -74,7 +85,7 @@ def main():
 
     args = get_args()
     random.seed(args.seed)
-    exercises = read_csv(args.file)
+    exercises = read_csv(args.file, args.delimiter)
 
     if not exercises:
         sys.exit(f'No usable data in --file "{args.file.name}"')