def test_filehandler_read_write_json(self): from pewtils.io import FileHandler h = FileHandler("tests/files", use_s3=False) h.write("temp", self.test_json, format="json") read = h.read("temp", format="json") import os os.unlink("tests/files/temp.json") self.assertEqual(repr(self.test_json), repr(dict(read)))
def test_filehandler_read_write_txt(self): from pewtils.io import FileHandler h = FileHandler("tests/files", use_s3=False) h.write("temp", "test", format="txt") read = h.read("temp", format="txt") import os os.unlink("tests/files/temp.txt") self.assertEqual(read, "test")
def test_filehandler_read_write_pkl(self): from pewtils.io import FileHandler h = FileHandler("tests/files", use_s3=False) h.write("temp", self.test_df, format="pkl") read = h.read("temp", format="pkl") import os os.unlink("tests/files/temp.pkl") self.assertEqual(repr(self.test_df), repr(read))
def test_filehandler_read_write_tab(self): from pewtils.io import FileHandler h = FileHandler("tests/files", use_s3=False) h.write("temp", self.test_df, format="tab") read = h.read("temp", format="tab") del read["Unnamed: 0"] import os os.unlink("tests/files/temp.tab") self.assertEqual(repr(self.test_df), repr(read))