Added test for -> in literal

pybind · Jan 10, 2025 · 9bb4f20 · 9bb4f20
1 parent b577012
commit 9bb4f20
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/tests/test_pytypes.cpp b/tests/test_pytypes.cpp
@@ -977,8 +977,12 @@ TEST_SUBMODULE(pytypes, m) {
     m.def("identity_literal_percent", [](const py::typing::Literal<"\"%\""> &x) { return x; });
     m.def("identity_literal_curly_open", [](const py::typing::Literal<"\"{\""> &x) { return x; });
     m.def("identity_literal_curly_close", [](const py::typing::Literal<"\"}\""> &x) { return x; });
+    m.def("identity_literal_arrow", [](const py::typing::Literal<"\"->\""> &x) { return x; });
+    m.def("identity_literal_arrow_with_callable",
+          [](const py::typing::Callable<RealNumber(const py::typing::Literal<"\"->\""> &,
+                                                   const RealNumber &)> &x) { return x; });
     m.def("identity_literal_all_special_chars",
-          [](const py::typing::Literal<"\"!@!!{%}\""> &x) { return x; });
+          [](const py::typing::Literal<"\"!@!!->{%}\""> &x) { return x; });
     m.def("annotate_generic_containers",
           [](const py::typing::List<typevar::TypeVarT> &l) -> py::typing::List<typevar::TypeVarV> {
               return l;

diff --git a/tests/test_pytypes.py b/tests/test_pytypes.py
@@ -1065,9 +1065,17 @@ def test_literal(doc):
         doc(m.identity_literal_curly_close)
         == 'identity_literal_curly_close(arg0: Literal["}"]) -> Literal["}"]'
     )
+    assert (
+        doc(m.identity_literal_arrow)
+        == 'identity_literal_arrow(arg0: Literal["->"]) -> Literal["->"]'
+    )
+    assert (
+        doc(m.identity_literal_arrow_with_callable)
+        == 'identity_literal_arrow_with_callable(arg0: Callable[[Literal["->"], Union[float, int]], float]) -> Callable[[Literal["->"], Union[float, int]], float]'
+    )
     assert (
         doc(m.identity_literal_all_special_chars)
-        == 'identity_literal_all_special_chars(arg0: Literal["!@!!{%}"]) -> Literal["!@!!{%}"]'
+        == 'identity_literal_all_special_chars(arg0: Literal["!@!!->{%}"]) -> Literal["!@!!->{%}"]'
     )