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