You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
2.7 KiB
111 lines
2.7 KiB
import os |
|
import shutil |
|
|
|
from cookiecutter import main |
|
import pytest |
|
|
|
CCDS_ROOT = os.path.abspath( |
|
os.path.join( |
|
__file__, |
|
os.pardir, |
|
os.pardir |
|
) |
|
) |
|
|
|
|
|
@pytest.fixture(scope='function') |
|
def default_baked_project(tmpdir): |
|
out_dir = str(tmpdir.mkdir('data-project')) |
|
|
|
main.cookiecutter( |
|
CCDS_ROOT, |
|
no_input=True, |
|
extra_context={}, |
|
output_dir=out_dir |
|
) |
|
|
|
# default project name is project_name |
|
yield os.path.join(out_dir, 'project_name') |
|
|
|
# cleanup after |
|
shutil.rmtree(out_dir) |
|
|
|
|
|
def test_readme(default_baked_project): |
|
readme_path = os.path.join(default_baked_project, 'README.md') |
|
|
|
assert os.path.exists(readme_path) |
|
assert no_curlies(readme_path) |
|
|
|
|
|
def test_license(default_baked_project): |
|
license_path = os.path.join(default_baked_project, 'LICENSE') |
|
|
|
assert os.path.exists(license_path) |
|
assert no_curlies(license_path) |
|
|
|
|
|
def test_requirements(default_baked_project): |
|
reqs_path = os.path.join(default_baked_project, 'requirements.txt') |
|
|
|
assert os.path.exists(reqs_path) |
|
assert no_curlies(reqs_path) |
|
|
|
|
|
def test_makefile(default_baked_project): |
|
makefile_path = os.path.join(default_baked_project, 'Makefile') |
|
|
|
assert os.path.exists(makefile_path) |
|
assert no_curlies(makefile_path) |
|
|
|
|
|
def test_folders(default_baked_project): |
|
expected_dirs = [ |
|
'data', |
|
os.path.join('data', 'external'), |
|
os.path.join('data', 'interim'), |
|
os.path.join('data', 'processed'), |
|
os.path.join('data', 'raw'), |
|
'docs', |
|
'models', |
|
'notebooks', |
|
'references', |
|
'reports', |
|
os.path.join('reports', 'figures'), |
|
'src', |
|
os.path.join('src', 'data'), |
|
os.path.join('src', 'features'), |
|
os.path.join('src', 'models'), |
|
os.path.join('src', 'visualization') |
|
] |
|
|
|
ignored_dirs = [ |
|
default_baked_project, |
|
os.path.join(default_baked_project, '__pycache__') |
|
] |
|
|
|
abs_expected_dirs = [os.path.join(default_baked_project, d) for |
|
d in expected_dirs] |
|
|
|
abs_dirs, _, _ = list(zip(*os.walk(default_baked_project))) |
|
|
|
assert len(set(abs_expected_dirs + ignored_dirs) - set(abs_dirs)) == 0 |
|
|
|
|
|
def no_curlies(filepath): |
|
""" Utility to make sure no curly braces appear in a file. |
|
That is, was jinja able to render everthing? |
|
""" |
|
with open(filepath, 'r') as f: |
|
data = f.read() |
|
|
|
template_strings = [ |
|
'{{', |
|
'}}', |
|
'{%', |
|
'%}' |
|
] |
|
|
|
template_strings_in_file = [s in data for s in template_strings] |
|
|
|
return not any(template_strings_in_file)
|
|
|