sympycore/tests/test_pickle.py

   1
   2 import os
   3 import tempfile
   4 import pickle
   5
   6 from sympycore.core import Expr
   7 from sympycore.utils import SYMBOL
   8
   9 class MyExpr(Expr):
  10     # see comment in sympycore/core.py for pickling pure Expr
  11     # instances.
  12     pass
  13
  14 def test_pickle():
  15
  16     fn = tempfile.mktemp()
  17     obj = MyExpr(SYMBOL, 'x')
  18
  19     f = open(fn, 'wb')
  20     pickle.dump(obj, f)
  21     f.close()
  22
  23     f = open(fn, 'rb')
  24     obj2 = pickle.load(f)
  25     f.close()
  26     os.remove(fn)
  27
  28     assert obj.head == obj2.head,`obj.head, obj2.head`
  29     assert obj.head is obj2.head,`obj.head, obj2.head, id(obj.head), id(obj2.head)`
  30     assert obj.data == obj2.data,`obj.data, obj2.data`
  31     assert obj.pair == obj2.pair,`obj.pair, obj2.pair`
  32     assert obj==obj2,`obj,obj2`
  33