def transpile(source, headers=False, testing=False): """ Transpile a single python translation unit (a python script) into C++ 14 code. """ tree = ast.parse(source) rewriter = PythonMainRewriter("cpp") tree = rewriter.visit(tree) add_variable_context(tree) add_scope_context(tree) add_list_calls(tree) add_imports(tree) transpiler = CppTranspiler() buf = [] if testing: buf += ['#include "catch.hpp"'] transpiler.use_catch_test_cases = True if headers: buf += transpiler._headers buf += transpiler._usings if testing or headers: buf.append("") # Force empty line cpp = transpiler.visit(tree) return "\n".join(buf) + cpp
def test_call_added(self): source = parse("results = []", "results.append(x)") add_list_calls(source) assert len(source.scopes[-1].vars[0].calls) == 1
def test_global_list_assignment_with_later_append(self): source = parse("results = []", "def add_x():", " results.append(2)") add_list_calls(source) results = source.body[0] t = value_type(results) assert t == "2"
def test_list_assignment_with_append_unknown_value(self): source = parse("results = []", "x = 3", "results.append(x)") add_list_calls(source) results = source.body[0] t = value_type(results) assert t == "3"
def test_list_assignment_based_on_later_append(self): source = parse("x = 3", "results = []", "results.append(x)") add_list_calls(source) results = source.body[1] t = value_type(results) assert t == "3"
def test_is_list(): source = parse("list1 = []", "list2 = list1") add_list_calls(source) list2 = source.body[1] assert is_list(list2)
def test_decltype_list_var(): source = parse("results = []", "x = 3", "results.append(x)") add_list_calls(source) results = source.body[0] t = decltype(results) assert t == "std::vector<decltype(3)>"