Function test_assembly_multi_subshape_import

tests/test_assembly.py:1110–1154

Test if a file containing subshape information can be imported correctly.

(tmpdir, multi_subshape_assy, kind)

@pytest.mark.parametrize("kind", ["step", "xml", "xbf"])
def test_assembly_multi_subshape_import(tmpdir, multi_subshape_assy, kind):
    """
    Test if a file containing subshape information can be imported correctly.
    """

    assy_step_path = f"multi_subshape_assy.{kind}"

    with chdir(tmpdir):
        # Export the assembly
        multi_subshape_assy.export(assy_step_path)

        # Import the file back in
        imported_assy = cq.Assembly.load(assy_step_path)

    # Check that the top-level assembly name is correct
    assert imported_assy.name == "top_level"

    # Check the advanced face name for the first cube
    assert len(imported_assy.children[0]._subshape_names) == 1
    assert (
        list(imported_assy.children[0]._subshape_names.values())[0] == "cube_1_top_face"
    )

    # Check the color for the first cube
    color = list(imported_assy.children[0]._subshape_colors.values())[0]
    assert Quantity_NameOfColor.Quantity_NOC_RED == color.wrapped.GetRGB().Name()

    # Check the layer info for the first cube
    layer_name = list(imported_assy.children[0]._subshape_layers.values())[0]
    assert layer_name == "cube_1_top_face"

    # Check the advanced face name for the second cube
    assert len(imported_assy.children[1]._subshape_names) == 1
    assert (
        list(imported_assy.children[1]._subshape_names.values())[0]
        == "cube_2_right_face"
    )

    # Check the color
    color = list(imported_assy.children[1]._subshape_colors.values())[0]
    assert Quantity_NameOfColor.Quantity_NOC_RED == color.wrapped.GetRGB().Name()

    # Check the layer info
    layer_name = list(imported_assy.children[1]._subshape_layers.values())[0]
    assert layer_name == "cube_2_right_face"


def test_bad_step_file_import(tmpdir):