Function test_nested_subassembly_step_import

tests/test_assembly.py:1272–1299

Tests if the STEP import works correctly with nested subassemblies.

(tmpdir)

def test_nested_subassembly_step_import(tmpdir):
    """
    Tests if the STEP import works correctly with nested subassemblies.
    """

    # Create a simple assembly
    assy = cq.Assembly()
    assy.add(cq.Workplane().box(10, 10, 10), name="box_1")

    # Create a simple subassembly
    subassy = cq.Assembly()
    subassy.add(cq.Workplane().box(5, 5, 5), name="box_2", loc=cq.Location(10, 10, 10))

    # Nest the subassembly
    assy.add(subassy)

    # Export and then re-import the nested assembly STEP
    with chdir(tmpdir):
        assy.export("nested_assembly_step.step")
        imported_assy = cq.Assembly.importStep("nested_assembly_step.step")

    # Check the locations
    assert imported_assy.children[0].loc.toTuple()[0] == (0.0, 0.0, 0.0)
    assert imported_assy.children[1].objects["box_2"].loc.toTuple()[0] == (
        10.0,
        10.0,
        10.0,
    )


@pytest.mark.parametrize("kind", ["step", "xml", "xbf"])