driscollis · driscollis · commit 8eb3191aaa0e · 2020-08-06T15:46:19.000-05:00
add test script
diff --git a/chapter43_packages/module/test_arithmetic.py b/chapter43_packages/module/test_arithmetic.py
@@ -0,0 +1,22 @@
+# test_arithmetic.py
+
+import arithmetic
+import unittest
+
+class TestArithmetic(unittest.TestCase):
+
+    def test_addition(self):
+        self.assertEqual(arithmetic.add(1, 2), 3)
+
+    def test_subtraction(self):
+        self.assertEqual(arithmetic.subtract(2, 1), 1)
+
+    def test_multiplication(self):
+        self.assertEqual(arithmetic.multiply(5, 5), 25)
+
+    def test_division(self):
+        self.assertEqual(arithmetic.divide(8, 2), 4)
+
+
+if __name__ == '__main__':
+    unittest.main()
