def test_codegen_write_simple_strings(module_compiler): gen = CodeChunk() gen.write('def sum(a, b):') gen.indent() gen.write('return a + b') code = str(gen) m = module_compiler(code) assert m.sum(2, 3) == 5
def test_codegen_write_template_strings_kwargs(module_compiler): gen = CodeChunk() gen.write('def {method}(a, b):', method='sum') gen.indent() gen.write('return a + b') code = str(gen) m = module_compiler(code) assert m.sum(2, 3) == 5
def test_codegen_write_lines(module_compiler): lines = ['@staticmethod', 'def sum(a, b):' ' return a + b'] gen = CodeChunk() gen.write('class Math:') gen.indent() gen.write_lines(lines) code = str(gen) m = module_compiler(code) assert m.Math.sum(2, 3) == 5
def test_codegen_write_lines(module_compiler): lines = [ '@staticmethod', 'def sum(a, b):' ' return a + b' ] gen = CodeChunk() gen.write('class Math:') gen.indent() gen.write_lines(lines) code = str(gen) m = module_compiler(code) assert m.Math.sum(2, 3) == 5