Test cases with Literal return type for __bool__

python · Aug 16, 2023 · ef1f9cc · ef1f9cc
1 parent bc684ea
commit ef1f9cc
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test-data/unit/check-protocols.test b/test-data/unit/check-protocols.test
@@ -2893,8 +2893,18 @@ ss(None)
 class SupportsBool(Protocol):
     def __bool__(self) -> bool: ...
 
+class SupportsBoolLiteralTrue(Protocol):
+    def __bool__(self) -> Literal[True]: ...
+
+class SupportsBoolLiteralFalse(Protocol):
+    def __bool__(self) -> Literal[False]: ...
+
 def sb(s: SupportsBool) -> None: pass
 sb(None)
+def sblt(s: SupportsBoolLiteralTrue) -> None: pass
+sblt(None)
+def sblf(s: SupportsBoolLiteralFalse) -> None: pass
+sblf(None)
 
 class HashableStr(Protocol):
     def __str__(self) -> str: ...