[ty] functional enum syntax

astral-sh/ty#876

Author: thejchap

crates/ty_python_semantic/resources/mdtest/dataclasses/dataclasses.md

@@ -1936,16 +1936,16 @@ from enum import Enum
 
 E = Enum("E", "A B C")
 
-# TODO: should emit `invalid-dataclass`
+# error: [invalid-dataclass] "Cannot use `dataclass()` on an enum class"
 dataclass(E)
 
-# TODO: should emit `invalid-dataclass`
+# error: [invalid-dataclass] "Cannot use `dataclass()` on an enum class"
 dataclass()(E)
 
-# TODO: should emit `invalid-dataclass`
+# error: [invalid-dataclass] "Cannot use `dataclass()` on an enum class"
 dataclass(Enum("Inline1", "X Y"))
 
-# TODO: should emit `invalid-dataclass`
+# error: [invalid-dataclass] "Cannot use `dataclass()` on an enum class"
 dataclass()(Enum("Inline2", "X Y"))
 ```
 

crates/ty_python_semantic/resources/mdtest/enums.md

@@ -1251,7 +1251,35 @@ def _(x: EnumWithSubclassOfEnumMetaMetaclass):
 
 ## Function syntax
 
-To do: <https://typing.python.org/en/latest/spec/enums.html#enum-definition>
+```py
+from enum import Enum
+from ty_extensions import enum_members
+
+Color = Enum("Color", "RED", "GREEN", "BLUE")
+
+# revealed: tuple[Literal["RED"], Literal["GREEN"], Literal["BLUE"]]
+reveal_type(enum_members(Color))
+
+Color = Enum("Color", "RED GREEN BLUE")
+
+# revealed: tuple[Literal["RED"], Literal["GREEN"], Literal["BLUE"]]
+reveal_type(enum_members(Color))
+
+Color = Enum("Color", "RED, GREEN, BLUE")
+
+# revealed: tuple[Literal["RED"], Literal["GREEN"], Literal["BLUE"]]
+reveal_type(enum_members(Color))
+
+Color = Enum("Color", [("RED", 1), ("GREEN", 2), ("BLUE", 3)])
+
+# TODO: This should be supported
+reveal_type(enum_members(Color))  # revealed: Unknown
+
+Color = Enum("Color", (("RED", 1), ("GREEN", 2), ("BLUE", 3)))
+
+# TODO: This should be supported
+reveal_type(enum_members(Color))  # revealed: Unknown
+```
 
 ## Exhaustiveness checking