word count

kyclark · kyclark · commit 84e498dd58b6 · 2019-07-14T16:50:08.000-06:00
diff --git a/bin/chapters.txt b/bin/chapters.txt
@@ -1,6 +1,7 @@
 article
 jump_the_five
 picnic
+wc
 howler
 apples_and_bananas
 telephone
diff --git a/book.md b/book.md
diff --git a/playful_python.pdf b/playful_python.pdf
diff --git a/wc/Makefile b/wc/Makefile
@@ -0,0 +1,4 @@
+.PHONY: test
+
+test:
+	pytest -xv test.py
diff --git a/wc/README.md b/wc/README.md
@@ -0,0 +1,44 @@
+# Words Count
+
+Write a Python program called `wc.py` that will emulate the venerable `wc` program in Unix that counts the lines, words, and characters in the given file arguments.
+
+````
+$ ./wc.py
+usage: wc.py [-h] FILE [FILE ...]
+wc.py: error: the following arguments are required: FILE
+$ ./wc.py -h
+usage: wc.py [-h] FILE [FILE ...]
+
+Argparse Python script
+
+positional arguments:
+  FILE        Input file(s)
+
+optional arguments:
+  -h, --help  show this help message and exit
+````
+
+
+
+````
+$ ./wc.py ../inputs/scarlet.txt
+    7035   68061  396320 ../inputs/scarlet.txt
+$ ./wc.py ../inputs/*.txt
+    1000    1000    5840 ../inputs/1000.txt
+     100     100     657 ../inputs/1945-boys.txt
+     100     100     684 ../inputs/1945-girls.txt
+     872    7652   45119 ../inputs/const.txt
+    2476    7436   41743 ../inputs/dickinson.txt
+       1       9      45 ../inputs/fox.txt
+      25     278    1476 ../inputs/gettysburg.txt
+      37      91     499 ../inputs/issa.txt
+       9      51     248 ../inputs/nobody.txt
+       1      16      65 ../inputs/now.txt
+       6      71     413 ../inputs/preamble.txt
+    7035   68061  396320 ../inputs/scarlet.txt
+      17     118     661 ../inputs/sonnet-29.txt
+       3       7      45 ../inputs/spiders.txt
+       9      34     192 ../inputs/the-bustle.txt
+   37842   48990  369949 ../inputs/uscities.txt
+     176    1340    8685 ../inputs/usdeclar.txt
+````
diff --git a/wc/solution.py b/wc/solution.py
@@ -0,0 +1,42 @@
+#!/usr/bin/env python3
+"""Emulate wc (word count)"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description='Argparse Python script',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+    parser.add_argument('file',
+                        metavar='FILE',
+                        nargs='+',
+                        type=argparse.FileType('r'),
+                        help='Input file(s)')
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+
+    for fh in args.file:
+        lines, words, chars = 0, 0, 0
+        for line in fh:
+            lines += 1
+            chars += len(line)
+            words += len(line.split())
+
+        print('{:8}{:8}{:8} {}'.format(lines, words, chars, fh.name))
+
+
+# --------------------------------------------------
+if __name__ == '__main__':
+    main()
diff --git a/wc/test.py b/wc/test.py
@@ -0,0 +1,69 @@
+#!/usr/bin/env python3
+"""tests for wc.py"""
+
+import os
+import random
+import re
+import string
+from subprocess import getstatusoutput
+
+prg = './wc.py'
+fox = '../inputs/fox.txt'
+sonnet = '../inputs/sonnet-29.txt'
+
+
+# --------------------------------------------------
+def test_exists():
+    """exists"""
+
+    assert os.path.isfile(prg)
+
+
+# --------------------------------------------------
+def test_usage():
+    """usage"""
+
+    for flag in ['-h', '--help']:
+        rv, out = getstatusoutput('{} {}'.format(prg, flag))
+        assert rv == 0
+        assert re.match("usage", out, re.IGNORECASE)
+
+
+# --------------------------------------------------
+def random_string():
+    """generate a random string"""
+
+    k = random.randint(5, 10)
+    return ''.join(random.choices(string.ascii_letters + string.digits, k=k))
+
+
+# --------------------------------------------------
+def test_bad_file():
+    """bad_file"""
+
+    bad = random_string()
+    rv, out = getstatusoutput('{} {}'.format(prg, bad))
+    assert rv != 0
+    assert re.search(f"No such file or directory: '{bad}'", out)
+
+
+# --------------------------------------------------
+def test_fox():
+    """Test on fox"""
+
+    rv, out = getstatusoutput('{} {}'.format(prg, fox))
+
+    assert rv == 0
+    assert out.rstrip() == '       1       9      45 ../inputs/fox.txt'
+
+
+# --------------------------------------------------
+def test_more():
+    """Test on more than one file"""
+
+    rv, out = getstatusoutput('{} {} {}'.format(prg, fox, sonnet))
+
+    expected = ('       1       9      45 ../inputs/fox.txt\n'
+                '      17     118     661 ../inputs/sonnet-29.txt')
+    assert rv == 0
+    assert out.rstrip() == expected

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +.PHONY: test
++
 +test:
 +	pytest -xv test.py