(self)
| 191 | compile(code, "<test>", "exec") |
| 192 | |
| 193 | def test_csv_reader_codegen(self): |
| 194 | from docetl.frame import read_csv |
| 195 | frame = read_csv("data.csv").map(prompt="p") |
| 196 | code = frame.to_python() |
| 197 | assert "docetl.read_csv(" in code |
| 198 | compile(code, "<test>", "exec") |
| 199 | |
| 200 | def test_parquet_reader_codegen(self): |
| 201 | from docetl.frame import read_parquet |