Source code for tests.test_Generator

import os
from mmdesigner.Generator import Generator
from mmdesigner.TreeWalker import TreeWalker


[docs]def test_generate_stl():
    """Check generator builds all STL files"""
    root = "tests/test_data/model"
    gen = Generator(root)
    tw = TreeWalker(root, "scad", None)
    tw.clean("stl")
    gen.process_all("stl")
    assert os.path.isfile(os.path.join(root, "wing/wing.stl"))


[docs]def test_generate_mass_properties():
    """Check creates json file with mass properties"""
    root = "tests/test_data/model"
    gen = Generator(root)
    tw = TreeWalker(root, "stl", None)
    tw.clean("json")
    gen.process_parts("mass")
    assert os.path.isfile(os.path.join(root, "wing/wing.json"))


[docs]def test_generates_excel():
    """Check generation of excel file"""
    root = "tests/test_data/model"
    gen = Generator(root)
    gen.gen_excel("model")
    assert os.path.isfile(os.path.join(root, "model.xlsx"))


def test_process_all():
    root = "tests/test_data/model"
    gen = Generator(root)
    gen.process_all('stl')
    gen.process_all('mass')
    assert os.path.isfile(os.path.join(root, "body/rudder/rudder.stl"))
    assert os.path.isfile(os.path.join(root, "body/rudder/rudder.json"))