Skip to content

Commit 3ad9e13

Browse files
committed
Test input coder roundtrip in cog test
1 parent 591291f commit 3ad9e13

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

python/coglet/runner.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,10 @@ def __init__(self, predictor: adt.Predictor):
2222

2323
async def test(self) -> Any:
2424
inputs = inspector.get_test_inputs(self.predictor, self.inputs)
25+
for k, v in inputs.items():
26+
tpe = self.inputs[k].type
27+
w = tpe.json_decode(tpe.json_encode(v))
28+
assert w == v, f'test input {k} does not encode properly'
2529
if self.is_iter():
2630
output = []
2731
async for x in self.predict_iter(inputs):

0 commit comments

Comments
 (0)