fischertechnik-factory-twin/simulator/test/test_add.py

import unittest


def add(x, y):
    """A simple function to add two numbers."""
    return x + y


class TestAdd(unittest.TestCase):

    def test_add_positive_numbers(self):
        """Test adding two positive numbers."""
        self.assertEqual(add(2, 3), 5)

    def test_add_negative_numbers(self):
        """Test adding two negative numbers."""
        self.assertEqual(add(-2, -3), -5)

    def test_add_mixed_numbers(self):
        """Test adding a positive and a negative number."""
        self.assertEqual(add(2, -3), -1)

    def test_add_zero(self):
        """Test adding zero."""
        self.assertEqual(add(5, 0), 5)


if __name__ == '__main__':
    unittest.main()