Problem statement

You’ve got a basic utility function that calculates the square of a number. It’s used across your app, but you want to track what values it receives—and what it returns—without modifying the original function. This is your first step into applying decorators for visibility and traceability.

Goal

Create a withLogging decorator that logs the input arguments and return value of the square function. Then, use it to decorate the original function.

Constraints

• Do not modify the square function.

• The withLogging decorator must return a new function.

• Logging must happen before and after the original function runs.

• Do not add logging directly inside the square function.

Sample output

The examples below illustrate what the output should look like:

Good luck trying the problem! If you’re unsure how to proceed, check the “Solution” tab above.