test_gen_unit_tables.py

from pathlib import Path
from unittest.mock import mock_open, patch

from gen_unit_tables import write_unit_table


def test_write_unit_table(snapshot):
    with patch.object(Path, "open", mock_open()) as mock_open_file:
        write_unit_table(filename=Path())

    lines = [
        args[0] for args in mock_open_file.return_value.write.call_args_list
    ]
    # Exclude the first line, because it contains "Generated by /home/username/..."
    assert lines[1:] == snapshot