I'm trying to write a basic unit test for a task in our workflow and am running into an issue. I've tried using the example from https://docs.flyte.org/projects/flytekit/en/latest/generated/flytekit.testing.task_mock.html and I've tried calling the python method directly. In both cases I'm not able to get the output from the function. Is there something very simple I'm missing?

If you share a simple example, I am happy to take a look. I have had success following these directions: https://docs.flyte.org/projects/cookbook/en/latest/auto/testing/mocking.html#sphx-glr-auto-testing-mocking-py.