def testFromCharDirRecursive(self):
directory = os.path.join(self.currdir, "data")
charcol = CharacterCollection.from_character_directory(directory,
check_duplicate=True)
self.assertEquals(charcol.get_set_list(), ["防", "三", "一", "二"])
self.assertEquals(len(charcol.get_characters("一")), 3)
self.assertEquals(len(charcol.get_characters("三")), 2)
self.assertEquals(len(charcol.get_characters("二")), 1)
self.assertEquals(len(charcol.get_characters("防")), 1)
def testFromCharDirRecursive(self):
directory = os.path.join(self.currdir, "data")
charcol = CharacterCollection.from_character_directory(
directory, check_duplicate=True)
self.assertEquals(sorted(charcol.get_set_list()),
sorted(["yo", "防", "三", "一", "二"]))
self.assertEquals(len(charcol.get_characters("一")), 3)
self.assertEquals(len(charcol.get_characters("三")), 2)
self.assertEquals(len(charcol.get_characters("二")), 1)
self.assertEquals(len(charcol.get_characters("防")), 1)
def _get_charcol(charcol_type, charcol_path):
if charcol_type == TYPE_DIRECTORY:
# charcol_path is actually a directory here
return CharacterCollection.from_character_directory(charcol_path)
elif charcol_type in (TYPE_CHARCOL, TYPE_CHARCOL_DB):
return CharacterCollection(charcol_path)
elif charcol_type == TYPE_TOMOE:
return tomoe_dict_to_character_collection(charcol_path)
elif charcol_type == TYPE_KUCHIBUE:
return kuchibue_to_character_collection(charcol_path)
def _get_charcol(charcol_type, charcol_path):
if charcol_type == TYPE_DIRECTORY:
# charcol_path is actually a directory here
return CharacterCollection.from_character_directory(charcol_path)
elif charcol_type in (TYPE_CHARCOL, TYPE_CHARCOL_DB):
return CharacterCollection(charcol_path)
elif charcol_type == TYPE_TOMOE:
return tomoe_dict_to_character_collection(charcol_path)
elif charcol_type == TYPE_KUCHIBUE:
return kuchibue_to_character_collection(charcol_path)
def testFromCharDirNotRecursive(self):
directory = os.path.join(self.currdir, "data")
charcol = CharacterCollection.from_character_directory(
directory, recursive=False, check_duplicate=True)
self.assertEquals(charcol.get_set_list(), ["防"])
self.assertEquals(len(charcol.get_characters("防")), 1)
