import logging

from jedi.parser import pytree
from jedi.parser.pgen2 import Driver


def test_basic():
    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)

    compare('\na #pass\n')
    compare('wblabla* 1\t\n')
    compare('def x(a, b:3): pass\n')

    print(d)
    tree = d.parse_string('wblabla* 1\t\n')
    print(repr(tree))
    print(tree)
    #import pdb; pdb.set_trace()
    print(repr(d.parse_string('def x(a, b:3): pass\n')))
    print()
    x = d.parse_string('\na #pass\n')
    print(repr(x))
    print(x.get_code())
    assert False