1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
import rpyc
import gc
import unittest
class TestRefcount(unittest.TestCase):
def setUp(self):
self.conn = rpyc.classic.connect_thread()
def tearDown(self):
self.conn.close()
def test_refcount(self):
self.conn.execute("""
deleted_objects = []
class DummyObject(object):
def __init__(self, name):
self.name = name
def __del__(self):
deleted_objects.append(self.name)""")
rDummyObject = self.conn.namespace["DummyObject"]
d1 = rDummyObject("d1")
d2 = rDummyObject("d2")
d3 = rDummyObject("d3")
d4 = rDummyObject("d4") # noqa
d2_copy = d2
del d1
del d3
gc.collect()
self.assertEqual(set(self.conn.namespace["deleted_objects"]), set(["d1", "d3"]))
del d2
gc.collect()
self.assertEqual(set(self.conn.namespace["deleted_objects"]), set(["d1", "d3"]))
del d2_copy
gc.collect()
self.assertEqual(set(self.conn.namespace["deleted_objects"]), set(["d1", "d2", "d3"]))
if __name__ == "__main__":
unittest.main()
|
