Sounds like nbdev which indeed exports code and documentation from a single notebook, plus unit tests for external execution.