Function test_step_roundtrip_with_materials

tests/test_assembly.py:890–920

Tests to make sure that once materials have been exported to a file format such as STEP, the materials can be imported again.

(kind, tmpdir)

@pytest.mark.parametrize("kind", ["step", "xml", "xbf"])
def test_step_roundtrip_with_materials(kind, tmpdir):
    """
    Tests to make sure that once materials have been exported to a file format
    such as STEP, the materials can be imported again.
    """

    materials_assy = cq.Assembly()
    materials_assy.add(
        cq.Workplane().box(10, 10, 10),
        name="cube_1",
        material=cq.Material(name="copper"),
    )

    materials_path = f"roundtrip_materials.{kind}"

    with chdir(tmpdir):
        exportAssembly(materials_assy, materials_path)

        # Read the contents as a step file as a string so we can check the outputs
        with open(materials_path, "r") as f:
            step_contents = f.read()

        # Make sure that the face name string is present in the exported STEP contents
        assert "copper" in step_contents

        # Import the STEP file back in as an assembly
        test_assy = cq.Assembly.importStep(materials_path)

    # Make sure that the material was re-imported
    assert test_assy.children[0].material is not None
    assert test_assy.children[0].material.name == "copper"


def test_step_export_assembly_unit_default(tmp_path_factory):