diff --git a/test/test_new_parser.py b/test/test_new_parser.py index b5cac098..49a4e7c5 100644 --- a/test/test_new_parser.py +++ b/test/test_new_parser.py @@ -1,22 +1,10 @@ -import logging - -from jedi.parser import pytree -from jedi.parser.pgen2 import Driver +from jedi.parser import Parser def test_basic_parsing(): def compare(string): """Generates the AST object and then regenerates the code.""" - assert d.parse_string(string).get_code() == string - - #if self.options["print_function"]: - # python_grammar = pygram.python_grammar_no_print_statement - #else: - # When this is True, the refactor*() methods will call write_file() for - # files processed even if they were not changed during refactoring. If - # and only if the refactor method's write parameter was True. - logger = logging.getLogger("RefactoringTool") - d = Driver(pytree.python_grammar, convert=pytree.convert, logger=logger) + assert Parser(string).module.get_code() == string compare('\na #pass\n') compare('wblabla* 1\t\n')