def test_cp15803(self):
        _testdir = 'cp15803'
        pickle_file = os.path.join(self.test_dir,
                                   "cp15803_%d.pickle" % os.getpid())

        module_txt = """
class K%s(object):
    static_member = 1
    def __init__(self):
        self.member = 2

%s = K%s()
"""

        _testdir_init = os.path.join(self.test_dir, _testdir, '__init__.py')
        self.write_to_file(_testdir_init, module_txt % ("", "FROM_INIT", ""))

        _testdir_mod = os.path.join(self.test_dir, _testdir, 'mod.py')
        self.write_to_file(
            _testdir_mod, module_txt % ("Mod", "FROM_MOD", "Mod") + """
from cp15803 import K
FROM_INIT_IN_MOD = K()
""")

        _testdir_sub = os.path.join(self.test_dir, _testdir, 'sub')
        _testdir_sub_init = os.path.join(_testdir_sub, '__init__.py')
        self.write_to_file(_testdir_sub_init)

        _testdir_sub_submod = os.path.join(_testdir_sub, 'submod.py')
        self.write_to_file(
            _testdir_sub_submod,
            module_txt % ("SubMod", "FROM_SUB_MOD", "SubMod") + """
from cp15803 import mod
FROM_MOD_IN_SUBMOD = mod.KMod()
""")

        try:
            with path_modifier(self.test_dir) as p:
                import cp15803
                import cp15803.mod
                import cp15803.sub.submod
                import cp15803.sub.submod as newname

                for x in [
                        cp15803.K(),
                        cp15803.FROM_INIT,
                        cp15803.mod.KMod(),
                        cp15803.mod.FROM_MOD,
                        cp15803.mod.K(),
                        cp15803.mod.FROM_INIT_IN_MOD,
                        cp15803.sub.submod.KSubMod(),
                        cp15803.sub.submod.FROM_SUB_MOD,
                        cp15803.sub.submod.FROM_MOD_IN_SUBMOD,
                        cp15803.sub.submod.mod.KMod(),
                        cp15803.sub.submod.mod.FROM_MOD,
                        cp15803.sub.submod.mod.K(),
                        cp15803.sub.submod.mod.FROM_INIT_IN_MOD,
                        newname.KSubMod(),
                        newname.FROM_SUB_MOD,
                        newname.FROM_MOD_IN_SUBMOD,
                        newname.mod.KMod(),
                        newname.mod.FROM_MOD,
                        newname.mod.K(),
                        newname.mod.FROM_INIT_IN_MOD,
                ]:
                    with open(pickle_file, "wb") as f:
                        cPickle.dump(x, f)

                    with open(pickle_file, "rb") as f:
                        x_unpickled = cPickle.load(f)

                        self.assertEqual(x.__class__.__name__,
                                         x_unpickled.__class__.__name__)
                        self.assertEqual(x.static_member,
                                         x_unpickled.static_member)
                        self.assertEqual(x.member, x_unpickled.member)

        finally:
            try:
                os.unlink(pickle_file)
                self.clean_directory(os.path.join(self.test_dir, _testdir))
            except:
                pass
Ejemplo n.º 2
0
def test_cp15803():
    '''
    '''
    _testdir = 'cp15803'
    pickle_file = path_combine(testpath.public_testdir, "cp15803.pickle")
    
    module_txt = """
class K%s(object):
    static_member = 1
    def __init__(self):
        self.member = 2
        
%s = K%s()
"""
    
    _testdir_init       = path_combine(testpath.public_testdir, _testdir, '__init__.py')
    write_to_file(_testdir_init, module_txt % ("", "FROM_INIT", ""))
    
    _testdir_mod        = path_combine(testpath.public_testdir, _testdir, 'mod.py')
    write_to_file(_testdir_mod, module_txt % ("Mod", "FROM_MOD", "Mod") + """
from cp15803 import K
FROM_INIT_IN_MOD = K()
"""
    )
    
    _testdir_sub        = path_combine(testpath.public_testdir, _testdir, 'sub')
    _testdir_sub_init   = path_combine(_testdir_sub, '__init__.py')
    write_to_file(_testdir_sub_init)
    
    _testdir_sub_submod = path_combine(_testdir_sub, 'submod.py')
    write_to_file(_testdir_sub_submod, module_txt % ("SubMod", "FROM_SUB_MOD", "SubMod") + """
from cp15803 import mod
FROM_MOD_IN_SUBMOD = mod.KMod()
"""
    )
   
    import cp15803
    import cp15803.mod
    import cp15803.sub.submod
    import cp15803.sub.submod as newname
    
    try:
        for x in [
                    cp15803.K(), cp15803.FROM_INIT,
                    cp15803.mod.KMod(), cp15803.mod.FROM_MOD, cp15803.mod.K(), cp15803.mod.FROM_INIT_IN_MOD,
                    cp15803.sub.submod.KSubMod(), cp15803.sub.submod.FROM_SUB_MOD, cp15803.sub.submod.FROM_MOD_IN_SUBMOD,
                    cp15803.sub.submod.mod.KMod(), cp15803.sub.submod.mod.FROM_MOD, cp15803.sub.submod.mod.K(), cp15803.sub.submod.mod.FROM_INIT_IN_MOD,
                        
                    newname.KSubMod(), newname.FROM_SUB_MOD, newname.FROM_MOD_IN_SUBMOD,
                    newname.mod.KMod(), newname.mod.FROM_MOD, newname.mod.K(), newname.mod.FROM_INIT_IN_MOD,
                    ]:
            with open(pickle_file, "w") as f:
                cPickle.dump(x, f)
            
            with open(pickle_file, "r") as f:
                x_unpickled = cPickle.load(f)
            
                AreEqual(x.__class__.__name__, x_unpickled.__class__.__name__)
                AreEqual(x.static_member, x_unpickled.static_member)
                AreEqual(x.member, x_unpickled.member)
                
    finally:
        import nt
        try:
            nt.unlink(pickle_file)
            for f_name in [ _testdir_init, _testdir_mod,
                            _testdir_sub_init, _testdir_sub_submod,
                            ]:
                nt.unlink(f_name)
                if sys.platform=="win32":
                    nt.unlink(f_name + "c")
            
            
            for dir_name in [ _testdir_sub, _testdir]:
                nt.rmdir(dir_name)
        except:
            pass