Pytest Databricks Notebook Return