Complete Chapter 8 exercises

rchopra · rchopra · commit e9c8047d8f9e · 2023-12-07T21:27:55.000-06:00
diff --git a/Work/fileparse.py b/Work/fileparse.py
@@ -2,6 +2,9 @@
 #
 # Exercise 3.3
 import csv
+import logging
+
+log = logging.getLogger(__name__)
 
 
 def parse_csv(
@@ -10,7 +13,7 @@ def parse_csv(
     types=None,
     has_headers=True,
     delimiter=",",
-    silence_errors=True,
+    silence_errors=False,
 ):
     """Parse a CSV file into a list of records"""
     if select and not has_headers:
@@ -42,8 +45,8 @@ def parse_csv(
                 row = [func(val) for func, val in zip(types, row)]
             except ValueError as e:
                 if not silence_errors:
-                    print(f"Row {rownum}: Couldn't convert {row}")
-                    print(f"Row {rownum}: Reason {e}")
+                    log.warning("Row %d: Couldn't convert %s", rownum, row)
+                    log.debug("Row %d: Reason %s", rownum, e)
                 continue
 
         # Make a dictionary if there are headers, otherwise make a tuple.
diff --git a/Work/report.py b/Work/report.py
@@ -62,6 +62,13 @@ def main(args):
 
 
 if __name__ == "__main__":
+    import logging
     import sys
 
+    logging.basicConfig(
+        filename="app.log",  # Name of the log file (omit to use stderr)
+        filemode="w",  # File mode (use 'a' to append)
+        level=logging.WARNING,  # Logging level (DEBUG, INFO, WARNING, ERROR, or CRITICAL)
+    )
+
     main(sys.argv)
diff --git a/Work/test_stock.py b/Work/test_stock.py
@@ -0,0 +1,31 @@
+# test_stock.py
+
+import unittest
+
+import stock
+
+
+class TestStock(unittest.TestCase):
+    def test_create(self):
+        s = stock.Stock("GOOG", 100, 490.1)
+        self.assertEqual(s.name, "GOOG")
+        self.assertEqual(s.shares, 100)
+        self.assertEqual(s.price, 490.1)
+
+    def test_cost(self):
+        s = stock.Stock("GOOG", 100, 490.1)
+        self.assertEqual(s.cost, 49010.0)
+
+    def test_sell(self):
+        s = stock.Stock("GOOG", 100, 490.1)
+        s.sell(50)
+        self.assertEqual(s.shares, 50)
+
+    def test_shares_is_integer(self):
+        s = stock.Stock("GOOG", 100, 490.1)
+        with self.assertRaises(TypeError):
+            s.shares = "200"
+
+
+if __name__ == "__main__":
+    unittest.main()
