22 lines
No EOL
520 B
Python
22 lines
No EOL
520 B
Python
import unittest
|
|
from utils.math_utils import add, subtract, multiply, divide
|
|
|
|
|
|
class TestMathUtils(unittest.TestCase):
|
|
def test_add(self):
|
|
self.assertEqual(add(1, 2), 3)
|
|
|
|
def test_subtract(self):
|
|
self.assertEqual(subtract(5, 3), 2)
|
|
|
|
def test_multiply(self):
|
|
self.assertEqual(multiply(3, 4), 12)
|
|
|
|
def test_divide(self):
|
|
self.assertEqual(divide(10, 2), 5)
|
|
with self.assertRaises(ValueError):
|
|
divide(10, 0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |