def test_load(self):
storage_factory = StorageFactory()
file_store_config = FileStorageConfiguration()
file_store_config._gender_storage = FileStoreConfiguration(
file=os.path.dirname(__file__) + os.sep + "test_files" + os.sep +
"gender.txt",
fileformat="text",
extension="txt",
encoding="utf-8",
delete_on_start=False)
storage_engine = FileStorageEngine(file_store_config)
storage_factory._storage_engines[
StorageFactory.GENDER] = storage_engine
storage_factory._store_to_engine_map[
StorageFactory.GENDER] = storage_engine
collection = GenderCollection()
self.assertIsNotNone(collection)
self.assertTrue(collection.load(storage_factory))
self.assertEqual(collection.gender(" WITH HIM "), [
re.compile('(^WITH HIM | WITH HIM | WITH HIM$)', re.IGNORECASE),
' WITH HER '
])
self.assertEqual(collection.genderise_string("This is with him "),
"This is with her")
def test_reload_jp(self):
storage_factory = StorageFactory()
tokenizer = TokenizerJP()
file_store_config = FileStorageConfiguration()
file_store_config._gender_storage = FileStoreConfiguration(
file=os.path.dirname(__file__) + os.sep + "test_files" + os.sep +
"gender_jp.txt",
format="text",
extension="txt",
encoding="utf-8",
delete_on_start=False)
storage_engine = FileStorageEngine(file_store_config)
storage_factory._storage_engines[
StorageFactory.GENDER] = storage_engine
storage_factory._store_to_engine_map[
StorageFactory.GENDER] = storage_engine
collection = GenderCollection()
self.assertIsNotNone(collection)
collection.load(storage_factory)
self.assertEqual(collection.gender("彼"), '彼女')
self.assertEqual(collection.genderise_string(tokenizer, "彼が来た"),
"彼女が来た")
collection.reload(storage_factory)
self.assertEqual(collection.gender("彼"), '彼女')
self.assertEqual(collection.genderise_string(tokenizer, "彼が来た"),
"彼女が来た")
def test_load_from_file(self):
config = FileStorageConfiguration()
config._gender_storage = FileStoreConfiguration(file=os.path.dirname(__file__) + os.sep + "data" + os.sep + "lookups" + os.sep + "text" + os.sep + "gender.txt", fileformat="text", encoding="utf-8", delete_on_start=False)
engine = FileStorageEngine(config)
engine.initialise()
store = FileGenderStore(engine)
gender_collection = GenderCollection()
store.load(gender_collection)
self.assertEqual(gender_collection.gender(" WITH HIM "), [re.compile('(^WITH HIM | WITH HIM | WITH HIM$)', re.IGNORECASE), ' WITH HER '])
self.assertEqual(gender_collection.genderise_string("This is with him "), "This is with her")
def test_load_with_exception(self):
storage_factory = StorageFactory()
file_store_config = FileStorageConfiguration()
file_store_config._gender_storage = FileStoreConfiguration(
file=os.path.dirname(__file__) + os.sep + "test_files" + os.sep +
"gender.txt",
fileformat="text",
extension="txt",
encoding="utf-8",
delete_on_start=False)
storage_engine = FileStorageEngine(file_store_config)
storage_factory._storage_engines[
StorageFactory.GENDER] = storage_engine
storage_factory._store_to_engine_map[
StorageFactory.GENDER] = storage_engine
collection = GenderCollection()
self.assertIsNotNone(collection)
self.assertFalse(collection.load(storage_factory))
