ner-study/dataset.test.py
2022-02-14 08:28:54 +09:00

9 lines
231 B
Python

from dataset import padding_array
import unittest
class Test(unittest.TestCase):
def test_padding_array(self):
self.assertEquals(padding_array([[1,2],[3]]),[[1,2],[3,0]])
if __name__ == '__main__':
unittest.main()