def test_add_get(self):
Corpus.create('/tmp/TEST_CORPUS')
c = Corpus('/tmp/TEST_CORPUS')
c.add(u'Gżegżółką jaźń', 1, p1=1, p2="2", p3=[1, 2, 3, u'ą'])
c.add(u'Chrząszcz brzmi w czcinie',
2,
p1=1,
p2="2",
p3=[1, 2, 3, u'ą'])
c.add(u'Żółte źrebie', 3, p1=1, p2="2", p3=[1, 2, 3, u'ą'])
c.save_indexes()
d = Corpus('/tmp/TEST_CORPUS')
self.assertEqual(d.get(3), ({
'p1': 1,
'p2': "2",
'p3': [1, 2, 3, u'ą'],
'id': 3
}, u'Żółte źrebie'))
self.assertEqual(d.get(1), ({
'p1': 1,
'p2': "2",
'p3': [1, 2, 3, u'ą'],
'id': 1
}, u'Gżegżółką jaźń'))
self.assertEqual(d.get(2), ({
'p1': 1,
'p2': "2",
'p3': [1, 2, 3, u'ą'],
'id': 2
}, u'Chrząszcz brzmi w czcinie'))
del c, d
shutil.rmtree('/tmp/TEST_CORPUS')
def test_add_get(self):
Corpus.create('/tmp/TEST_CORPUS')
c = Corpus('/tmp/TEST_CORPUS')
c.add(u'Gżegżółką jaźń', 1, p1=1, p2="2", p3=[1,2,3,u'ą'])
c.add(u'Chrząszcz brzmi w czcinie', 2, p1=1, p2="2", p3=[1,2,3,u'ą'])
c.add(u'Żółte źrebie', 3, p1=1, p2="2", p3=[1,2,3,u'ą'])
c.save_indexes()
d = Corpus('/tmp/TEST_CORPUS')
self.assertEqual(d.get(3), ( { 'p1':1, 'p2':"2", 'p3':[1,2,3,u'ą'], 'id':3}, u'Żółte źrebie' ) )
self.assertEqual(d.get(1), ( { 'p1':1, 'p2':"2", 'p3':[1,2,3,u'ą'], 'id':1}, u'Gżegżółką jaźń' ) )
self.assertEqual(d.get(2), ( { 'p1':1, 'p2':"2", 'p3':[1,2,3,u'ą'], 'id':2}, u'Chrząszcz brzmi w czcinie' ) )
del c, d
shutil.rmtree('/tmp/TEST_CORPUS')
def test_iter(self):
Corpus.create('/tmp/TEST_CORPUS')
c = Corpus('/tmp/TEST_CORPUS')
c.add(u'Gżegżółką jaźń', 3, p1=1, p2="2", p3=[1,2,3,u'ą'])
c.add(u'Chrząszcz brzmi w czcinie', 1, p1=1, p2="2", p3=[1,2,3,u'ą'])
c.add(u'Żółte źrebie', 2, p1=1, p2="2", p3=[1,2,3,u'ą'])
c.save_indexes()
d = Corpus('/tmp/TEST_CORPUS')
l = []
for t in d:
l.append(t[0]['id'])
self.assertEqual(l, [3,1,2])
del c, d
shutil.rmtree('/tmp/TEST_CORPUS')
def test_iter(self):
Corpus.create('/tmp/TEST_CORPUS')
c = Corpus('/tmp/TEST_CORPUS')
c.add(u'Gżegżółką jaźń', 3, p1=1, p2="2", p3=[1, 2, 3, u'ą'])
c.add(u'Chrząszcz brzmi w czcinie',
1,
p1=1,
p2="2",
p3=[1, 2, 3, u'ą'])
c.add(u'Żółte źrebie', 2, p1=1, p2="2", p3=[1, 2, 3, u'ą'])
c.save_indexes()
d = Corpus('/tmp/TEST_CORPUS')
l = []
for t in d:
l.append(t[0]['id'])
self.assertEqual(l, [3, 1, 2])
del c, d
shutil.rmtree('/tmp/TEST_CORPUS')
