def test_write_file(tmpdir): fname, content = 'test1.bin', b'TorchVision\211\n' fpath = os.path.join(tmpdir, fname) content_tensor = torch.tensor(list(content), dtype=torch.uint8) write_file(fpath, content_tensor) with open(fpath, 'rb') as f: saved_content = f.read() os.unlink(fpath) assert content == saved_content
def test_write_file_non_ascii(tmpdir): fname, content = "日本語(Japanese).bin", b"TorchVision\211\n" fpath = os.path.join(tmpdir, fname) content_tensor = torch.tensor(list(content), dtype=torch.uint8) write_file(fpath, content_tensor) with open(fpath, "rb") as f: saved_content = f.read() os.unlink(fpath) assert content == saved_content
def test_write_file_non_ascii(self): with get_tmp_dir() as d: fname, content = '日本語(Japanese).bin', b'TorchVision\211\n' fpath = os.path.join(d, fname) content_tensor = torch.tensor(list(content), dtype=torch.uint8) write_file(fpath, content_tensor) with open(fpath, 'rb') as f: saved_content = f.read() self.assertEqual(content, saved_content) os.unlink(fpath)