import pytest from importer.uploader import create_folders @pytest.fixture def nxc(mocker): return mocker.MagicMock() @pytest.mark.parametrize("path,folders", [ ("folder1", ["/folder1"]), ("folder1/folder2", ["/folder1", "/folder1/folder2"]), ("/folder1/folder2", ["/folder1", "/folder1/folder2"]) ]) def test_create_folders(nxc, path, folders): create_folders(path, nxc) for call, folder in zip(nxc.method_calls, folders): assert folder == call.args[0]