Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 542bdef7 编写于 作者: F fengjiayi
浏览文件
操作

fix a unit test

上级 0bb9c80e
显示空白变更内容
内联 并排
Showing with 0 addition and 3 deletion
python/paddle/v2/fluid/tests/test_protobuf_descs.py
浏览文件 @ 542bdef7
...@@ -120,7 +120,6 @@ class TestVarDesc(unittest.TestCase): ...@@ -120,7 +120,6 @@ class TestVarDesc(unittest.TestCase):
block = program_desc.block(0) block = program_desc.block(0)
var = block.var('my_reader') var = block.var('my_reader')
var.set_type(core.VarDesc.VarType.READER) var.set_type(core.VarDesc.VarType.READER)
var.set_tensor_num(3)
src_shapes = [[2, 3, 3], [4, 5], [6, 7, 8, 9]] src_shapes = [[2, 3, 3], [4, 5], [6, 7, 8, 9]]
var.set_shapes(src_shapes) var.set_shapes(src_shapes)
res_shapes = var.shapes() res_shapes = var.shapes()
...@@ -141,7 +140,6 @@ class TestVarDesc(unittest.TestCase): ...@@ -141,7 +140,6 @@ class TestVarDesc(unittest.TestCase):
block = program_desc.block(0) block = program_desc.block(0)
var = block.var('my_reader') var = block.var('my_reader')
var.set_type(core.VarDesc.VarType.READER) var.set_type(core.VarDesc.VarType.READER)
var.set_tensor_num(3)
src_types = [ src_types = [
core.DataType.INT32, core.DataType.FP64, core.DataType.FP32 core.DataType.INT32, core.DataType.FP64, core.DataType.FP32
] ]
...@@ -154,7 +152,6 @@ class TestVarDesc(unittest.TestCase): ...@@ -154,7 +152,6 @@ class TestVarDesc(unittest.TestCase):
block = program_desc.block(0) block = program_desc.block(0)
var = block.var('my_reader') var = block.var('my_reader')
var.set_type(core.VarDesc.VarType.READER) var.set_type(core.VarDesc.VarType.READER)
var.set_tensor_num(3)
src_types = [3, 1, 2] src_types = [3, 1, 2]
var.set_lod_levels(src_types) var.set_lod_levels(src_types)
self.assertEqual(src_types, var.lod_levels()) self.assertEqual(src_types, var.lod_levels())
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或