import pytest
import attr
@pytest.mark.parametrize("slots", [True, False])
def test_init_subclass_vanilla(slots):
@attr.s(slots=slots)
class Base:
def __init_subclass__(cls, param, **kw):
super().__init_subclass__(**kw)
cls.param = param
class Vanilla(Base, param="foo"):
pass
assert "foo" == Vanilla().param
def test_init_subclass_attrs():
@attr.s(slots=False)
class Base:
def __init_subclass__(cls, param, **kw):
super().__init_subclass__(**kw)
cls.param = param
@attr.s
class Attrs(Base, param="foo"):
pass
assert "foo" == Attrs().param