test: add unit tests for addition module in tests/test_addition.py

noclocks · Apr 30, 2024 · 3970584 · 3970584
1 parent fe351c2
commit 3970584
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/test_addition.py b/tests/test_addition.py
@@ -0,0 +1,19 @@
+"""Unit Tests for Addition Module"""
+
+import unittest
+
+from pkg.addition import add_one, add_n 
+
+
+class TestAddition(unittest.TestCase):
+
+    def test_add_one(self):
+      self.assertEqual(add_one(5), 6)
+
+
+    def test_add_n(self):
+        self.assertEqual(add_n(5, 1), 6)
+
+
+if __name__ == '__main__':
+    unittest.main()