修复 test

经过 pipeline rework 之后，之前的一些 test 代码需要修复以适配新的数据格式

我在 !104 (merged) 有一些未完成的工作，可以参考。