def METHOD_NAME(self, client, created_entities):
    # Test setting on creation
    registered_model = client.get_or_create_registered_model(pii=True)
    created_entities.append(registered_model)
    assert registered_model.get_pii() is True
    registered_model = client.create_registered_model(pii=True)
    created_entities.append(registered_model)
    assert registered_model.get_pii() is True
    # Test updating
    registered_model = client.get_or_create_registered_model()
    created_entities.append(registered_model)
    registered_model.set_pii(True)
    assert registered_model.get_pii() is True
    registered_model.set_pii(False)
    assert registered_model.get_pii() is False