Zoox Interview Question

Write a simple function that does a particular mathematical operation. Think about corner cases and implement tests.