Add test case for subtract.

michaelkawaicheung · Aug 16, 2024 · 12fbd78 · 12fbd78
1 parent b6ad200
commit 12fbd78
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_calculator.py b/tests/test_calculator.py
@@ -1,10 +1,13 @@
 import unittest
-from calculator.calculator import add
+from calculator.calculator import add, subtract
 
 class TestCalculator(unittest.TestCase):
 
     def test_add(self):
         self.assertEqual(add(1, 2), 3)
 
+    def test_subtract(self):
+        self.assertEqual(subtract(5, 3), 2)
+
 if __name__=='__main__':
     unittest.main()