def test_save_and_load(self):
# test for dict with warning:
# save a file to disk to test it:
data = {"wuwu": {"warning": {"warning1", "warning2"}}}
GenderNounDataHandler.save_to_disk(data, "test.gdn")
self.assertEqual(data, GenderNounDataHandler.load_from_disk("test.gdn"))
# test for dict without warning:
# save a file to disk to test it:
data = {"wuwu": {"fufu": "wawa"}}
GenderNounDataHandler.save_to_disk(data, "test.gdn")
self.assertEqual(data, GenderNounDataHandler.load_from_disk("test.gdn"))
# test for words with underscores:
data = {"fu_fu": {"wawa": "wuwu"}}
GenderNounDataHandler.save_to_disk(data, "test.gdn")
self.assertEqual(data, GenderNounDataHandler.load_from_disk("test.gdn"))
# test for a mixture:
# save a file to disk to test it:
data = {"wuwu": {"warning": {"warning1", "warning2"}}, "wa_wa": {"fufu": "wawa"}}
GenderNounDataHandler.save_to_disk(data, "test.gdn")
self.assertEqual(data, GenderNounDataHandler.load_from_disk("test.gdn"))
# finally delete the file:
os.remove("test.gdn")
def test_save_to_disk(self):
# test for dict with warning:
# save a file to disk to test it:
data_in_file = {"wuwu": {"warning": ["warning1", "warning2"]}}
data_loaded = {"wuwu": {"warning": {"warning1", "warning2"}}}
GenderNounDataHandler.save_to_disk(data_loaded, "test.gdn")
with open("test.gdn", "r") as test_file:
self.assertEqual(json.loads(test_file.read()), data_in_file)
# test for dict without warning:
# save a file to disk to test it:
data_in_file = {"wuwu": {"fufu": "wawa"}}
data_loaded = data_in_file
GenderNounDataHandler.save_to_disk(data_loaded, "test.gdn")
with open("test.gdn", "r") as test_file:
self.assertEqual(json.loads(test_file.read()), data_in_file)
# test for a mixture:
# save a file to disk to test it:
data_in_file = {"wuwu": {"warning": ["warning1", "warning2"]}, "wawa": {"fufu": "wawa"}}
data_loaded = {"wuwu": {"warning": {"warning1", "warning2"}}, "wawa": {"fufu": "wawa"}}
GenderNounDataHandler.save_to_disk(data_loaded, "test.gdn")
with open("test.gdn", "r") as test_file:
self.assertEqual(json.loads(test_file.read()), data_in_file)
# finally delete the file:
os.remove("test.gdn")