Traktor/myenv/Lib/site-packages/mpmath/tests/test_pickle.py

28 lines
401 B
Python
Raw Normal View History

2024-05-23 01:57:24 +02:00
import os
import tempfile
import pickle
from mpmath import *
def pickler(obj):
fn = tempfile.mktemp()
f = open(fn, 'wb')
pickle.dump(obj, f)
f.close()
f = open(fn, 'rb')
obj2 = pickle.load(f)
f.close()
os.remove(fn)
return obj2
def test_pickle():
obj = mpf('0.5')
assert obj == pickler(obj)
obj = mpc('0.5','0.2')
assert obj == pickler(obj)