Add unit test case for no tokens to be erased

add_depthwiseConv_op_gpu
Yibing Liu 7 years ago
parent d1d614b9f8
commit 5ae0c97faf

@ -44,5 +44,20 @@ class TestSequenceEraseOp(OpTest):
self.check_output()
class TestSequenceEraseOpEmpty(OpTest):
def setUp(self):
self.op_type = "sequence_erase"
in_seq = np.random.randint(0, 10, (30, 1)).astype("int32")
lod = [[0, 9, 13, 24, 30]]
tokens = []
out_seq, new_lod0 = sequence_erase(in_seq, lod[0], tokens)
self.attrs = {'tokens': tokens}
self.inputs = {'X': (in_seq, lod)}
self.outputs = {'Out': (out_seq, [new_lod0])}
def test_check_output(self):
self.check_output()
if __name__ == '__main__':
unittest.main()

Loading…
Cancel
Save