Some preparations to get Python 2 tests working.

This commit is contained in:
Dave Halter
2017-07-31 12:33:47 +02:00
parent 5fbbb225dd
commit 58c32591d0
8 changed files with 25 additions and 24 deletions

View File

@@ -154,24 +154,24 @@ class NodeOrLeaf(object):
e.g. a statement.
"""
def _get_normalizer(self, normalizer_config):
def _get_normalizer(self, normalizer_config, grammar):
if normalizer_config is None:
normalizer_config = self.default_normalizer_config
if normalizer_config is None:
raise ValueError("You need to specify a normalizer, because "
"there's no default normalizer for this tree.")
return normalizer_config.create_normalizer()
return normalizer_config.create_normalizer(grammar)
def _normalize(self, normalizer_config=None):
def _normalize(self, grammar, normalizer_config=None):
"""
TODO this is not public, yet.
The returned code will be normalized, e.g. PEP8 for Python.
"""
normalizer = self._get_normalizer(normalizer_config)
normalizer = self._get_normalizer(normalizer_config, grammar)
return normalizer.walk(self)
def _get_normalizer_issues(self, normalizer_config=None):
normalizer = self._get_normalizer(normalizer_config)
def _get_normalizer_issues(self, grammar, normalizer_config=None):
normalizer = self._get_normalizer(normalizer_config, grammar)
normalizer.walk(self)
return normalizer.issues