38 lines
952 B
Python
38 lines
952 B
Python
import unittest
|
|
|
|
|
|
# Define some test cases
|
|
class TestAddition(unittest.TestCase):
|
|
def test_add_positive(self):
|
|
self.assertEqual(1 + 2, 3)
|
|
|
|
def test_add_negative(self):
|
|
self.assertEqual(-1 + (-1), -2)
|
|
|
|
|
|
class TestSubtraction(unittest.TestCase):
|
|
def test_subtract_positive(self):
|
|
self.assertEqual(5 - 3, 2)
|
|
|
|
def test_subtract_negative(self):
|
|
self.assertEqual(-5 - (-2), -3)
|
|
|
|
|
|
# Now let's create a TestSuite
|
|
def suite():
|
|
suite = unittest.TestSuite()
|
|
|
|
# Add tests to the suite
|
|
suite.addTest(TestAddition('test_add_positive'))
|
|
suite.addTest(TestAddition('test_add_negative'))
|
|
suite.addTest(TestSubtraction('test_subtract_positive'))
|
|
# suite.addTest(TestSubtraction('test_subtract_negative'))
|
|
# suite.addTest(TestSubtraction('test_subtract_negative2'))
|
|
|
|
return suite
|
|
|
|
|
|
# Running the suite
|
|
if __name__ == "__main__":
|
|
runner = unittest.TextTestRunner()
|
|
runner.run(suite()) |