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")