def test_trivial(self):
import socket
sock_class = socket.socket
del socket
saved_class = save_class(sock_class)
restored_class = restore_class(saved_class)
self.assertEqual(restored_class, sock_class)
def test_deep(self):
from xml.sax.handler import ContentHandler
ch_class = ContentHandler
del ContentHandler
saved_class = save_class(ch_class)
restored_class = restore_class(saved_class)
self.assertEqual(restored_class, ch_class)