input: Write a function to add two numbers and use unit test to validate it.
  assert add(2, 3) == 5
  assert add(4, 5) == 9
output: def add(a, b):
      return a + b
  
  # Unit tests
  assert add(2, 3) == 5
  assert add(4, 5) == 9