def assertIO(self, input, output):
stdout, stdin = sys.stdout, sys.stdin
sys.stdout, sys.stdin = StringIO(), StringIO(input)
A.a()
sys.stdout.seek(0)
out = sys.stdout.read()[:-1]
sys.stdout, sys.stdin = stdout, stdin
self.assertEqual(out, output)
def test_impor(self):
import importlib
import os
import A
A.a()
os.rename('A.py', 'A_change.py')
os.rename('B.py', 'A.py')
importlib.reload(A)
A.a()
os.rename('A.py', 'B.py')
os.rename('A_change.py', 'A.py')
