(self, sample_input)
| 149 | compile(code, "<test>", "exec") |
| 150 | |
| 151 | def test_multiline_prompt(self, sample_input): |
| 152 | frame = ( |
| 153 | read_json(sample_input) |
| 154 | .map("op", prompt="Line 1\nLine 2\nLine 3", |
| 155 | output={"schema": {"x": "string"}}) |
| 156 | ) |
| 157 | code = frame.to_python() |
| 158 | assert '"""' in code |
| 159 | compile(code, "<test>", "exec") |
| 160 | |
| 161 | def test_multi_op_step_codegen(self, tmp_dir): |
| 162 | config_path = os.path.join(tmp_dir, "multi.yaml") |