blob: 17551bb864ee8a3c415b31f946571c9d19fed300 (
plain)
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
42
43
44
45
|
import os
import sys
import shutil
import unittest
orig_path = os.path.join(os.path.dirname(__file__))
workdir = os.getcwd()
src = os.path.join(orig_path, 'test_module_template.py')
dst = os.path.join(workdir, 'test_module.py')
shutil.copyfile(src, dst)
sys.path.append(workdir)
def increment_module_value():
modfile = open(dst, 'a')
modfile.write('MyOtherObjectType.value += 1' + os.linesep)
modfile.flush()
modfile.close()
try:
os.remove(dst + 'c')
except:
os.remove(dst + 'o')
class TestModuleReloading(unittest.TestCase):
def testModuleReloading(self):
'''Test module reloading with on-the-fly modifications.'''
import test_module
self.assertEqual(test_module.MyOtherObjectType.value, 10)
increment_module_value()
reload(sys.modules['test_module'])
self.assertEqual(test_module.MyOtherObjectType.value, 11)
reload(sys.modules['test_module'])
self.assertEqual(test_module.MyOtherObjectType.value, 11)
increment_module_value()
reload(sys.modules['test_module'])
self.assertEqual(test_module.MyOtherObjectType.value, 12)
if __name__ == "__main__":
unittest.main()
|