def test__set_to_vec(self):
morpheme_loader = MorphemesLoader()
morpheme_loader.max_pos_id=2
self.assertEqual(list(morpheme_loader._set_to_vec({0,1})),[1,1])
self.assertEqual(list(morpheme_loader._set_to_vec({})),[0,0])
self.assertEqual(list(morpheme_loader._set_to_vec({0})),[1,0])
morpheme_loader.max_pos_id = 3
self.assertEqual(len(morpheme_loader._set_to_vec({})), 3)
def test__set_to_vec_power_set(self):
morpheme_loader = MorphemesLoader()
morpheme_loader.use_power_set = True
morpheme_loader.max_morpheme_count=2
self.assertEqual(morpheme_loader._set_to_vec({0,1}),[0])
self.assertEqual(morpheme_loader._set_to_vec({}),[1])
self.assertEqual(morpheme_loader._set_to_vec({0}),[2])
self.assertEqual(morpheme_loader.max_power_set_key,3)
self.assertEqual(morpheme_loader._set_to_vec({0, 1}), [0])
self.assertEqual(morpheme_loader.max_power_set_key, 3)