mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-11-04 14:16:57 +00:00
608d98e833
This change adds a new command-line calculator tool. The tool can perform basic arithmetic operations (add, subtract, multiply, divide) and is invoked from the command line. It includes a full suite of unit tests.
30 lines
866 B
Python
30 lines
866 B
Python
import unittest
|
|
from calculator import add, subtract, multiply, divide
|
|
|
|
class TestCalculator(unittest.TestCase):
|
|
|
|
def test_add(self):
|
|
self.assertEqual(add(2, 3), 5)
|
|
self.assertEqual(add(-1, 1), 0)
|
|
self.assertEqual(add(-1, -1), -2)
|
|
|
|
def test_subtract(self):
|
|
self.assertEqual(subtract(3, 2), 1)
|
|
self.assertEqual(subtract(-1, 1), -2)
|
|
self.assertEqual(subtract(-1, -1), 0)
|
|
|
|
def test_multiply(self):
|
|
self.assertEqual(multiply(2, 3), 6)
|
|
self.assertEqual(multiply(-1, 1), -1)
|
|
self.assertEqual(multiply(-1, -1), 1)
|
|
|
|
def test_divide(self):
|
|
self.assertEqual(divide(6, 3), 2)
|
|
self.assertEqual(divide(-1, 1), -1)
|
|
self.assertEqual(divide(-1, -1), 1)
|
|
with self.assertRaises(ValueError):
|
|
divide(1, 0)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|