pointnet++/Pointnet2_PyTorch/tests/test_semseg.py

import pytest


@pytest.mark.parametrize("use_xyz", ["True", "False"])
@pytest.mark.parametrize("model", ["ssg", "msg"])
def test_semseg(use_xyz, model):
    model = pytest.helpers.get_model(
        ["task=semseg", f"model={model}", f"model.use_xyz={use_xyz}"]
    )
    pytest.helpers.semseg_test(model)