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')