Function test_copied_assembly_import

tests/test_assembly.py:1230–1269

Tests to make sure that copied children in assemblies work correctly.

(tmpdir)

def test_copied_assembly_import(tmpdir):
    """
    Tests to make sure that copied children in assemblies work correctly.
    """
    from cadquery import Assembly, Location, Color
    from cadquery.func import box, rect

    # prepare the model
    def make_model(name: str, COPY: bool):

        b = box(1, 1, 1)

        assy = Assembly(name="test_assy")
        assy.add(box(1, 2, 5), color=Color("green"))

        for i, v in enumerate(rect(10, 10).vertices()):
            assy.add(
                b.copy() if COPY else b,
                name=f"element_{i}",
                loc=Location(v.Center()),
                color=Color("red"),
            )

        with chdir(tmpdir):
            assy.export(name)

        return assy

    make_model("test_assy_copy.step", True)
    make_model("test_assy.step", False)

    # import the assy with copies
    with chdir(tmpdir):
        assy_copy = Assembly.importStep("test_assy_copy.step")
    assert 5 == len(assy_copy.children)

    # import the assy without copies
    with chdir(tmpdir):
        assy_normal = Assembly.importStep("test_assy.step")
    assert 5 == len(assy_normal.children)


def test_nested_subassembly_step_import(tmpdir):