self.assertIsInstance(NonPR(), PR)
self.assertIsSubclass(NonPR, PR)
+ self.assertNotIn("__protocol_attrs__", vars(NonP))
+ self.assertNotIn("__protocol_attrs__", vars(NonPR))
+ self.assertNotIn("__callable_proto_members_only__", vars(NonP))
+ self.assertNotIn("__callable_proto_members_only__", vars(NonPR))
+
+ acceptable_extra_attrs = {
+ '_is_protocol', '_is_runtime_protocol', '__parameters__',
+ '__subclasshook__', '__abstractmethods__', '_abc_impl',
+ '__init__', '__annotations__',
+ }
+ self.assertLessEqual(vars(NonP).keys(), vars(C).keys() | acceptable_extra_attrs)
+ self.assertLessEqual(
+ vars(NonPR).keys(), vars(D).keys() | acceptable_extra_attrs
+ )
+
def test_custom_subclasshook(self):
class P(Protocol):
x = 1
# but is necessary for several reasons...
def __init__(cls, *args, **kwargs):
super().__init__(*args, **kwargs)
- cls.__protocol_attrs__ = _get_protocol_attrs(cls)
- # PEP 544 prohibits using issubclass()
- # with protocols that have non-method members.
- cls.__callable_proto_members_only__ = all(
- callable(getattr(cls, attr, None)) for attr in cls.__protocol_attrs__
- )
+ if getattr(cls, "_is_protocol", False):
+ cls.__protocol_attrs__ = _get_protocol_attrs(cls)
+ # PEP 544 prohibits using issubclass()
+ # with protocols that have non-method members.
+ cls.__callable_proto_members_only__ = all(
+ callable(getattr(cls, attr, None)) for attr in cls.__protocol_attrs__
+ )
def __subclasscheck__(cls, other):
if (