Add multiply logic.

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