diff options
Diffstat (limited to 'tests/test_nsjail.py')
| -rw-r--r-- | tests/test_nsjail.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_nsjail.py b/tests/test_nsjail.py index ccbca56..839d3ec 100644 --- a/tests/test_nsjail.py +++ b/tests/test_nsjail.py @@ -26,7 +26,7 @@ class NsJailTests(unittest.TestCase): return self.nsjail.python3(["-c", code]) def eval_file(self, code: str, name: str = "test.py", **kwargs): - file = FileAttachment(name, code) + file = FileAttachment(name, code.encode()) return self.nsjail.python3([name], [file], **kwargs) def test_print_returns_0(self): @@ -59,8 +59,8 @@ class NsJailTests(unittest.TestCase): def test_multi_files(self): files = [ - FileAttachment("main.py", "import lib; print(lib.x)"), - FileAttachment("lib.py", "x = 'hello'"), + FileAttachment("main.py", "import lib; print(lib.x)".encode()), + FileAttachment("lib.py", "x = 'hello'".encode()), ] result = self.nsjail.python3(["main.py"], files) @@ -209,7 +209,7 @@ class NsJailTests(unittest.TestCase): def test_file_write_error(self): """Test errors during file write.""" - result = self.nsjail.python3([""], [FileAttachment("output", "hello")]) + result = self.nsjail.python3([""], [FileAttachment("output", "hello".encode())]) self.assertEqual(result.returncode, None) self.assertEqual(result.stdout, "IsADirectoryError: Failed to create file 'output'.") |