Test the ast_compat layer works with removed types

Author: dflook

test/test_is_constant_node.py

@@ -13,6 +13,8 @@
 @pytest.mark.filterwarnings("ignore:ast.NameConstant is deprecated:DeprecationWarning")
 @pytest.mark.filterwarnings("ignore:ast.Ellipsis is deprecated:DeprecationWarning")
 def test_type_nodes():
+    if sys.version_info >= (3, 14):
+        pytest.skip('Deprecated AST types removed in Python 3.14')
     assert is_constant_node(ast.Str('a'), ast.Str)
 
     if hasattr(ast, 'Bytes'):
@@ -42,6 +44,8 @@ def test_constant_nodes():
     # only test on python 3.8+
     if sys.version_info < (3, 8):
         pytest.skip('Constant not available')
+    if sys.version_info >= (3, 14):
+        pytest.skip('Deprecated AST types removed in Python 3.14')
 
     assert is_constant_node(ast.Constant('a'), ast.Str)
     assert is_constant_node(ast.Constant(b'a'), ast.Bytes)
@@ -51,3 +55,39 @@ def test_constant_nodes():
     assert is_constant_node(ast.Constant(False), ast.NameConstant)
     assert is_constant_node(ast.Constant(None), ast.NameConstant)
     assert is_constant_node(ast.Constant(ast.literal_eval('...')), ast.Ellipsis)
+
+
+def test_ast_compat_types_python314():
+    """Test that ast_compat provides the removed AST types in Python 3.14+"""
+    if sys.version_info < (3, 14):
+        pytest.skip('ast_compat types test only for Python 3.14+')
+    
+    import python_minifier.ast_compat as ast_compat
+    
+    # Test that ast_compat provides the removed types
+    assert is_constant_node(ast_compat.Str('a'), ast_compat.Str)
+    assert is_constant_node(ast_compat.Bytes(b'a'), ast_compat.Bytes)
+    assert is_constant_node(ast_compat.Num(1), ast_compat.Num)
+    assert is_constant_node(ast_compat.Num(0), ast_compat.Num)
+    assert is_constant_node(ast_compat.NameConstant(True), ast_compat.NameConstant)
+    assert is_constant_node(ast_compat.NameConstant(False), ast_compat.NameConstant)
+    assert is_constant_node(ast_compat.NameConstant(None), ast_compat.NameConstant)
+    assert is_constant_node(ast_compat.Ellipsis(), ast_compat.Ellipsis)
+
+
+def test_ast_compat_constant_nodes_python314():
+    """Test that ast_compat works with Constant nodes in Python 3.14+"""
+    if sys.version_info < (3, 14):
+        pytest.skip('ast_compat constant test only for Python 3.14+')
+    
+    import python_minifier.ast_compat as ast_compat
+    
+    # Test that Constant nodes work with ast_compat types
+    assert is_constant_node(ast.Constant('a'), ast_compat.Str)
+    assert is_constant_node(ast.Constant(b'a'), ast_compat.Bytes)
+    assert is_constant_node(ast.Constant(1), ast_compat.Num)
+    assert is_constant_node(ast.Constant(0), ast_compat.Num)
+    assert is_constant_node(ast.Constant(True), ast_compat.NameConstant)
+    assert is_constant_node(ast.Constant(False), ast_compat.NameConstant)
+    assert is_constant_node(ast.Constant(None), ast_compat.NameConstant)
+    assert is_constant_node(ast.Constant(...), ast_compat.Ellipsis)