Test that weakref works on Coords too

1 files changed, 10 insertions, 1 deletions

diff --git a/tests/test_c_ast.py b/tests/test_c_ast.py
index 18fd542..bfe301a 100644
--- a/tests/test_c_ast.py
+++ b/tests/test_c_ast.py
@@ -6,6 +6,7 @@ import weakref
 
 sys.path.insert(0, '..')
 import pycparser.c_ast as c_ast
+import pycparser.plyparser as plyparser
 
 
 class Test_c_ast(unittest.TestCase):
@@ -22,13 +23,21 @@ class Test_c_ast(unittest.TestCase):
         self.failUnless(isinstance(b1.right, c_ast.ID))
         self.assertEqual(b1.right.name, 'joe')
 
-    def test_weakref_works(self):
+    def test_weakref_works_on_nodes(self):
         c1 = c_ast.Constant(type='float', value='3.14')
         wr = weakref.ref(c1)
         cref = wr()
         self.assertEqual(cref.type, 'float')
         self.assertEqual(weakref.getweakrefcount(c1), 1)
 
+    def test_weakref_works_on_coord(self):
+        coord = plyparser.Coord(file='a', line=2)
+        wr = weakref.ref(coord)
+        cref = wr()
+        self.assertEqual(cref.line, 2)
+        self.assertEqual(weakref.getweakrefcount(coord), 1)
+
+
 
 class TestNodeVisitor(unittest.TestCase):
     class ConstantVisitor(c_ast.NodeVisitor):