| Server IP : 84.32.84.139 / Your IP : 216.73.217.69 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/self/root/opt/cloudlinux/venv/lib64/python3.11/site-packages/dill/tests/ |
Upload File : |
#!/usr/bin/env python
# Author: Leonardo Gama (@leogama)
# Copyright (c) 2022-2023 The Uncertainty Quantification Foundation.
# License: 3-clause BSD. The full license text is available at:
# - https://github.com/uqfoundation/dill/blob/master/LICENSE
import logging
import re
import tempfile
import dill
from dill import detect
from dill.logger import stderr_handler, adapter as logger
try:
from StringIO import StringIO
except ImportError:
from io import StringIO
test_obj = {'a': (1, 2), 'b': object(), 'f': lambda x: x**2, 'big': list(range(10))}
def test_logging(should_trace):
buffer = StringIO()
handler = logging.StreamHandler(buffer)
logger.addHandler(handler)
try:
dill.dumps(test_obj)
if should_trace:
regex = re.compile(r'(\S*┬ \w.*[^)]' # begin pickling object
r'|│*└ # \w.* \[\d+ (\wi)?B])' # object written (with size)
)
for line in buffer.getvalue().splitlines():
assert regex.fullmatch(line)
return buffer.getvalue()
else:
assert buffer.getvalue() == ""
finally:
logger.removeHandler(handler)
buffer.close()
def test_trace_to_file(stream_trace):
file = tempfile.NamedTemporaryFile(mode='r')
with detect.trace(file.name, mode='w'):
dill.dumps(test_obj)
file_trace = file.read()
file.close()
# Apparently, objects can change location in memory...
reghex = re.compile(r'0x[0-9A-Za-z]+')
file_trace, stream_trace = reghex.sub('0x', file_trace), reghex.sub('0x', stream_trace)
# PyPy prints dictionary contents with repr(dict)...
regdict = re.compile(r'(dict\.__repr__ of ).*')
file_trace, stream_trace = regdict.sub(r'\1{}>', file_trace), regdict.sub(r'\1{}>', stream_trace)
assert file_trace == stream_trace
if __name__ == '__main__':
logger.removeHandler(stderr_handler)
test_logging(should_trace=False)
detect.trace(True)
test_logging(should_trace=True)
detect.trace(False)
test_logging(should_trace=False)
loglevel = logging.ERROR
logger.setLevel(loglevel)
with detect.trace():
stream_trace = test_logging(should_trace=True)
test_logging(should_trace=False)
assert logger.getEffectiveLevel() == loglevel
test_trace_to_file(stream_trace)