mirror of
https://gitee.com/fastnlp/fastNLP.git
synced 2024-12-05 05:38:31 +08:00
修改Padder的测试用例
This commit is contained in:
parent
8091a734ee
commit
3e33a23042
@ -118,11 +118,12 @@ class TestPadder(unittest.TestCase):
|
||||
self.assertListEqual([[1,2], [3, 0], [4, 0]],
|
||||
padder(content, None, np.int64).tolist())
|
||||
|
||||
contents = [
|
||||
content = [
|
||||
[[1, 2, 3], [4, 5], [7,8,9,10]],
|
||||
[[1]]
|
||||
]
|
||||
print(padder(contents, None, np.int64))
|
||||
self.assertListEqual(content,
|
||||
padder(content, None, np.int64).tolist())
|
||||
|
||||
def test02(self):
|
||||
"""
|
||||
|
Loading…
Reference in New Issue
Block a user