Merge pull request #139 from asottile/312_syntax

fix tests for python 3.12
asottile · Jun 10, 2023 · 2b12235 · 2b12235
2 parents be01ce6 + 3a67f63
commit 2b12235
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -10,4 +10,4 @@ jobs:
   main:
     uses: asottile/workflows/.github/workflows/tox.yml@v1.0.0
     with:
-      env: '["py38", "py39", "py310"]'
+      env: '["py38", "py39", "py310", "py311", "py312"]'
diff --git a/tests/astpretty_test.py b/tests/astpretty_test.py
@@ -302,7 +302,26 @@ def f(
 
 
 def test_pformat_py38_type_comments(tmpdir, capsys):
-    expected = '''\
+    expected_38 = '''\
+Module(
+    body=[
+        FunctionDef(
+            lineno=1,
+            col_offset=0,
+            end_lineno=2,
+            end_col_offset=8,
+            name='f',
+            args=arguments(posonlyargs=[], args=[], vararg=None, kwonlyargs=[], kw_defaults=[], kwarg=None, defaults=[]),
+            body=[Pass(lineno=2, col_offset=4, end_lineno=2, end_col_offset=8)],
+            decorator_list=[],
+            returns=None,
+            type_comment='() -> None',
+        ),
+    ],
+    type_ignores=[TypeIgnore(lineno=2, tag='')],
+)
+'''  # noqa: E501
+    expected_312 = '''\
 Module(
     body=[
         FunctionDef(
@@ -316,11 +335,13 @@ def test_pformat_py38_type_comments(tmpdir, capsys):
             decorator_list=[],
             returns=None,
             type_comment='() -> None',
+            type_params=[],
         ),
     ],
     type_ignores=[TypeIgnore(lineno=2, tag='')],
 )
 '''  # noqa: E501
+    expected = expected_312 if sys.version_info >= (3, 12) else expected_38
     mod = (
         'def f():  # type: () -> None\n'
         '    pass  # type: ignore\n'