def METHOD_NAME(self, etype=2):
    if etype in {'cell', 2}:
        return self.ds.cell
    elif etype in {'edge', 1}:
        return self.ds.edge
    elif etype in {'node', 0}:
        return self.mesh.node
    else:
        raise ValueError("`entitytype` is wrong!")