(self)
| 137 | self.assertTrue(all(t == 5 for _, t in calls)) |
| 138 | |
| 139 | async def test_cfg_forwarded(self) -> None: |
| 140 | seen_cfg: list[Any] = [] |
| 141 | |
| 142 | async def flow(input: RawText, *, cfg: Any = None) -> str: |
| 143 | seen_cfg.append(cfg) |
| 144 | return "ok" |
| 145 | |
| 146 | cfg = PaperFlowCfg(model="sentinel-model") |
| 147 | await batch_run(flow, [RawText(text="x")], cfg=cfg, concurrency=1) |
| 148 | self.assertIs(seen_cfg[0], cfg) |
| 149 | |
| 150 | async def test_extra_kwargs_forwarded(self) -> None: |
| 151 | seen: list[Any] = [] |
nothing calls this directly
no test coverage detected