How I can test my Python code that I wrote using notebooks?

alexott
Databricks Employee
Databricks Employee

I've written the code in the notebooks using the Python, and I want to add tests to it to make sure that it won't break when I do more changes.

What tools can I use for that tasks?