class TestPyDicManager(unittest.TestCase): def __init__(self, methodName='runTest'): self.current_dir = os.path.dirname(os.path.realpath(__file__)) self.temp_dict1_path = tempfile.mkdtemp() self.temp_dict2_path = tempfile.mkdtemp() self.dict1_file = open(os.path.join(self.current_dir, 'dict1.txt')) self.dict2_file = open(os.path.join(self.current_dir, 'dict2.txt')) PyDicCreator().generate(self.dict1_file, self.temp_dict1_path, 'dict1', verbose=False) PyDicCreator().generate(self.dict2_file, self.temp_dict2_path, 'dict2', verbose=False) return super(TestPyDicManager, self).__init__(methodName) def setUp(self): self.dict = PyDicManager(self.temp_dict1_path, self.temp_dict2_path) def test_dictionaries(self): self.assertEquals(set(self.dict.get_dictionaries()), set(['dict1', 'dict2'])) def test_id(self): self.assertEquals(set(self.dict.id(u'kotem')), set(['1@dict1', '6@dict2'])) self.assertEquals(set(self.dict.id(u'płaszcz')), set(['13@dict1', '9@dict2'])) self.assertEquals(self.dict.id(u'zatelegrafowałobym'), ['5@dict2']) self.assertEquals(self.dict.id(u'zatelegrafowałobymmmmm'), []) def test_id_forms(self): self.assertEquals(self.dict.id_forms('6@dict2'), self.dict.id_forms('1@dict1')) self.assertEquals(self.dict.id_forms('4@dict1'), [u"pszczoła", u"pszczoły", u"pszczole", u"pszczołę", u"pszczołą", u"pszczoło", u"pszczół", u"pszczołom", u"pszczołami", u"pszczołach", ]) self.assertEquals(self.dict.id_forms('8@dict2'), [u"spodnie", u"spodni", u"spodniom", u"spodniami", u"spodniach", ]) self.assertEquals(self.dict.id_forms('88888888@dict2'), []) def test_id_base(self): self.assertEquals(self.dict.id_base('6@dict2'), self.dict.id_base('1@dict1')) self.assertEquals(self.dict.id_base('4@dict1'), u"pszczoła") self.assertEquals(self.dict.id_base('8@dict2'), u"spodnie") self.assertEquals(self.dict.id_base('88888888@dict2'), None) def test_word_forms(self): self.assertEquals(self.dict.word_forms(u'psem'), [(u"pies", u"psa", u"psu", u"psem", u"psie", u"psy", u"psów", u"psom", u"psami", u"psach",)]) def test_word_base(self): self.assertEquals(self.dict.word_base(u'psem'), [u"pies"]) self.assertEquals(self.dict.word_base(u'pilotem'), [u"pilot"]) self.assertEquals(self.dict.word_base(u'pilotemmmmmm'), []) self.assertEquals(set(self.dict.word_base(u'płaszcz')), set([u"płaszczyć", u"płaszcz"])) def test_a_id(self): self.assertEquals(self.dict.a_id(u'pszczola'),[u"4@dict1"]) self.assertEquals(self.dict.a_id(u'psow'), [u"2@dict1",u"7@dict2"]) self.assertEquals(self.dict.a_id(u'psów'), [u"2@dict1",u"7@dict2"]) self.assertEquals(self.dict.a_id(u'pśów'), []) def test_a_word_forms(self): self.assertEquals(self.dict.a_word_forms(u'psow'), [(u"pies", u"psa", u"psu", u"psem", u"psie", u"psy", u"psów", u"psom", u"psami", u"psach",)]) self.assertEquals(self.dict.a_word_forms(u'pszczol'), [(u"pszczoła", u"pszczoły", u"pszczole", u"pszczołę", u"pszczołą", u"pszczoło", u"pszczół", u"pszczołom", u"pszczołami", u"pszczołach",)]) self.assertEquals(self.dict.a_word_forms(u'pszczól'), [(u"pszczoła", u"pszczoły", u"pszczole", u"pszczołę", u"pszczołą", u"pszczoło", u"pszczół", u"pszczołom", u"pszczołami", u"pszczołach",)]) self.assertEquals(self.dict.a_word_forms(u'pszcżól'), []) def test_a_word_base(self): self.assertEquals(self.dict.a_word_base(u'pszczol'),[u"pszczoła"]) self.assertEquals(self.dict.a_word_base(u'autopilotow'),[u"autopilot"]) self.assertEquals(self.dict.a_word_base(u'autopilotów'),[u"autopilot"]) self.assertEquals(self.dict.a_word_base(u'autopiłotow'),[]) def test_pydic_id_formats(self): self.assertRaises(ValueError, self.dict.id_forms, 88888888)
def setUp(self): self.dict = PyDicManager(self.temp_dict1_path, self.temp_dict2_path)