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')