import calc import unittest class TestAdd(unittest.TestCase): """ Test the add function from the calc library """ def test_add_integers(self): """ Test that the addition of two integers returns the correct total """ result = calc.add(1, 2) self.assertEqual(result, 3) def test_add_floats(self): """ Test that the addition of two floats returns the correct result """ result = calc.add(10.5, 2) self.assertEqual(result, 12.5) def test_add_strings(self): """ Test the addition of two strings returns the two string as one concatenated string """ result = calc.add('abc', 'def') self.assertEqual(result, 'abcdef') if __name__ == '__main__': unittest.main()