Test tasks returning a file - use other temporary directory

[HansBambel]

I have a task that I want to test that returns a CSVFile:

from flytekit.types.file import CSVFile
import polars as pl
from pathlib import Path

@task
def save_curriculum_csv(curriculum_df: pl.DataFrame) -> CSVFile:
    """Save curriculum to a csv file under curriculum/curriculum_map.csv."""
    # Set output folder and create it if it doesn't exist
    output_folder = Path(fl.current_context().working_directory) / "curriculum"
    output_folder.mkdir(parents=True, exist_ok=True)

    # Write the data to curriculum_map.csv file
    curriculum_df.write_csv(output_folder / "curriculum_map.csv")
    return CSVFile(str(output_folder / "curriculum_map.csv"))

This is my test:

import polars as pl
from polars.testing import assert_frame_equal

def test_get_curriculum_map_csv_created_correctly():
    """Test that save_curriculum_csv creates the expected csv file."""
    expected = pl.DataFrame(
        {
            "grade": [1],
            "subject": ["Math"],
            "strand": ["Numbers"],
            "substrand": ["Addition"],
            "content_id": ["important/slashes"],
        }
    )
    csv_file = save_curriculum_csv(expected)
    result = pl.read_csv(csv_file.path)
    assert_frame_equal(result, expected)

I am using fl.current_context().working_directory for a path. If I understood it correctly, this path is just temporary in /var/folders which will be cleared after a reboot.
When testing, I want to use pytest's temporary folders via the fixture tmp_path (https://docs.pytest.org/en/7.1.x/how-to/tmp_path.html) as this at most contains 3 instances of runs.

When I am not rebooting a lot or not manually cleaning out the /var/folders directory it gets cluttered with a lot of files from running tests.

How would I go about this? Is that even possible?

[Future-Outlier]

Here can help you!
https://github.com/flyteorg/flytekit/blob/d42a0024436a3bd70b3abddc12b2cd9541dbb67f/tests/flytekit/unit/types/file/test_file.py#L11-L27
cc @HansBambel