Generates dummy test tree

generate_tests_tree.py

"""
Creates a dummy tree with test files, with or without conftest.py files.

Usage:

    generate_tests_tree.py 4 6 10 0

Generates a tree 4 levels deep, 6 folders per level and 10 test files each, without any conftest.py files. Each
test file contains 10 dummy tests.
"""

import os
import sys

import shutil

levels, folders, tests, conftests = sys.argv[1:]

if os.path.isdir('root'):
    shutil.rmtree('root')
os.mkdir('root')
for level_index in range(int(levels)):
    level_folder = 'root/pkg%d' % level_index
    os.mkdir(level_folder)

    with open(level_folder + '/__init__.py', 'w') as f:
        pass

    for folder_index in range(int(folders)):
        directory = level_folder + '/mod%s' % folder_index
        os.mkdir(directory)

        with open(directory + '/__init__.py', 'w') as f:
            pass

        for test_index in range(int(tests)):
            if int(conftests):
                with open('%s/conftest.py' % directory, 'w') as conftest_file:
                    pass

            with open('%s/test_%d.py' % (directory, test_index), 'w') as test_file:
                code = (
                    "import pytest\n"
                    "@pytest.mark.parametrize('i', range(10))\n"
                    "def test(i): pass\n"
                )
                test_file.write(code)