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_load_from_disk(self):
# test for dict with warning:
# save a file to disk to test it:
inp = {"wuwu": {"warning": ["warning1", "warning2"]}}
out = {"wuwu": {"warning": {"warning1", "warning2"}}}
with open("test.gdn", "w") as test_file:
test_file.write(json.dumps(inp))
# test if it loads as expected:
self.assertEqual(GenderNounDataHandler.load_from_disk("test.gdn"), out)
# test for dict without warning:
# save a file to disk to test it:
inp = {"wuwu": {"fufu": "wawa"}}
out = inp
with open("test.gdn", "w") as test_file:
test_file.write(json.dumps(inp))
# test if it loads directly:
self.assertEqual(GenderNounDataHandler.load_from_disk("test.gdn"), out)
# test for a mixture:
# save a file to disk to test it:
inp = {"wuwu": {"warning": ["warning1", "warning2"]}, "wawa": {"fufu": "wawa"}}
out = {"wuwu": {"warning": {"warning1", "warning2"}}, "wawa": {"fufu": "wawa"}}
with open("test.gdn", "w") as test_file:
test_file.write(json.dumps(inp))
# test if it loads as expected:
self.assertEqual(GenderNounDataHandler.load_from_disk("test.gdn"), out)
# finally delete the file:
os.remove("test.gdn")