def assertCompiles(self, expr, code): ec = ExpressionCompiler({'foo', 'baz'}) py_expr = ec.visit(to_expr(expr, {})) first = astor.to_source(py_expr) if not PY3: first = first.replace("u'", "'") second = dedent(code).strip() if first != second: msg = ('Compiled code is not equal:\n\n{}'.format('\n'.join( difflib.ndiff(first.splitlines(), second.splitlines())))) raise self.failureException(msg)
def assertCompiles(self, expr, code): ec = ExpressionCompiler({'foo', 'baz'}) py_expr = ec.visit(to_expr(expr, {})) first = astor.to_source(py_expr) if not PY3: first = first.replace("u'", "'") second = dedent(code).strip() if first != second: msg = ('Compiled code is not equal:\n\n{}' .format('\n'.join(difflib.ndiff(first.splitlines(), second.splitlines())))) raise self.failureException(msg)