def test_exhausted_gen():
def gfunc():
yield 5
g = gfunc()
for i in g:
print i
g1 = statesaver.copy(g)
assert iter(g1) is g1
py.test.raises(StopIteration, g1.next)
g2 = statesaver.copy(g1)
assert iter(g2) is g2
py.test.raises(StopIteration, g2.next)
def test_stringiter():
s = "hello"
it = iter(s)
assert it.next() == 'h'
assert it.next() == 'e'
it1 = statesaver.copy(it)
assert list(it) == ['l', 'l', 'o']
assert list(it1) == ['l', 'l', 'o']
def test_tupleiter():
tup = (2, 4, 6, 8)
it = iter(tup)
assert it.next() == 2
assert it.next() == 4
it1 = statesaver.copy(it)
assert list(it) == [6, 8]
assert list(it1) == [6, 8]
def test_listiter():
lst = [2, 4, 6, 8]
it = iter(lst)
assert it.next() == 2
assert it.next() == 4
it1 = statesaver.copy(it)
lst.append(10)
assert list(it) == [6, 8, 10]
assert list(it1) == [6, 8]
def test_seqiter():
from UserList import UserList
seq = UserList([2, 4, 6, 8])
it = iter(seq)
assert it.next() == 2
assert it.next() == 4
it1 = statesaver.copy(it)
assert list(it) == [6, 8]
assert list(it1) == [6, 8]
def test_generator():
def gfunc():
lst = [5, 6]
yield lst.pop()
yield lst.pop()
g = gfunc()
assert g.next() == 6
g1 = statesaver.copy(g)
assert g.next() == 5
py.test.raises(StopIteration, g.next)
assert g1.next() == 5
py.test.raises(StopIteration, g1.next)
def test_instance():
class Foo:
def inst_build(self):
return Bar()
class Bar:
pass
x = Foo()
x.attr = [1, 2, 3]
y = statesaver.copy(x)
assert y.__class__ is Bar
assert y.attr == [1, 2, 3]
assert y.attr is not x.attr
def test_method():
def func(x, y=[]):
assert glob == 2
y.append(x)
return y
m = new.instancemethod(func, {})
assert m() == [{}]
m1 = statesaver.copy(m)
assert m() == [{}, {}]
assert m() == [{}, {}, {}]
assert m1() == [{}, {}]
assert m1() == [{}, {}, {}]
l = m1()
assert l[0] is l[1] is l[2] is l[3]
def test_function():
# XXX closures not supported
def func(x, y=[]):
assert glob == 2
y.append(x)
return y
l = func(5)
l = func(6)
assert l == [5, 6]
func1 = statesaver.copy(func)
l = func(7)
l = func(8)
assert l == [5, 6, 7, 8]
l = func1(9)
l = func1(10)
assert l == [5, 6, 9, 10]
def test_list():
lst = [None, 12, "hello", 3.4, ("foo", (), [])]
lst1 = statesaver.copy(lst)
assert lst1 == lst
assert lst1 is not lst
assert lst1[-1][-1] is not lst[-1][-1]
def test_dict():
dct = {1: "hi", 2: {}}
dct1 = statesaver.copy(dct)
assert dct1 == dct
assert dct1 is not dct
assert dct1[2] is not dct[2]
