mirror of
https://github.com/deepseek-ai/smallpond
synced 2025-06-26 18:27:45 +00:00
init
This commit is contained in:
25
tests/test_filesystem.py
Normal file
25
tests/test_filesystem.py
Normal file
@@ -0,0 +1,25 @@
|
||||
import os.path
|
||||
import tempfile
|
||||
import threading
|
||||
import unittest
|
||||
|
||||
from smallpond.io.filesystem import dump, load
|
||||
from tests.test_fabric import TestFabric
|
||||
|
||||
|
||||
class TestFilesystem(TestFabric, unittest.TestCase):
|
||||
def test_pickle_runtime_ctx(self):
|
||||
with tempfile.TemporaryDirectory(dir=self.output_root_abspath) as output_dir:
|
||||
pickle_path = os.path.join(output_dir, "runtime_ctx.pickle")
|
||||
dump(self.runtime_ctx, pickle_path)
|
||||
runtime_ctx = load(pickle_path)
|
||||
self.assertEqual(self.runtime_ctx.job_id, runtime_ctx.job_id)
|
||||
|
||||
def test_pickle_trace(self):
|
||||
with self.assertRaises(TypeError) as context:
|
||||
with tempfile.TemporaryDirectory(
|
||||
dir=self.output_root_abspath
|
||||
) as output_dir:
|
||||
thread = threading.Thread()
|
||||
pickle_path = os.path.join(output_dir, "thread.pickle")
|
||||
dump(thread, pickle_path)
|
||||
Reference in New Issue
Block a user