17 أسطر
361 B
Python
17 أسطر
361 B
Python
import unittest
|
|
from main import add, subtract
|
|
|
|
|
|
class TestCalculator(unittest.TestCase):
|
|
def test_add(self):
|
|
self.assertEqual(add(2, 3), 5)
|
|
self.assertEqual(add(-1, 1), 0)
|
|
|
|
def test_subtract(self):
|
|
self.assertEqual(subtract(5, 2), 2)
|
|
self.assertEqual(subtract(0, 1), -1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|