<filename>openpnm/geometry/_circles.py
from openpnm.models.collections.geometry import circles_and_rectangles
from openpnm.geometry import GenericGeometry
from openpnm.utils import Docorator


docstr = Docorator()


@docstr.dedent
class CirclesAndRectangles(GenericGeometry):
    r"""
    2D representation of spheres and cylinders suitable for 2D simulations

    Parameters
    ----------
    %(GenericGeometry.parameters)s

    """

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.models.update(circles_and_rectangles)
        self.regenerate_models()
