import unittest from operator import add import pytest class AddTestCase(unittest.TestCase): def test_1_2(self): a = 1 b = 2 desired = 3 actual = add(a, b) assert actual == desired, f'{actual} != {desired}' def test_2_3(self): a = 2 b = 3 desired = 5 self.assertEqual(add(2, 3), desired) def test_add_5_2(): a = 5 b = 2 assert 7 == add(a, b) @pytest.mark.parametrize('a,b,desired', [(1, 2, 3), (3, 3, 5)]) def test_many(a, b, desired): assert desired == add(a, b) if __name__ == '__main__': unittest.main()