diff --git a/test_ast.py b/test_ast.py index 7979100..d0108c5 100755 --- a/test_ast.py +++ b/test_ast.py @@ -71,6 +71,8 @@ def _install_equal_methods(): _install_generic_equal(ast.Typedef, 'name alias namespace') _install_generic_equal(ast.VariableDeclaration, 'name type initial_value namespace') + + _install_equal_methods() @@ -1140,5 +1142,6 @@ def test_inline_function(self): Function('fn', list(get_tokens('void')), []), nodes[0]) + if __name__ == '__main__': unittest.main() diff --git a/test_tokenize.py b/test_tokenize.py index 49b78e8..53aa110 100755 --- a/test_tokenize.py +++ b/test_tokenize.py @@ -53,6 +53,7 @@ def __eq__(self, other): self.start == other.start and self.end == other.end) + tokenize.Token.__eq__ = __eq__