Function test_meta_step_export_edge_cases

tests/test_assembly.py:838–886

Test all the edge cases of the STEP export function.

(tmpdir)

def test_meta_step_export_edge_cases(tmpdir):
    """
    Test all the edge cases of the STEP export function.
    """

    # Create an assembly where the child is empty
    assy = cq.Assembly(name="top-level")
    subassy = cq.Assembly(name="second-level")
    assy.add(subassy)

    with chdir(tmpdir):
        # Write an assembly with no children
        success = exportStepMeta(assy, "meta_edge_cases1.step")
        assert success

        # Test an object with no color set
        cube = cq.Workplane().box(10.0, 10.0, 10.0)
        assy.add(cube, name="cube")
        success = exportStepMeta(assy, "meta_edge_cases2.step")
        assert success

        # Tag a face that does not match the object
        with pytest.raises(ValueError):
            assy.addSubshape(plane(1, 1), name="cube_top_face")

        # Tag the name but nothing else
        assy.addSubshape(cube.faces(">Z").val(), name="cube_top_face")
        success = exportStepMeta(assy, "meta_edge_cases3.step")
        assert success

        # Reset the assembly
        assy.remove("cube")
        cube = cq.Workplane().box(9.9, 9.9, 9.9)
        assy.add(cube, name="cube")

        # Tag the color but nothing else
        assy.addSubshape(cube.faces(">Z").val(), color=cq.Color(1.0, 0.0, 0.0))
        success = exportStepMeta(assy, "meta_edge_cases4.step")
        assert success

        # Reset the assembly
        assy.remove("cube")
        cube = cq.Workplane().box(9.8, 9.8, 9.8)
        assy.add(cube, name="cube")

        # Tag the layer but nothing else
        assy.addSubshape(cube.faces(">Z").val(), layer="cube_top_face")
        success = exportStepMeta(assy, "meta_edge_cases5.step")
        assert success


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