mirror of
https://github.com/deepseek-ai/smallpond
synced 2025-06-26 18:27:45 +00:00
24 lines
908 B
Python
24 lines
908 B
Python
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)
|