# Define square(n) returning n * n and add_one(n) returning n + 1

Canonical URL: <https://datadriven.io/problems/define-squaren-returning-n-n-and-addonen-returning-n-d3337268>

Domain: Python · Difficulty: medium

## Problem

Define square(n) returning n * n and add_one(n) returning n + 1. Print the result of add_one(square(4)).

## Related

- [All practice problems](https://datadriven.io/problems)
- [Mock interview mode](https://datadriven.io/interview/define-squaren-returning-n-n-and-addonen-returning-n-d3337268)
- [Python Interview Questions](https://datadriven.io/python-interview-questions)
- [Data Engineering Interview Prep Guide](https://datadriven.io/data-engineer-interview-prep)
- [Daily Challenge](https://datadriven.io/daily)

---

Source: DataDriven (https://datadriven.io). 100% free data engineering interview prep. Live code execution against Postgres 16, Python 3.11, and Spark sandboxes. No paywall, no premium tier, no signup gate.